Abstract
Motivation Prediction of drug resistance and identification of its mechanisms in bacteria such as Mycobacterium tuberculosis, the etiological agent of tuberculosis, is a challenging problem. Solving this problem requires a transparent, accurate, and flexible predictive model. The methods currently used for this purpose rarely satisfy all of these criteria. On the one hand, approaches based on testing strains against a catalogue of previously identified mutations often yield poor predictive performance; on the other hand, machine learning techniques typically have higher predictive accuracy, but often lack interpretability and may learn patterns that produce accurate predictions for the wrong reasons. Current interpretable methods may either exhibit a lower accuracy or lack the flexibility needed to generalize them to previously unseen data.
Contribution In this paper we propose a novel technique, inspired by the group testing and Boolean compressed sensing, which yields highly accurate predictions, interpretable results, and is flexible enough to be optimized for various evaluation metrics at the same time.
Results We test the predictive accuracy of our approach on five first-line and seven second-line antibiotics used for treating tuberculosis. We find that it has a higher or comparable accuracy to that of commonly used machine learning models, and is able to identify variants in genes with previously reported association to drug resistance. Our method is intrinsically interpretable, and can be customized for different evaluation metrics. Our implementation is available at github.com/hoomanzabeti/INGOT_DR and can be installed via The Python Package Index (Pypi) under ingotdr. This package is also compatible with most of the tools in the Scikit-learn machine learning library.
Competing Interest Statement
The authors have declared no competing interest.
Footnotes
Full list of author information is available at the end of the article
New title. Method section is revised. New results and more ML models are included in the comparison. Conclusion is revised.
https://github.com/AmirHoseinSafari/M.tuberculosis-dataset-for-drug-resistant
Abbreviations
- CS
- compressed sensing
- FN
- false negatives
- FP
- false positives
- GWAS
- genome-wide association study
- ILP
- integer linear programming
- LD
- linkage disequilibrium
- LR
- logistic regression
- MTB
- Mycobacterium tuberculosis
- RF
- random forests
- SCM
- set covering machine
- SNP
- single-nucleotide polymorphism
- SVM
- support vector machine
- WGS
- whole-genome sequencing