Skip to main navigation menu Skip to main content Skip to site footer

Feature-Based Child Mortality Prediction Using Ensemble and Traditional Machine Learning Models

Abstract

Child mortality is a big problem around the world, especially in low- and middle-income nations where there are big differences in health care and social conditions. This investigation seeks to create a predictive model for child mortality and pinpoint the key factors that significantly contribute to it, employing machine learning (ML) methodologies. The dataset includes various features such as parental age, maternal education, birth weight, wealth index, and access to healthcare services. Thirteen machine learning classifiers were used, categorized into four model groups: Traditional Models (Logistic Regression, K-Nearest Neighbors, Support Vector Machine, Naive Bayes), Tree-Based Models (Decision Tree, Random Forest, Extra Trees), Boosting Models (AdaBoost, Gradient Boosting, XGBoost), and Ensemble Learning Models (Soft Voting, Hard Voting, Stacking). The efficacy of each model was assessed using classification metrics, including Accuracy, Precision, Recall, and F1-Score within a 10-fold cross-validation framework to guarantee robustness. Results indicate that ensemble models, particularly AdaBoost, achieved the highest predictive accuracy, with perfect scores across all metrics (1.00). XGBoost and Stacking also demonstrated strong and consistent performance. The findings indicate that ensemble learning methods are effective in predicting child mortality and can assist policymakers and healthcare planners in identifying high-risk populations and implementing targeted interventions to reduce child mortality.

Keywords

Child Mortality Prediction, Machine Learning Classifiers, Ensemble Learning Models, Explainable AI (XAI), Healthcare Data Analytics

PDF

References

  1. C. Di Novi and H. Thakare, “Inequality of Opportunity in Accessing Maternal and Newborn Healthcare Services: Evidence from the Bangladesh Demographic and Health Survey,” Social Indicators Research, vol. 164, no. 3, pp. 1505–1529, Sep. 2022, doi: 10.1007/s11205-022-03004-6.
  2. F. Yu, Z. Yan, R. Pu, S. Tang, B. Ghose, and R. Huang, “Do Mothers with Lower Socioeconomic Status Contribute to the Rate of All-Cause Child Mortality in Kazakhstan?,” BioMed Research International, vol. 2018, pp. 1–8, Jan. 2018, doi: 10.1155/2018/3629109.
  3. “End preventable deaths of newborns and children under 5 years of age,” Mar. 14, 2024. https://www.who.int/data/gho/data/themes/topics/sdg-target-3_2-newborn-and-child-mortality
  4. E. K. Ekwuazi, C. O. Chigbu, and N. C. Ngene, “Reducing maternal mortality in low- and middle-income countries,” Case Reports in Women S Health, vol. 39, p. e00542, Sep. 2023, doi: 10.1016/j.crwh.2023.e00542.
  5. P. Pandey, S. Shukla, N. K. Singh, and M. Kumar, “Predicting Child Mortality Determinants in Uttar Pradesh Using Machine Learning: Insights from the National Family and Health Survey (2019-21),” Clinical Epidemiology and Global Health, p. 101949, Feb. 2025, doi: 10.1016/j.cegh.2025.101949.
  6. F. Iqbal, M. I. Satti, A. Irshad, and M. A. Shah, “Predictive analytics in smart healthcare for child mortality prediction using a machine learning approach,” Open Life Sciences, vol. 18, no. 1, 2023, doi: 10.1515/biol-2022-0609.
  7. A. W. Demsash, “Using best performance machine learning algorithm to predict child death before celebrating their fifth birthday,” Informatics in Medicine Unlocked, vol. 40, p. 101298, 2023, doi: 10.1016/j.imu.2023.101298.
  8. C. Chivardi et al., “Understanding the social determinants of child mortality in Latin America over the last two decades: a machine learning approach,” Scientific Reports, vol. 13, no. 1, 2023, doi: 10.1038/s41598-023-47994-w.
  9. O. Samuel, T. Zewotir, and D. North, “Application of machine learning methods for predicting under-five mortality: analysis of Nigerian demographic health survey 2018 dataset,” BMC Medical Informatics and Decision Making, vol. 24, no. 1, 2024, doi: 10.1186/s12911-024-02476-5.
  10. L. P. Garcia, I. J. C. Schneider, C. De Oliveira, E. Traebert, and J. Traebert, “What is the impact of national public expenditure and its allocation on neonatal and child mortality? A machine learning analysis,” BMC Public Health, vol. 23, no. 1, 2023, doi: 10.1186/s12889-023-15683-y.
  11. S. Holcroft, I. Karangwa, F. Little, J. Behoor, and O. Bazirete, “Predictive modelling of postpartum haemorrhage using early risk factors: A comparative analysis of statistical and machine learning models,” International Journal of Environmental Research and Public Health, vol. 21, no. 5, p. 600, 2024, doi: 10.3390/ijerph21050600.
  12. R. K. Saroj, P. K. Yadav, R. Singh, and Obvious. N. Chilyabanyama, “Machine Learning Algorithms for understanding the determinants of under-five Mortality,” BioData Mining, vol. 15, no. 1, 2022, doi: 10.1186/s13040-022-00308-8.
  13. A. Satty, G. S. M. Khamis, Z. M. Mohammed, A. F. Mahmoud, F. A. Abdalla, M. Salih, ... & E. A. Gumma, “Statistical insights into machine learning models for predicting under-five mortality: An analysis from multiple indicator cluster survey (MICS),” IEEE Access, p. 1, 2025, doi: 10.1109/access.2025.3549097.
  14. S. Naznin, M. J. Uddin, I. Ahmad, and A. Kabir, “Analyzing and forecasting under-5 mortality trends in Bangladesh using machine learning techniques,” PLoS ONE, vol. 20, no. 2, p. e0317715, 2025, doi: 10.1371/journal.pone.0317715.
  15. S. Das, L., D. Brals, B. Boczek, S. T. Hasan, P. Massara, ... & W. Voskuijl, “Development of machine learning models predicting mortality using routinely collected observational health data from 0-59 months old children admitted to an intensive care unit in Bangladesh: critical role of biochemistry and haematology data,” BMJ Paediatrics Open, vol. 8, no. 1, p. e002365, 2024, doi: 10.1136/bmjpo-2023-002365.
  16. S. Adithya, T. S. Rezinald, A. V. A. Mary, J. Refonaa, S. L. J. Shabu, and P. Jeyanthi, “Child Mortality Prediction using Machine Learning,” 2022 6th International Conference on Trends in Electronics and Informatics (ICOEI), vol. 5, pp. 1253–1256, 2022, doi: 10.1109/icoei53556.2022.9776827.
  17. C. Ashwini, S. R. Bose, M. S. D. Padmavathy, C. Raj, and J. C. Ajay, “Predicting child mortality with diverse regression algorithms using a machine learning approach,” in Advances in computer and electrical engineering book series, 2024, pp. 329–352. doi: 10.4018/979-8-3693-3739-4.ch017.
  18. J. Prithula et al., “Improved pediatric ICU mortality prediction for respiratory diseases: machine learning and data subdivision insights,” Respiratory Research, vol. 25, no. 1, May 2024, doi: 10.1186/s12931-024-02753-x.
  19. M. H. Abdullah and Y. T. Mustafa, “Machine Learning and SHAP-Based analysis of deforestation and forest degradation dynamics along the Iraq–Turkey border,” Earth, vol. 6, no. 2, p. 49, Jun. 2025, doi: 10.3390/earth6020049.
  20. A. A. Abdullah, M. M. Hassan, and Y. T. Mustafa, “Leveraging Bayesian deep learning and ensemble methods for uncertainty quantification in image classification: A ranking-based approach,” Heliyon, vol. 10, no. 2, p. e24188, Jan. 2024, doi: 10.1016/j.heliyon.2024.e24188.
  21. A. A. Abdullah, M. M. Hassan and Y. T. Mustafa, "A Review on Bayesian Deep Learning in Healthcare: Applications and Challenges," in IEEE Access, vol. 10, pp. 36538-36562, 2022, doi: 10.1109/ACCESS.2022.3163384.
  22. M. S. Rao, S. K. UmaMaheswaran, N. C. Sattaru, K. H. Abdullah, U. K. Pandey and L. Biban, "A Critical Understanding of Integrated Artificial Intelligence Techniques for the Healthcare Prediction System," 2022 2nd International Conference on Advance Computing and Innovative Technologies in Engineering (ICACITE), Greater Noida, India, 2022, pp. 728-731. https://doi: 10.1109/ICACITE53722.2022.9823678.
  23. B. A. Sullivan et al., “Comparing machine learning techniques for neonatal mortality prediction: insights from a modeling competition,” Pediatric Research, Dec. 2024, doi: 10.1038/s41390-024-03773-5.
  24. J. Lee, J. Cai, F. Li, and Z. A. Vesoulis, “Predicting mortality risk for preterm infants using random forest,” Scientific Reports, vol. 11, no. 1, Mar. 2021, doi: 10.1038/s41598-021-86748-4.
  25. C. Mangold, S. Zoretic, K. Thallapureddy, A. Moreira, K. Chorath, and A. Moreira, “Machine Learning Models for Predicting Neonatal Mortality: A Systematic review,” Neonatology, vol. 118, no. 4, pp. 394–405, Jan. 2021, doi: 10.1159/000516891.
  26. K. . Pal, J. . K Pal, S. R. Trivedi, N. Sharma, P. . P Patavardhan, and S. Goswami, “Predictive Modeling of Neonatal Mortality Trends Using Machine Learning: A Cross-sectional Data Science Approach”, J Neonatal Surg, vol. 14, no. 22S, pp. 1033–1039, May 2025.
  27. A. Sheikhtaheri, M. R. Zarkesh, R. Moradi, and F. Kermani, “Prediction of neonatal deaths in NICUs: development and validation of machine learning models,” BMC Medical Informatics and Decision Making, vol. 21, no. 1, Apr. 2021, doi: 10.1186/s12911-021-01497-8.
  28. S. K. Dey et al., “Analyzing infant Cry to detect birth asphyxia using a hybrid CNN and feature extraction approach,” Neuroscience Informatics, p. 100193, Feb. 2025, doi: 10.1016/j.neuri.2025.100193.
  29. M. S. Rao, Ch. Sekhar, and D. Bhattacharyya, “Comparative analysis of machine learning models on loan risk analysis,” in Advances in intelligent systems and computing, 2021, pp. 81–90. https://doi: 10.1007/978-981-15-9516-5_7.
  30. A. F. M. Batista, C. S. G. Diniz, E. A. Bonilha, I. Kawachi, and A. D. P. C. Filho, “Neonatal mortality prediction with routinely collected data: a machine learning approach,” BMC Pediatrics, vol. 21, no. 1, Jul. 2021, doi: 10.1186/s12887-021-02788-9.
  31. E. Keles and U. Bagci, “The past, current, and future of neonatal intensive care units with artificial intelligence: a systematic review,” Npj Digital Medicine, vol. 6, no. 1, Nov. 2023, doi: 10.1038/s41746-023-00941-5.
  32. F. Juraev, S. El-Sappagh, E. Abdukhamidov, F. Ali, and T. Abuhmed, “Multilayer dynamic ensemble model for intensive care unit mortality prediction of neonate patients,” Journal of Biomedical Informatics, vol. 135, p. 104216, Oct. 2022, doi: 10.1016/j.jbi.2022.104216.
  33. T. W?odarczyk et al., “Machine Learning Methods for Preterm Birth Prediction: A review,” Electronics, vol. 10, no. 5, p. 586, Mar. 2021, doi: 10.3390/electronics10050586.
  34. A. Rahman and Md. H. Rahman, “Explore the factors related to the death of offspring under age five and appraise the hazard of child mortality using machine learning techniques in Bangladesh,” BMC Public Health, vol. 25, no. 1, Jan. 2025, doi: 10.1186/s12889-025-21460-w.
  35. N. Rai, N. Kaushik, D. Kumar, C. Raj, and A. Ali, “Mortality prediction of COVID-19 patients using soft voting classifier,” International Journal of Cognitive Computing in Engineering, vol. 3, pp. 172–179, Jun. 2022, doi: 10.1016/j.ijcce.2022.09.001.

Downloads

Download data is not yet available.

Similar Articles

1-10 of 65

You may also start an advanced similarity search for this article.