A comparative analysis of classical and machine learning methods for forecasting TB/HIV co-infection
TB/HIV coinfection poses a complex public health challenge. Accurate forecasting of future trends is essential for efficient resource allocation and intervention strategy development. This study compares classical statistical and machine learning models to predict TB/HIV coinfection cases stratified by gender and the general populations. We analyzed time series data using exponential smoothing and ARIMA to establish the baseline trend and seasonality. Subsequently, machine learning models (SVR, XGBoost, LSTM, CNN, GRU, CNN-GRU, and CNN-LSTM) were employed to capture the complex dynamics and inherent non-linearities of TB/HIV coinfection data. Performance metrics (MSE, MAE, sMAPE) and the Diebold-Mariano test were used to evaluate the model performance. Results revealed that Deep Learning models, particularly Bidirectional LSTM and CNN-LSTM, significantly outperformed classical methods. This demonstrates the effectiveness of Deep Learning for modeling TB/HIV coinfection time series and generating more accurate forecasts.
Authors
- Thaís Zamboni Berra (ORCID: https://orcid.org/0000-0002-4163-8719)
- Antônio Carlos Vieira Ramos (ORCID: https://orcid.org/0000-0002-7862-1355)
- Alessandro Rolim Scholze (ORCID: https://orcid.org/0000-0003-4045-3584)
- Lucas Faria Porto (ORCID: https://orcid.org/0000-0003-2170-427X)
- Nathan da Silva Barros (ORCID: https://orcid.org/0009-0008-1691-1577)
- André Abade (ORCID: https://orcid.org/0000-0001-9771-9123)
- Ricardo Alexandre Arcêncio (ORCID: https://orcid.org/0000-0003-4792-8714)
- Josilene Dália Alves (ORCID: https://orcid.org/0000-0001-5007-9536)
- Daniely Kuntath
Institutions
- Universidade de Ribeirão Preto (BR)
- Universidade de São Paulo (BR)
- Universidade do Estado de Minas Gerais (BR)
- Instituto Federal de Educação, Ciência e Tecnologia de Mato Grosso (BR)
- Universidade Federal de Mato Grosso (BR)
- Universidade Estadual do Norte do Paraná (BR)
Publication Details
- Journal
- Scientific Reports
- Published
- 2024-08-16
- DOI
- https://doi.org/10.1038/s41598-024-69580-4
- Citations
- 16
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 5.56
Funders
- Conselho Nacional de Desenvolvimento Científico e Tecnológico
- Fundação de Amparo à Pesquisa do Estado de Mato Grosso
- Universidade Federal de Mato Grosso do Sul