A comparative analysis of classical and machine learning methods for forecasting TB/HIV co-infection.

Journal: Scientific reports
PMID:

Abstract

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

  • André Abade
    Computer Science, Federal Institute of Science and Technology, Barra do Garças, Brazil.
  • Lucas Faria Porto
    Department of Health Sciences, Barra do Garças, Campus Araguaia, Federal University of Mato Grosso, Cuiabá, Mato Grosso, Brazil.
  • Alessandro Rolim Scholze
    State University of Northern Paraná, Luiz Meneghel Campus, Bandeirantes, Paraná, Brazil.
  • Daniely Kuntath
    Department of Health Sciences, Barra do Garças, Campus Araguaia, Federal University of Mato Grosso, Cuiabá, Mato Grosso, Brazil.
  • Nathan da Silva Barros
    Department of Health Sciences, Barra do Garças, Campus Araguaia, Federal University of Mato Grosso, Cuiabá, Mato Grosso, Brazil.
  • Thaís Zamboni Berra
    University of São Paulo College of Nursing at Ribeirão Preto, Ribeirão Preto, São Paulo, Brazil.
  • Antonio Carlos Vieira Ramos
    Nursing Department, University of the State of Minas Gerais, Passos, Minas Gerais, Brazil.
  • Ricardo Alexandre Arcêncio
    Universidade de São Paulo - Ribeirão Preto (SP), Brazil.
  • Josilene Dália Alves
    Department of Health Sciences, Barra do Garças, Campus Araguaia, Federal University of Mato Grosso, Cuiabá, Mato Grosso, Brazil.