Application of Ensemble Machine Learning for Infectious Diseases with Vaccine Intervention: A Global COVID-19 Case Study

Main Article Content

Egi Safitri
Ruki Rizalnul Fikri
Rini Nurlistiani

Abstract

The COVID-19 pandemic has posed significant challenges worldwide, especially in controlling the spread of the disease through vaccination and active case monitoring. This study aims to evaluate the effectiveness of various ensemble machine-learning models in predicting the number of daily vaccinations and the number of active cases of COVID-19 based on global data. The models used include Random Forest, Bagging, Gradient Boosting Machine (GBM), AdaBoost, and XGBoost. The evaluation results show that Random Forest provides the best performance in predicting both the number of daily vaccinations and active COVID-19 cases, with a MSE value of 4.7e+09, MAE of 16,971.1, and RMSE of 68,557.2 for daily vaccinations, as well as an R² Score of 0.989, indicating a high ability to explain data variability. The Bagging model also showed excellent results with MSE of 4.78e+09 and MAE of 17,039.8. In contrast, the AdaBoost model performed the worst in predicting both variables, with an MSE of 5.54e+10 and an MAE of 106,228.6. These findings suggest that Random Forest and Bagging are superior models for predicting the number of daily vaccinations and active COVID-19 cases. This study provides important insights into using machine learning to predict vaccination effectiveness and active case dynamics, aiding decision-making in global pandemic control efforts.

Downloads

Download data is not yet available.

Article Details

How to Cite
[1]
E. Safitri, R. Fikri, and R. Nurlistiani, “Application of Ensemble Machine Learning for Infectious Diseases with Vaccine Intervention: A Global COVID-19 Case Study”, INFOTEL, vol. 16, no. 4, pp. 819–836, Dec. 2024.
Section
Informatics

References

\bibitem{ref1}
R. Mousa and P. K. Ozili, “Reimagining financial inclusion in the post COVID-19 world: the case of Grameen America,” \textit{International Journal of Ethics and Systems}, vol. 39, no. 3, 2023, doi: 10.1108/IJOES-12-2021-0230.

\bibitem{ref2}
K. Yeoh \textit{et al.}, “Global Health System Resilience during Encounters with Stressors – Lessons Learnt from Cancer Services during the COVID-19 Pandemic,” \textit{Clin Oncol}, vol. 35, no. 4, 2023, doi: 10.1016/j.clon.2023.01.004.

\bibitem{ref3}
Y. Rosilawati, R. Bustami, M. Fajar, and M. I. Khatami, “University social responsibility during covid-19 outbreaks: application case from Indonesia and Malaysia,” \textit{Multidisciplinary Science Journal}, vol. 6, no. 3, 2024, doi: 10.31893/multiscience.2024035.

\bibitem{ref4}
T. Cheung \textit{et al.}, “Psychological impact on healthcare workers, general population and affected individuals of SARS and COVID-19: A systematic review and meta-analysis,” 2022. doi: 10.3389/fpubh.2022.1004558.

\bibitem{ref5}
N. Xiong, K. Fritzsche, Y. Pan, J. Löhlein, and R. Leonhart, “The psychological impact of COVID-19 on Chinese healthcare workers: a systematic review and meta-analysis,” 2022. doi: 10.1007/s00127-022-02264-4.

\bibitem{ref6}
N. Salari \textit{et al.}, “Prevalence of stress, anxiety, depression among the general population during the COVID-19 pandemic: A systematic review and meta-analysis,” 2020. doi: 10.1186/s12992-020-00589-w.

\bibitem{ref7}
U. Phalswal, V. Pujari, R. Sethi, and R. Verma, “Impact of social media on mental health of the general population during Covid-19 pandemic: A systematic review,” 2023. doi: 10.4103/jehp.jehp-460-22.

\bibitem{ref8}
S. Araki, “S-42-6: COVID-19 MORTALITY CORRELATES WITH CHRONIC DISEASE BURDEN AMONG COUNTRIES’ ONGOING SOCIAL DEVELOPMENT,” \textit{J Hypertens}, vol. 41, no. Suppl 1, 2023, doi: 10.1097/01.hjh.0000913640.08435.81.

\bibitem{ref9}
A. M. Wu \textit{et al.}, “Global, regional, and national burden of neck pain, 1990–2020, and projections to 2050: a systematic analysis of the Global Burden of Disease Study 2021,” \textit{Lancet Rheumatol}, vol. 6, no. 3, 2024, doi: 10.1016/S2665-9913(23)00321-1.

\bibitem{ref10}
C. E. Stauber \textit{et al.}, “Mobile Health Technologies Are Essential for Reimagining the Future of Water, Sanitation, and Hygiene,” \textit{Am J Trop Med Hyg}, vol. 106, no. 4, 2022, doi: 10.4269/ajtmh.21-1040.

\bibitem{ref11}
M. Raynaud \textit{et al.}, “Impact of the COVID-19 pandemic on publication dynamics and non-COVID-19 research production,” \textit{BMC Med Res Methodol}, vol. 21, no. 1, 2021, doi: 10.1186/s12874-021-01404-9.

\bibitem{ref12}
V. Recchia, A. Aloisi, and A. Zizza, “Risk management and communication plans from SARS to COVID-19 and beyond,” 2022. doi: 10.1002/hpm.3545.

\bibitem{ref13}
L. Cai \textit{et al.}, “The research progress of Chikungunya fever,” 2023. doi: 10.3389/fpubh.2022.1095549.

\bibitem{ref14}
D. Zhang \textit{et al.}, “Ecological Barrier Deterioration Driven by Human Activities Poses Fatal Threats to Public Health due to Emerging Infectious Diseases,” 2022. doi: 10.1016/j.eng.2020.11.002.

\bibitem{ref15}
L. Li, S. Zhang, J. Wang, X. Yang, and L. Wang, “Governing public health emergencies during the coronavirus disease outbreak: Lessons from four Chinese cities in the first wave,” \textit{Urban Studies}, vol. 60, no. 9, 2023, doi: 10.1177/00420980211049350.

\bibitem{ref16}
V. N. E. Malange \textit{et al.}, “The perinatal health challenges of emerging and re-emerging infectious diseases: A narrative review,” 2023. doi: 10.3389/fpubh.2022.1039779.

\bibitem{ref17}
Q. Wu, Q. Li, and J. Lu, “A One Health strategy for emerging infectious diseases based on the COVID-19 outbreak,” 2022. doi: 10.1016/j.jobb.2021.09.003.

\bibitem{ref18}
R. A. Karron \textit{et al.}, “Assessment of Clinical and Virological Characteristics of SARS-CoV-2 Infection among Children Aged 0 to 4 Years and Their Household Members,” \textit{JAMA Netw Open}, vol. 5, no. 8, 2022, doi: 10.1001/jamanetworkopen.2022.27348.

\bibitem{ref19}
N. Zheng \textit{et al.}, “A Novel Linear B-Cell Epitope on the P54 Protein of African Swine Fever Virus Identified Using Monoclonal Antibodies,” \textit{Viruses}, vol. 15, no. 4, 2023, doi: 10.3390/v15040867.

\bibitem{ref20}
J. Liu \textit{et al.}, “CD8 T cells contribute to vaccine protection against SARS-CoV-2 in macaques,” \textit{Sci Immunol}, vol. 7, no. 77, 2022, doi: 10.1126/sciimmunol.abq7647.

\bibitem{ref21}
A. F. Aqil, H.-C. Lee, and S. I. Wardani, “Forecasting COVID-19 Vaccination Trends in Indonesia using Machine Learning,” \textit{Indonesian Scholars Scientific Summit Taiwan Proceeding}, vol. 3, 2021, doi: 10.52162/3.2021118.

\bibitem{ref22}
H. Emami, R. Rabiei, S. Sohrabei, and A. Atashi, “Predicting the mortality of patients with Covid-19: A machine learning approach,” \textit{Health Sci Rep}, vol. 6, no. 4, 2023, doi: 10.1002/hsr2.1162.

\bibitem{ref23}
Y. Zoabi, S. Deri-Rozov, and N. Shomron, “Machine learning-based prediction of COVID-19 diagnosis based on symptoms,” \textit{NPJ Digit Med}, vol. 4, no. 1, 2021, doi: 10.1038/s41746-020-00372-6.

\bibitem{ref24}
H. Wang \textit{et al.}, “Neural-SEIR: A flexible data-driven framework for precise prediction of epidemic disease,” \textit{Mathematical Biosciences and Engineering}, vol. 20, no. 9, 2023, doi: 10.3934/mbe.2023749.

\bibitem{ref25}
A. Mahmoudian, N. Tajik, M. M. Taleshi, M. Shakiba, and M. Yekrangnia, “Ensemble machine learning-based approach with genetic algorithm optimization for predicting bond strength and failure mode in concrete-GFRP mat anchorage interface,” \textit{Structures}, vol. 57, 2023, doi: 10.1016/j.istruc.2023.105173.

\bibitem{ref26}
H. Yuan, J. Liu, Y. Zhou, and H. Pei, “State of Charge Estimation of Lithium Battery Based on Integrated Kalman Filter Framework and Machine Learning Algorithm,” \textit{Energies (Basel)}, vol. 16, no. 5, 2023, doi: 10.3390/en16052155.

\bibitem{ref27}
H.-W. Zhang, D.-L. Huang, Y.-R. Wang, H.-S. Zhong, and H.-W. Pang, “CT radiomics based on different machine learning models for classifying gross tumor volume and normal liver tissue in hepatocellular carcinoma,” \textit{Cancer Imaging}, vol. 24, no. 1, 2024, doi: 10.1186/s40644-024-00652-4.

\bibitem{ref28}
R. Z. Pek, S. T. Ozyer, T. Elhage, T. Ozyer, and R. Alhajj, “The Role of Machine Learning in Identifying Students At-Risk and Minimizing Failure,” \textit{IEEE Access}, vol. 11, 2023, doi: 10.1109/ACCESS.2022.3232984.

\bibitem{ref29}
WHO, “COVID-19 cases WHO COVID-19 dashboard,” 2024.

\bibitem{ref30}
A. N. Khan \textit{et al.}, “A New Method for Determination of Optimal Borehole Drilling Location Considering Drilling Cost Minimization and Sustainable Groundwater Management,” \textit{ACS Omega}, vol. 8, no. 12, 2023, doi: 10.1021/acsomega.2c06854.

\bibitem{ref31}
S. J. Lee \textit{et al.}, “Random RotBoost: An Ensemble Classification Method Based on Rotation Forest and AdaBoost in Random Subsets and Its Application to Clinical Decision Support,” \textit{Entropy}, vol. 24, no. 5, 2022, doi: 10.3390/e24050617.

\bibitem{ref32}
Z. Liang, “Predict Customer Churn based on Machine Learning Algorithms,” \textit{Highlights in Business, Economics and Management}, vol. 10, 2023, doi: 10.54097/hbem.v10i.8051.

\bibitem{ref33}
Y. Baashar \textit{et al.}, “Effectiveness of Artificial Intelligence Models for Cardiovascular Disease Prediction: Network Meta-Analysis,” \textit{Comput Intell Neurosci}, vol. 2022, 2022, doi: 10.1155/2022/5849995.

\bibitem{ref34}
K. Matuszelański and K. Kopczewska, “Customer Churn in Retail E-Commerce Business: Spatial and Machine Learning Approach,” \textit{Journal of Theoretical and Applied Electronic Commerce Research}, vol. 17, no. 1, 2022, doi: 10.3390/jtaer17010009.

\bibitem{ref35}
X. Li and K. Li, “Imbalanced data classification based on improved EIWAPSO-AdaBoost-C ensemble algorithm,” \textit{Applied Intelligence}, vol. 52, no. 6, 2022, doi: 10.1007/s10489-021-02708-5.

\bibitem{ref36}
S. B. Oyong, U. O. Ekong, and O. U. Obot, “Dynamic analysis of malware intrusion in mobile devices using Adaboost Algorithm, KNN and SVM base classifiers,” \textit{World Journal of Applied Science \& Technology}, vol. 15, no. 1, 2023, doi: 10.4314/wojast.v15i1.78.

\bibitem{ref37}
A. K. Dey, G. P. Gupta, and S. P. Sahu, “A metaheuristic-based ensemble feature selection framework for cyber threat detection in IoT-enabled networks,” \textit{Decision Analytics Journal}, vol. 7, 2023, doi: 10.1016/j.dajour.2023.100206.

\bibitem{ref38}
E. Kesriklioğlu, E. Oktay, and A. Karaaslan, “Predicting total household energy expenditures using ensemble learning methods,” \textit{Energy}, vol. 276, 2023, doi: 10.1016/j.energy.2023.127581.

\bibitem{ref39}
K. Kim \textit{et al.}, “Pre-diagnosis of flooding and drying in proton exchange membrane fuel cells by bagging ensemble deep learning models using long short-term memory and convolutional neural networks,” \textit{Energy}, vol. 266, 2023, doi: 10.1016/j.energy.2022.126441.

\bibitem{ref40}
K. V. Archana and G. Komarasamy, “A novel deep learning-based brain tumor detection using the Bagging ensemble with K-nearest neighbor,” \textit{Journal of Intelligent Systems}, vol. 32, no. 1, 2023, doi: 10.1515/jisys-2022-0206.

\bibitem{ref41}
N. Kati and F. Ucar, “An Intelligent Model for Supercapacitors with a Graphene-Based Fractal Electrode to Investigate the Cyclic Voltammetry,” \textit{Fractal and Fractional}, vol. 7, no. 3, 2023, doi: 10.3390/fractalfract7030218.

\bibitem{ref42}
J. Y. Zhu \textit{et al.}, “Ultrasound-based radiomics analysis for differentiating benign and malignant breast lesions: From static images to CEUS video analysis,” \textit{Front Oncol}, vol. 12, 2022, doi: 10.3389/fonc.2022.951973.

\bibitem{ref43}
S. Mohapatra, R. Mukherjee, A. Roy, A. Sengupta, and A. Puniyani, “Can Ensemble Machine Learning Methods Predict Stock Returns for Indian Banks Using Technical Indicators?,” \textit{Journal of Risk and Financial Management}, vol. 15, no. 8, 2022, doi: 10.3390/jrfm15080350.

\bibitem{ref44}
M. Das and E. K. Akpinar, “Investigation of pear drying performance by different methods and regression of convective heat transfer coefficient with support vector machine,” \textit{Applied Sciences (Switzerland)}, vol. 8, no. 2, 2018, doi: 10.3390/app8020215.