Back

Identifying Optimal Machine Learning Approaches for Microbiome-Metabolomics Integration with Stable Feature Selection

Palmer, S. N.; Mishra, A. A.; Gan, S.; Liu, D.; Koh, A.; Zhan, X.

2025-06-30 bioinformatics
10.1101/2025.06.21.660858 bioRxiv
Show abstract

Microbiome research has been limited by methodological inconsistencies. Taxonomy-based profiling presents challenges such as data sparsity, variable taxonomic resolution, and the reliance on DNA-based profiling, which provides limited functional insight. Multi-omics integration has emerged as a promising approach to link microbiome composition with function. However, the lack of standardized methodologies and inconsistencies in machine learning strategies has hindered reproducibility. Additionally, while machine learning can be used to identify key microbial and metabolic features, the stability of feature selection across models and data types remains underexplored, despite its importance for downstream experimental validation and biomarker discovery. Here, we systematically compare Elastic Net, Random Forest, and XGBoost across five multi-omics integration strategies: Concatenation, Averaged Stacking, Weighted Non-negative Least Squares (NNLS), Lasso Stacking, and Partial Least Squares (PLS), as well as individual omics models. We evaluate performance across 588 binary and 735 continuous models using human gut microbiome-derived metabolomics and taxonomic data derived from metagenomics shotgun sequencing data. Additionally, we assess the impact of feature reduction on model performance and feature selection stability. Among the approaches tested, Random Forest combined with NNLS yielded the highest overall performance across diverse datasets. Tree-based methods also demonstrated consistent feature selection across data types and dimensionalities. These results demonstrate how integration strategies, algorithm selection, data dimensionality, and response type impact both predictive performance and the stability of selected features in multi-omics microbiome modeling. Key PointsO_LIA total of 1,323 models were developed to comprehensively evaluate prediction performance and the robustness of feature selection for human gut microbiome (metabolomics and taxonomy from metagenomics shotgun sequencing) datasets. These models included three widely used machine learning algorithms - Elastic Net, Random Forest and XGBoost - applied across five integration strategies and single-omics approaches on datasets with binary and continuous outcomes. C_LIO_LIFor continuous outcomes, Random Forest combined with NNLS integration achieved the highest performance and maintained strong predictive performance across full-dimensional and feature-reduced datasets. C_LIO_LIFor binary outcomes, Random Forest consistently performed well regardless of the integration strategy. Notably, single-omics models, especially those using metabolomics data, outperformed integrative approaches. C_LIO_LITree-based models demonstrated greater consistency in feature selection across different dimensionalities and integration strategies. C_LI

Matching journals

The top 8 journals account for 50% of the predicted probability mass.

50% of probability mass above

"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.