Evaluating Machine Learning Models for Predicting Late Leprosy Diagnosis by Physical Disability Grade in Brazil (2018-2022).

Journal: Tropical medicine and infectious disease
Published Date:

Abstract

The severity of physical disability at leprosy diagnosis reflects the timeliness of case detection and the effectiveness of disease surveillance. This study evaluates machine learning models to predict factors associated with late leprosy diagnosis-defined as grade 2 physical disability (G2D)-in Brazil from 2018 to 2022. Using an observational cross-sectional design, we analyzed data from the Notifiable Diseases Information System and trained four machine learning models: Random Forest, LightGBM, CatBoost, XGBoost, and an Ensemble model. Model performance was assessed through accuracy, area under the receiver operating characteristic curve (AUC-ROC), recall, precision, F1 score, specificity, and Matthew's correlation coefficient (MCC). An increasing trend in G2D prevalence was observed, averaging 11.6% over the study period and rising to 13.1% in 2022. The Ensemble model and LightGBM demonstrated the highest predictive performance, particularly in the north and northeast regions (accuracy: 0.85, AUC-ROC: 0.93, recall: 0.90, F1 score: 0.83, MCC: 0.70), with similar results in other regions. Key predictors of G2D included the number of nerves affected, clinical form, education level, and case detection mode. These findings underscore the potential of machine learning to enhance early detection strategies and reduce the burden of disability in leprosy, particularly in regions with persistent health disparities.

Authors

  • Lucia Rolim Santana de Freitas
    Faculdade de Medicina, Universidade de Brasília, Brasília 70910-900, Brazil.
  • José Antônio Oliveira de Freitas
    Departamento de Matemática, Universidade de Brasília, Brasília 70910-900, Brazil.
  • Gerson Oliveira Penna
    Escola de Governo Fiocruz Brasília, Fundação Oswaldo Cruz, Brasília 70904-130, Brazil.
  • Elisabeth Carmen Duarte
    Faculdade de Medicina, Universidade de Brasília, Brasília 70910-900, Brazil.

Keywords

No keywords available for this article.