Machine learning for improved dengue diagnosis, Puerto Rico
Zachary J. Madewell,Dania M. Rodriguez,Maile B. Thayer,Vanessa Rivera-Amill,Jomil Torres Aponte,Melissa Marzan-Rodriguez,Gabriela Paz-Bailey,Laura E. Adams,Joshua M. Wong
DOI: https://doi.org/10.1101/2024.11.13.24317272
2024-11-13
Abstract:Background: Diagnosing dengue accurately, especially in resource-limited settings, remains challenging due to overlapping symptoms with other febrile illnesses and limitations of current diagnostic methods. This study aimed to develop machine learning (ML) models that leverage readily available clinical data to improve diagnostic accuracy for dengue, potentially offering a more accessible and rapid diagnostic tool for healthcare providers.
Methods: We used data from the Sentinel Enhanced Dengue Surveillance System (SEDSS) in Puerto Rico (May 2012-June 2024). SEDSS primarily targets acute febrile illness but also includes cases with other symptoms during outbreaks (e.g., Zika and COVID-19). ML models (logistic regression, random forest, support vector machine, artificial neural network, adaptive boosting, light gradient boosting machine [LightGBM], and extreme gradient boosting [XGBoost]) were evaluated across different feature sets, including demographic, clinical, laboratory, and epidemiological variables. Model performance was assessed using the area under the receiver operating characteristic curve (AUC), where higher AUC values indicate better performance in distinguishing dengue cases from non-dengue cases.
Results: Among 49,679 patients in SEDSS, 1,640 laboratory-confirmed dengue cases were identified. The XGBoost and LightGBM models achieved the highest diagnostic accuracy, with AUCs exceeding 90%, particularly with comprehensive feature sets. Incorporating predictors such as monthly dengue incidence, leukopenia, thrombocytopenia, rash, age, and absence of nasal discharge significantly enhanced model sensitivity and specificity for diagnosing dengue. Adding more relevant clinical and epidemiological features consistently improved the models' ability to correctly identify dengue cases.
Conclusions: ML models, especially XGBoost and LightGBM, show promise for improving diagnostic accuracy for dengue using widely accessible clinical data, even in resource-limited settings. Future research should focus on developing user-friendly tools, such as mobile apps, web-based platforms, or clinical decision systems integrated into electronic health records, to implement these models in clinical practice and exploring their application for predicting dengue.