Benchmarking and optimizing microbiome-based bioinformatics workflow for non-invasive detection of intestinal tumors

Abstract

Background: The human gut microbiome is closely linked to disease states, offering substantial potential for novel disease detection tools based on machine learning (ML). However, variations in feature types, data preprocessing strategies, feature selection strategies, and classification algorithms can all influence the model’s predictive performance and robustness.

Methods: To develop an optimized and systematically evaluated workflow, we conducted a comprehensive evaluation of ML methods for classifying colorectal cancer and adenoma using 4,217 fecal samples from diverse global regions. The area under the receiver operating characteristic curve was used to quantify model performance. We benchmarked 6,468 unique analytical pipelines, defined by distinct tools, parameters, and algorithms, utilizing a dual validation strategy that included both cross-validation and leave-one-dataset-out validation.

Results: Our findings revealed that shotgun metagenomic (WGS) data generally outperformed 16S ribosomal RNA gene (16S) sequencing data, with features at the species-level genome bin, species, and genus levels demonstrating the greatest discriminatory power. For 16S data, Amplicon Sequence Variant-based features yielded the best disease classification performance. Furthermore, the application of specific feature selection tools, such as the Wilcoxon rank-sum test method, combined with appropriate data normalization, also optimized model performance. Finally, in the algorithm selection phase, we identified ensemble learning models (eXtreme Gradient Boosting and Random Forest) as the best-performing classifiers.

Conclusion: Based on the comprehensive evaluation results, we developed an optimized Microbiome-based Detection Framework (MiDx) and validated its robust generalizability on an independent dataset, offering a systematic and practical framework for future 16S and WGS-based intestinal disease detection.

Keywords

Colorectal cancer / adenoma / machine learning / benchmarking