Benchmarking the translational potential of AI-based drug-resistance prediction from Mycobacterium tuberculosis whole-genome sequencing data
C X LiuiD, Haoyue ZhuiD, Pengfei Zhou, Nguyen Trung Thanh, Nguyen Quoc DatiD, Ines I. AtmosukartoiD, Io Hong CheongiD, Zisis KozlakidisiD, Wiku Adisasmito, Xiaoqi ZhengiD, H Y Wang, Yang Yang
Abstract
BackgroundTuberculosis, especially drug-resistant tuberculosis (DR-TB) including multidrug-resistant (MDR) and extensively drug-resistant (XDR) strains, remains a leading cause of infectious death worldwide. The rapid accumulation of whole-genome sequencing (WGS) data had spurred numerous computational methods for predicting antimicrobial resistance in Mycobacterium tuberculosis. However, heterogeneous datasets, preprocessing pipelines, and evaluation protocols have made fair comparisons impossible and have hindered clinical translation. A critical yet missing resource is a large-scale, unified benchmark to systematically assess and compare existing methods. MethodsWe curated an integrated MTB WGS-phenotypic drug susceptibility testing (pDST) dataset from three sources: the CRyPTIC dataset (Comprehensive Resistance Prediction for Tuberculosis: an International Consortium), a published multi-study compilation, and newly curated literature-derived datasets. The final benchmark contains 54,364 paired WGS-pDST records with broad geographic, lineage, and drug coverage. After harmonizing phenotypes and generating standardized variant features, we evaluated seven models (including classical machine learning and deep learning architectures) across 18 drug-level and six clinical resistance category prediction tasks. ResultsXGBoost achieved the highest mean drug-level AUPRC (0.674) and F1-score (0.620) and ranked first in AUPRC for 11 of 18 drugs, whereas WDNN achieved the highest mean AUROC. Random forest yielded the highest mean specificity (0.956) and accuracy (0.933), whereas logistic regression achieved the highest mean recall (0.774), highlighting distinct clinical trade-offs. Drug-level difficulty was highly heterogeneous: rifampicin and isoniazid were predicted robustly, whereas bedaquiline, delamanid, linezolid, and clofazimine remained persistently difficult. In clinical resistance category evaluation, RR-TB, MDR-TB, and pan-susceptibility were well predicted, but XDR-TB and other resistance categories constituted major bottlenecks. ConclusionsUnder the largest unified benchmark to date, classical machine-learning methods, particularly XGBoost, provided the strongest precision-recall and F1 performance overall, while neural models remained competitive by AUROC. Emerging drugs (bedaquiline, delamanid, linezolid, clofazimine) and XDR cases remain persistently difficult to predict, identifying key bottlenecks for future method development. This benchmark can serve as a community standard for evaluating MTB resistance prediction and the provided evaluation pipeline offers an actionable baseline for regulatory qualification and clinical decision support system validation, accelerating the translation of WGS-based resistance prediction into practice.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 0
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- TB-Bench: A Systematic Benchmark of Machine Learning and Deep Learning Methods for Second-Line TB Drug Resistance Prediction
V P Brintha, Saish Jaiswal, … Manikandan Narayanan·bioRxiv·Apr 13, 2026
- Geographic heterogeneity impacts drug resistance predictions in Mycobacterium tuberculosis
Guo Liang Gan, Matthew H. Nguyen, … Leonid Chindelevitch·bioRxiv·Sep 17, 2020
- Genomic Diagnostics for Drug-Resistant Mycobacterium tuberculosis: Computational Prediction of Antimicrobial Resistance
Mohammadali Serajian, Yunheng Han, Christina Boucher·bioRxiv·May 25, 2026
- Deep learning predicts tuberculosis drug resistance status from genome sequencing data
Michael L. Chen, Akshith Doddi, … Maha Reda Farhat·EBioMedicinePeer-reviewed·Apr 29, 2019
- GenTB: A user-friendly genome-based predictor for tuberculosis resistance powered by machine learning
Matthias I. Gröschel, Martin Owens, … Maha Reda Farhat·Genome MedicinePeer-reviewed·Aug 30, 2021
- Whole genome sequencing for drug resistance profile prediction in Mycobacterium tuberculosis
S. M. Gygli, P. M. Keller, … E. C. Böttger·bioRxiv·Aug 28, 2018