Early Prediction of Severe Dengue Using Machine Learning
conference paperAuthors: Janani Halwala, Sulanie Perera, Uvini Ranaweera, Shehan Silva
Abstract: Dengue remains a major public health challenge in tropical regions such as Sri Lanka, where early identification of severe dengue is critical to reducing mortality and optimizing resource allocation. This study developed and evaluated machine learning models to predict severe dengue progression within 1, 2, and 3 days of hospital admission. Since clinical data contain noise, several preprocessing steps were performed before fitting any model, including feature selection and handling of class imbalance. Multiple algorithms, including Random Forest, Extra Trees, Gradient Boosting, XGBoost, LightGBM, CatBoost, and Support Vector Machines, were trained using stratified cross-validation and hyperparameter tuning. Model performance was evaluated using standard classification metrics, with an emphasis on sensitivity to ensure early detection of high-risk patients. The results demonstrated that the Extra Trees model achieved the best performance within 2 days of admission, with a recall of 93.4% and an F1-score of 0.937, outperforming both time windows of day 1 and day 3. SHAP analysis revealed plasma leakage (USS), platelet count, white blood cell count, and liver function markers (AST/ALT) as the most influential predictors, aligning with established dengue pathophysiology. This study uniquely evaluates severe dengue prediction across multiple early time windows (days 1-3), finding that the 2-day window achieves the best sensitivity. Unlike prior research focusing on diagnosis or single timepoints, the proposed framework uses routinely collected hospital data to provide practical clinical decision support for timely interventions in resource-limited, dengue-endemic regions.
Keywords: Sensitivity, Mortality, Predictive models, Feature extraction
Presented: 10th International Conference on Information Technology Research ICITR 20255
DOI: https://doi.org/10.1109/ICITR69413.2025.11353903
Cite the paper:
@InProceedings{10.1109/ICITR69413.2025.11353903.,
author = "J. Halwala, S. Perera, U. Ranaweera and S. Silva",
title = "Early Prediction of Severe Dengue Using Machine Learning",
year = "2025",
publisher = "IEEE",
pages="1--6"
}