Abstract:Scientific research is shedding light on the interaction of the gut microbiome with the human host and on its role in human health. Existing machine learning methods have shown great potential in discriminating healthy from diseased microbiome states. Most of them leverage shotgun metagenomic sequencing to extract gut microbial species-relative abundances or strain-level markers. Each of these gut microbial profiling modalities showed diagnostic potential when tested separately; however, no existing approach combines them in a single predictive framework. Here, we propose the Multimodal Variational Information Bottleneck (MVIB), a novel deep learning model capable of learning a joint representation of multiple heterogeneous data modalities. MVIB achieves competitive classification performance while being faster than existing methods. Additionally, MVIB offers interpretable results. Our model adopts an information theoretic interpretation of deep neural networks and computes a joint stochastic encoding of different input data modalities. We use MVIB to predict whether human hosts are affected by a certain disease by jointly analysing gut microbial species-relative abundances and strain-level markers. MVIB is evaluated on human gut metagenomic samples from 11 publicly available disease cohorts covering 6 different diseases. We achieve high performance (0.80 < ROC AUC < 0.95) on 5 cohorts and at least medium performance on the remaining ones. We adopt a saliency technique to interpret the output of MVIB and identify the most relevant microbial species and strain-level markers to the model's predictions. We also perform cross-study generalisation experiments, where we train and test MVIB on different cohorts of the same disease, and overall we achieve comparable results to the baseline approach, i.e. the Random Forest. Further, we evaluate our model by adding metabolomic data derived from mass spectrometry as a third input modality. Our method is scalable with respect to input data modalities and has an average training time of < 1.4 seconds. The source code and the datasets used in this work are publicly available.

MDITRE: Scalable and Interpretable Machine Learning for Predicting Host Status from Temporal Microbiome Dynamics

Human Host Status Inference from Temporal Microbiome Changes Via Recurrent Neural Networks.

MMETHANE: interpretable AI for predicting host status from microbial composition and metabolomics data

Longitudinal Microbiome-based Interpretable Machine Learning for Identification of Time-Varying Biomarkers in Early Prediction of Disease Outcomes

Host Trait Prediction of Metagenomic Data for Topology-Based Visualization

MicroHDF: predicting host phenotypes with metagenomic data using a deep forest-based framework

BMCMDA: a Novel Model for Predicting Human Microbe-Disease Associations Via Binary Matrix Completion

DeepMicro: Deep Representation Learning for Disease Prediction Based on Microbiome Data

Statistical and Machine Learning Techniques in Human Microbiome Studies: Contemporary Challenges and Solutions

The Human Microbiota in Health and Disease

Computational Biology and Machine Learning Approaches to Understand Mechanistic Microbiome-Host Interactions

Predicting Microbiome Growth Dynamics under Environmental Perturbations

Microbiome-based disease prediction with multimodal variational information bottlenecks

Latent Space Temporal Model of Microbial Abundance to Predict Domination and Bacteremia

Multitask Knowledge-primed Neural Network for Predicting Missing Metadata and Host Phenotype based on Human Microbiome

Applications of Machine Learning in Human Microbiome Studies: A Review on Feature Selection, Biomarker Identification, Disease Prediction and Treatment

A comparative study of supervised and unsupervised machine learning algorithms applied to human microbiome

A Review and Tutorial of Machine Learning Methods for Microbiome Host Trait Prediction

Multimodal deep learning applied to classify healthy and disease states of human microbiome

mbtransfer: Microbiome intervention analysis using transfer functions and mirror statistics

Machine learning applications in microbial ecology, human microbiome studies, and environmental monitoring