Machine Learning for Environmental Antimicrobial Resistance Risk Prediction
DOI:
https://doi.org/10.51137/wrp.ijdht.550Keywords:
Antimicrobial Resistance, Environmental Resistome, Machine Learning, One Health, Predictive ModelingAbstract
Antimicrobial resistance (AMR) poses a major global health threat, and environmental reservoirs play a critical role in the dissemination of antimicrobial resistance genes (ARGs). This study proposes a machine learning (ML) framework to predict environmental ARG dissemination risk using real-world surveillance data. Gene-level, environmental, spatial, and abundance-related features were integrated into a supervised learning pipeline. Nine ML models were systematically evaluated, including Logistic Regression, Random Forest, Gradient Boosting, XGBoost, LightGBM, and neural networks. Ensemble-based models demonstrated superior performance, with LightGBM achieving the highest accuracy (98.96%) and F1-score (97.24%). Feature importance and SHAP-based explainability analyses identified ARG abundance, habitat type, and geographic attributes as dominant drivers of dissemination risk. The findings demonstrate that ML can effectively capture complex nonlinear ARG-environment interactions and provide interpretable, risk-oriented predictions. The proposed framework offers a scalable approach for environmental AMR surveillance and supports integration into digital health systems within a One Health context.
References
Bengtsson-Palme, J., Kristiansson, E., & Larsson, D. G. J. (2018). Environmental factors influencing antibiotic resistance. FEMS Microbiology Reviews, 42, 68–80. https://doi.org/10.1093/femsre/fux053
Huijbers, P. M. C., Blaak, H., de Jong, M. C. M., et al. (2019). The role of the environment in the transmission of antimicrobial resistance. The Lancet Planetary Health, 3, e145–e153. https://doi.org/10.1016/S2542-5196(19)30052-9
Larsson, D. G. J., & Flach, C.-F. (2022). Antibiotic resistance in the environment. Nature Reviews Microbiology, 20, 257–269. https://doi.org/10.1038/s41579-021-00649-x
Manaia, C. M. (2020). Assessing antibiotic resistance in the environment. Current Opinion in Environmental Science & Health, 16, 100–107. https://doi.org/10.1016/j.coesh.2020.04.003
Singer, A. C., Shaw, H., Rhodes, V., & Hart, A. (2016). Review of antimicrobial resistance in the environment. Environment International, 85, 1–10. https://doi.org/10.1016/j.envint.2015.08.014
Murray, C. J. L., Ikuta, K. S., Sharara, F., et al. (2022). Global burden of bacterial antimicrobial resistance in 2019: A systematic analysis. The Lancet, 399(10325), 629–655. https://doi.org/10.1016/S0140-6736(21)02724-0
World Health Organization. (2024). Global antimicrobial resistance and use surveillance system (GLASS) report 2024. WHO. https://www.who.int/teams/surveillance-prevention-control-AMR
FAO, UNEP, & World Health Organization. (2022). One Health joint plan of action on antimicrobial resistance (2022–2026). WHO. https://www.who.int/initiatives/one-health
Ju, F., & Zhang, T. (2015). Bacterial assembly and antibiotic resistance in wastewater systems. Current Opinion in Microbiology, 27, 14–19. https://doi.org/10.1016/j.mib.2015.08.003
Karkman, A., Pärnänen, K., & Larsson, D. G. J. (2019). Fecal pollution explains antibiotic resistance gene abundances. The ISME Journal, 13, 304–317. https://doi.org/10.1038/s41396-018-0277-0
Zhang, A. N., Li, L. G., Yin, X., et al. (2021). Occurrence of antibiotic resistance genes in wastewater treatment plants. Environmental Science & Technology, 55, 6888–6898. https://doi.org/10.1021/acs.est.1c05235
Zhang, Q., Zhang, Z., Lu, T., et al. (2021). Occurrence and removal of ARGs in aquatic environments. Science of the Total Environment, 759, 143476. https://doi.org/10.1016/j.scitotenv.2020.143476
Breiman, L. (2001). Random forests. Machine Learning, 45, 5–32. https://doi.org/10.1023/A:1010933404324
Chen, T., & Guestrin, C. (2016). XGBoost: A scalable tree boosting system. Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, 785–794. https://doi.org/10.1145/2939672.2939785
Ke, G., Meng, Q., Finley, T., et al. (2017). LightGBM. Advances in Neural Information Processing Systems. https://proceedings.neurips.cc
Lundberg, S. M., & Lee, S. I. (2017). A unified approach to interpreting model predictions. Advances in Neural Information Processing Systems. https://proceedings.neurips.cc
Molnar, C. (2022). Interpretable machine learning (2nd ed.). Leanpub.
Ma, R. A., Ding, Y. H., Zhong, S., et al. (2025). Machine learning prediction of environmental resistome abundance. Environment International, 202, 109655. https://doi.org/10.1016/j.envint.2025.109655
Walsh, C., & Toleman, M. (2024). Pitfalls in machine learning-based AMR prediction. PLOS Computational Biology, 20, e1011987. https://doi.org/10.1371/journal.pcbi.1011987
Nsubuga, M., Galiwango, R., Jjingo, D., & Mboowa, G. (2024). Generalizability of machine learning for AMR prediction. BMC Genomics, 25, 287. https://doi.org/10.1186/s12864-024-10214-4
Yang, Y., Li, B., Zou, S., et al. (2024). Environmental AMR forecasting using machine learning. Environmental Modelling & Software, 173, 105908. https://doi.org/10.1016/j.envsoft.2024.105908
Wu, L. S., Li, X., Huang, Y., et al. (2024). AI tools for identifying antibiotic resistance genes. Frontiers in Microbiology, 15, 1437602. https://doi.org/10.3389/fmicb.2024.1437602
Li, H., Chen, Y., Wang, J., et al. (2024). Integrating metagenomics and machine learning for ARG risk prediction. Bioinformatics. https://doi.org/10.1093/bioinformatics/btad845
Gootenberg, D. B., & Turnbaugh, P. J. (2025). Translating AMR prediction into actionable risk. Trends in Microbiology. https://doi.org/10.1016/j.tim.2024.11.003
Hendriksen, R. S., Munk, P., Njage, P., et al. (2019). Global monitoring of antimicrobial resistance. Nature Communications, 10, 1124. https://doi.org/10.1038/s41467-019-09145-6
Downloads
Published
Issue
Section
License
Copyright (c) 2026 Afsana Munni, Gwaliwa Peter Mashaka (Author)
This work is licensed under a Creative Commons Attribution 4.0 International License.
