PhD Dissertation Proposal: Mahbuba Tasmin, Integrating Biological Signal to improve Generalizability and Interpretability of Machine Learning in Genomics
Content
Speaker:
Abstract:
The rise of drug-resistant Mycobacterium tuberculosis remains a critical global health challenge, motivating rapid genotype-based approaches for antimicrobial resistance prediction. While machine learning and foundation models provide promising tools for mapping genomic variation to resistance phenotypes, their clinical utility is limited by two persistent challenges: scarce clinically labeled data and limited biological interpretability. Sequence-only models can achieve strong predictive performance, but they may fail to reliably recover known resistance-conferring variants and can rely on spurious genomic correlations. This thesis develops a biologically- grounded machine learning framework that improves generalization and interpretability by integrating three-dimensional protein structure, evolutionary constraint, and molecular energetics into pathogen genotype–phenotype prediction.
The thesis is organized around five connected contributions. First, I establish BIG-TB, a benchmark of more than 17,000 M. tuberculosis genomes that evaluates both phenotype prediction and resistance-variant discovery, exposing the interpretability limits of current sequence-based models. Second, using a proteome-wide unsupervised analysis of more than 31,000 clinical isolates, I show that resistance-associated mutations cluster in three-dimensional protein space, including in non-essential genes, supporting structural proximity as a biologically meaningful signal for resistance. Third, I incorporate this structural information directly into predictive models through Fused Ridge Regression, where coefficients are regularized according to residue adjacency in 3D protein space to improve generalization when learning from scarce training instances with highly sparse and redundant features.
Building on these structural insights, the fourth contribution develops a multimodal forecasting framework for variants of uncertain significance. By combining 3D structural context, Rosetta- derived biophysical energetics, and ESM-2 evolutionary likelihoods, this model achieves strong performance on a held-out test set, and forecasts future World Health Organization variant reclassifications with 79.1% accuracy. The framework reveals distinct biological regimes: resistance in essential genes is more strongly associated with structural and energetic constraints, whereas resistance in non-essential genes is more strongly associated with sequence-level evolutionary likelihood. Finally, the thesis develops methods for biologically constrained protein data augmentation using single-amino-acid substitutions filtered by ESM-2 pseudo-perplexity, improving model performance under limited training data in a biologically plausible fashion. Together, this thesis integrates biological signals into traditional sequence modeling to build interpretable, data-efficient AI systems for antimicrobial resistance forecasting.
Advisor:
Anna Green