Using Conditional Generative Adversarial Networks to Boost the Performance of Machine Learning in Microbiome Datasets
Reiman, D.; Dai, Y.
Show abstract
The microbiome of the human body has been shown to have profound effects on physiological regulation and disease pathogenesis. However, association analysis based on statistical modeling of microbiome data has continued to be a challenge due to inherent noise, complexity of the data, and high cost of collecting large number of samples. To address this challenge, we employed a deep learning framework to construct a data-driven simulation of microbiome data using a conditional generative adversarial network. Conditional generative adversarial networks train two models against each other while leveraging side information learn from a given dataset to compute larger simulated datasets that are representative of the original dataset. In our study, we used a cohorts of patients with inflammatory bowel disease to show that not only can the generative adversarial network generate samples representative of the original data based on multiple diversity metrics, but also that training machine learning models on the synthetic samples can improve disease prediction through data augmentation. In addition, we also show that the synthetic samples generated by this cohort can boost disease prediction of a different external cohort.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
- Decoding the Language of Microbiomes: Leveraging Patterns in 16S Public Data using Word-Embedding Techniques and Applications in Inflammatory Bowel Disease 95%
- CBEA: Competitive balances for taxonomic enrichmentanalysis 95%
- MiMeNet: Exploring Microbiome-Metabolome Relationships using Neural Networks 95%
Similar papers in this journal
- parafac4microbiome: Exploratory analysis of longitudinal microbiome data using Parallel Factor Analysis 95%
- Directional Gaussian Mixture Models of the gut microbiome elucidate microbial spatial structure 94%
- A mixed model approach for estimating drivers of microbiota community composition and differential taxonomic abundance 94%
Similar papers in this journal
- VBayesMM: Variational Bayesian neural network to prioritize important relationships of high-dimensional microbiome multiomics data 95%
- Feature selection with vector-symbolic architectures: a case study on microbial profiles of shotgun metagenomic samples of colorectal cancer 94%
- Comprehensive evaluation of methods for differential expression analysis of metatranscriptomics data 94%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.