Optimization of Machine Learning Models for Risk Prediction of DHF Spread to Support Management Strategies in Urban Areas

Yesica Devis, Muhamadiah Muhamadiah, Yulanda Yulanda, Yuda Irawan, Refni Wahyuni

Abstract


Dengue fever is an endemic disease that poses a serious threat to public health in tropical regions such as Indonesia. Efforts to control this disease require a data-based approach that is able to accurately predict the level of risk so that interventions can be targeted. This study aims to develop a predictive model of DHF risk using ensemble stacking method optimized with Optuna algorithm and integrated into an interactive dashboard based on Streamlit. The dataset used includes environmental, climate, and socio-demographic indicators from 2015 to 2024 with a total of 1,440 data entries. The preprocessing process includes normalization with Standard Scaler, feature selection using LASSO, and label data balancing with the SMOTE method. Model validation was performed using 10-Fold Cross Validation to ensure model generalization to new data. The stacking model is built with three basic algorithms, namely SVM, KNN, and Random Forest, which are combined using Logistic Regression as a meta-learner. The evaluation results show that the model is able to achieve an average accuracy of 97.57%, with high precision, recall, and f1-score values in all three prediction classes (low, medium, high). The ROC-AUC for each class also showed near-perfect performance. The implementation of the model in the Streamlit dashboard allows non-technical users such as health center or health office staff to perform regional risk prediction and obtain data-driven intervention recommendations automatically. This research not only contributes to the development of predictive technology, but also strengthens evidence-based health promotion practices in urban areas. Further research is recommended to integrate IoT-based real-time data and expand the scope of application areas.


Keywords


DHF; Machine Learning; Stacking Ensemble; OPTUNA; Public Health

Full Text:

PDF

References


S. Masyeni, I. M. W. Wardhana, and F. Nainu, “Cytokine profiles in dengue fever and dengue hemorrhagic fever: A study from Indonesia.,” Narra J, vol. 4, no. 1, p. e309, Apr. 2024, doi: 10.52225/narra.v4i1.309.

R. Bur et al., “Early intervention of 5% albumin shown superior control of vascular integrity and function compared to ringer’s lactatein hospitalized adult with grade I & II Dengue hemorrhagic fever: a multicenter randomized controlled trial in Indonesia,” Trop. Dis. Travel Med. Vaccines, vol. 10, no. 1, p. 20, 2024, doi: 10.1186/s40794-024-00230-3.

M. P. Patel, V. M. Oza, H. B. Tanna, A. D. Khadela, P. D. Bharadia, and J. K. Patel, “Current Perspectives in Dengue Hemorrhagic Fever,” in Rising Contagious Diseases, 2024, pp. 72–86. doi: https://doi.org/10.1002/9781394188741.ch8.

M. Riaz et al., “Evaluation of clinical and laboratory characteristics of dengue viral infection and risk factors of dengue hemorrhagic fever: a multi-center retrospective analysis,” BMC Infect. Dis., vol. 24, no. 1, p. 500, 2024, doi: 10.1186/s12879-024-09384-z.

E. A. Giofandi, P. Purwantiningrum, F. Madino, and A. Lumbantobing, “Analisis Faktor Spasial Terhadap Kejadian Demam Berdarah Dengue Menggunakan Pendekatan Geographically Weighted Regression di Kota Pekanbaru, Provinsi Riau,” J. Ilmu Lingkung., vol. 22, no. 1, pp. 50–59, 2023, doi: 10.14710/jil.22.1.50-59.

Kominfo7/RD2, “Dinkes Pekanbaru Imbau Waspadai Kasus DBD di Pekanbaru Pada Awal Tahun 2025,” Portal Resmi Pemerintah Kota Pekanbaru Provinsi Riau, 2025. https://www.pekanbaru.go.id/p/news/dinkes-pekanbaru-imbau-waspadai-kasus-dbd-di-pekanbaru-pada-awal-tahun-2025

N. Azizah and A. R. Masithoh, “Promosi Kesehatan 3M Plus Upaya Pencegahan Demam Berdarah Dan Peningkatan Perilaku Hidup Bersih Sehat,” J. Abdimas Indones., vol. 4, no. 1, pp. 30–33, 2022, doi: 10.26751/jai.v4i1.1500.

A. M. R. Arivadany, “Dampak Perubahan Iklim Terhadap Penyebaran Demam Berdarah : Tinjauan Literatur,” J. Kesehat. Tambusai, vol. 5, no. 3, pp. 7107–7119, 2024.

P. A. D. Amiri and S. Pierre, “An Ensemble-Based Machine Learning Model for Forecasting Network Traffic in VANET,” IEEE Access, vol. 11, no. March, pp. 22855–22870, 2023, doi: 10.1109/ACCESS.2023.3253625.

I. D. Mienye and Y. Sun, “A Survey of Ensemble Learning: Concepts, Algorithms, Applications, and Prospects,” IEEE Access, vol. 10, no. September, pp. 99129–99149, 2022, doi: 10.1109/ACCESS.2022.3207287.

M. S. Al Reshan et al., “Enhancing Breast Cancer Detection and Classification Using Advanced Multi-Model Features and Ensemble Machine Learning Techniques,” Life, vol. 13, no. 10, 2023, doi: 10.3390/life13102093.

S. Ghimire and S. Pangeni, “A mixed method evaluation of knowledge, attitude and practice on dengue fever among Lalitpur Metropolitan City residents: a cross-sectional investigation,” BMC Infect. Dis., vol. 24, no. 1, p. 1124, 2024, doi: 10.1186/s12879-024-10025-8.

N. A. M. H. Abdullah, N. C. Dom, S. A. Salleh, H. Salim, and N. Precha, “The association between dengue case and climate: A systematic review and meta-analysis.,” One Heal. (Amsterdam, Netherlands), vol. 15, p. 100452, Dec. 2022, doi: 10.1016/j.onehlt.2022.100452.

P. S. Singh and H. K. Chaturvedi, “Socio-ecological predictors of dengue in high incidence area of Delhi, India,” Sci. Rep., vol. 14, no. 1, p. 17005, 2024, doi: 10.1038/s41598-024-67909-7.

M. Hamed, “Knowledge, attitude, and practices toward dengue fever among the public: a cross-sectional study in the Western region of Saudi Arabia,” Front. Public Heal., vol. 12, 2024, doi: 10.3389/fpubh.2024.1327427.

M. N. Chaudhary et al., “Assessing the basic knowledge and awareness of dengue fever prevention among migrant workers in Klang Valley, Malaysia,” PLoS One, vol. 19, no. 2, p. e0297527, Feb. 2024, [Online]. Available: https://doi.org/10.1371/journal.pone.0297527

S. Fahri and S. Suhermanto, “Enhancing Student Knowledge in Dengue Hemorrhagic Fever Control Through Educational Modeling,” Heal. Educ. Heal. Promot., vol. 12, no. 1, 2024, doi: 10.58209/hehp.12.1.37.

F. Aslam et al., “Effect of Educational Interventions on Awareness of Dengue Fever and Its Preventive Measures among High School Students: Educational Interventions On Awareness of Dengue Fever,” NURSEARCHER (Journal Nurs. Midwifery Sci., vol. 4, no. 03 SE-Original Articles, pp. 24–29, Sep. 2024, doi: 10.54393/nrs.v4i03.114.

R. Dapari et al., “School-based health education effect on knowledge, attitude, and practices of dengue prevention among school children: a systematic review,” Discov. Soc. Sci. Heal., vol. 5, no. 1, p. 31, 2025, doi: 10.1007/s44155-025-00181-w.

López-Saleme Rossana, Escobar-Velásquez Katty, and Barajas-Lizarazo Mayra, “Knowledge, Attitudes, and Practices for the Prevention and Vector Control of Dengue in a Colombian Rural Population: A Mixed Method Study,” SAGE Open Nurs., vol. 11, p. 23779608241302710, Jan. 2025, doi: 10.1177/23779608241302713.

M. K. Jamal et al., “Investigating the recent outbreak of dengue fever in Iran: a systematic review,” Egypt. J. Intern. Med., vol. 37, no. 1, p. 37, 2025, doi: 10.1186/s43162-025-00411-2.

N. Tian et al., “Precision Prediction for Dengue Fever in Singapore: A Machine Learning Approach Incorporating Meteorological Data,” Trop. Med. Infect. Dis., vol. 9, no. 4, 2024, doi: 10.3390/tropicalmed9040072.

Z. J. Madewell et al., “Machine learning for predicting severe dengue in Puerto Rico,” Infect. Dis. Poverty, vol. 14, no. 1, p. 5, 2025, doi: 10.1186/s40249-025-01273-0.

I. P. Hawa, S. S. Prasetiyowati, and Y. Sibaroni, “Classification Prediction of Dengue Fever Spread Using Decision Tree with Time-Based Feature Expansion,” Int. J. Inf. Commun. Technol., vol. 10, no. 2, pp. 225–241, 2025, doi: 10.21108/ijoict.v10i2.1026.

M. Sugiono, “Analysis of Dengue Fever Spread Prediction Using Ensemble Learning Approach with Xgboost and Random,” Instal J. Komput., vol. 15, no. June, pp. 113–119, 2019, [Online]. Available: https://scholar.archive.org/work/te4arwrevjhhdg47urjqwoha2y/access/wayback/https://journalinstal.cattleyadf.org/index.php/Instal/article/download/8/8

D. A. Tuan and T. N. Dang, “Leveraging Climate Data for Dengue Forecasting in Ba Ria Vung Tau Province, Vietnam: An Advanced Machine Learning Approach.,” Trop. Med. Infect. Dis., vol. 9, no. 10, Oct. 2024, doi: 10.3390/tropicalmed9100250.

A. S. Khan et al., “Integrating BERT Embeddings with SVM for Prostate Cancer Prediction,” in 2024 6th International Conference on Electrical Engineering and Information & Communication Technology (ICEEICT), 2024, pp. 1–6. doi: 10.1109/ICEEICT62016.2024.10534547.

S. Samantaray and A. Sahoo, “Prediction of flow discharge in Mahanadi River Basin, India, based on novel hybrid SVM approaches,” Environ. Dev. Sustain., vol. 26, no. 7, pp. 18699–18723, 2024, doi: 10.1007/s10668-023-03412-9.

A. Lubis, Y. Irawan, Junadhi, and S. Defit, “Leveraging K-Nearest Neighbors with SMOTE and Boosting Techniques for Data Imbalance and Accuracy Improvement,” J. Appl. Data Sci., vol. 5, no. 4, pp. 1625–1638, 2024, doi: 10.47738/jads.v5i4.343.

K. Alnowaiser, “Improving Healthcare Prediction of Diabetic Patients Using KNN Imputed Features and Tri-Ensemble Model,” IEEE Access, vol. 12, pp. 16783–16793, 2024, doi: 10.1109/ACCESS.2024.3359760.

Z. Xu, C. Qiao, Y. Xiong, and J. Yu, “Enhancing Equipment Health Prediction with Enhanced SMOTE-KNN,” J. Ind. Eng. Appl. Sci., vol. 2, no. 2, pp. 13–20, 2024.

Herianto, B. Kurniawan, Z. H. Hartomi, Y. Irawan, and M. K. Anam, “Machine Learning Algorithm Optimization using Stacking Technique for Graduation Prediction,” J. Appl. Data Sci., vol. 5, no. 3, pp. 1272–1285, 2024.

H. Fonda, Y. Irawan, R. Melyanti, R. Wahyuni, and A. Muhaimin, “A Comprehensive Stacking Ensemble Approach for Stress Level Classification in Higher Education,” J. Appl. Data Sci., vol. 5, no. 4, pp. 1701–1714, 2024.

S. Rezaei Melal, M. Aminian, and S. M. Shekarian, “A machine learning method based on stacking heterogeneous ensemble learning for prediction of indoor humidity of greenhouse,” J. Agric. Food Res., vol. 16, p. 101107, 2024, doi: 10.1016/j.jafr.2024.101107.

A. Zaidi, “Predicting wildfires in Algerian forests using machine learning models,” Heliyon, vol. 9, no. 7, 2023, doi: 10.1016/j.heliyon.2023.e18064.

T. R. Mahesh, V. K. V, D. K. V, O. Geman, and M. Margala, “Healthcare Analytics The stratified K-folds cross-validation and class-balancing methods with high-performance ensemble classifiers for breast cancer classification,” Healthc. Anal., vol. 4, no. July, pp. 1–10, 2023.

W. Kaleem, S. Tewari, M. Fogat, and D. A. Martyushev, “A hybrid machine learning approach based study of production forecasting and factors influencing the multiphase flow through surface chokes,” Petroleum, no. xxxx, 2023, doi: 10.1016/j.petlm.2023.06.001.

M. Hasan, P. A. Bath, C. Marincowitz, L. Sutton, R. Pilbery, and F. Hopfgartner, “Pre-hospital prediction of adverse outcomes in patients with suspected COVID-19 : Development , application and comparison of machine learning and deep learning methods,” Comput. Biol. Med. J., vol. 151, no. May, 2022.

S. Hou, Y. Liu, and Q. Yang, “Journal of Rock Mechanics and Geotechnical Engineering Real-time prediction of rock mass classi fi cation based on TBM operation big data and stacking technique of ensemble learning,” vol. 14, 2022.

R. Liu et al., “Stacking Ensemble Method for Gestational Diabetes Mellitus Prediction in Chinese Pregnant Women : A Prospective Cohort Study,” vol. 2022, 2022, doi: 10.1155/2022/8948082.

H. Zheng, S. W. A. Sherazi, and J. Y. Lee, “A Stacking Ensemble Prediction Model for the Occurrences of Major Adverse Cardiovascular Events in Patients with Acute Coronary Syndrome on Imbalanced Data,” IEEE Access, vol. 9, pp. 113692–113704, 2021, doi: 10.1109/ACCESS.2021.3099795.

S. S. Kshatri, D. Singh, B. Narain, S. Bhatia, M. T. Quasim, and G. R. Sinha, “An Empirical Analysis of Machine Learning Algorithms for Crime Prediction Using Stacked Generalization: An Ensemble Approach,” IEEE Access, vol. 9, pp. 67488–67500, 2021, doi: 10.1109/ACCESS.2021.3075140.




DOI: https://doi.org/10.47738/jads.v6i4.898

Refbacks

  • There are currently no refbacks.



Barcode

Journal of Applied Data Sciences

ISSN:2723-6471 (Online)
Publisher:Bright Publisher
Website:http://bright-journal.org/JADS
Email:taqwa@amikompurwokerto.ac.id (principal contact)
  support@bright-journal.org (technical issues)

This work is licensed under a Creative Commons Attribution-ShareAlike 4.0