Author:
Mfateneza Emmanuel,Rutayisire Pierre Claver,Biracyaza Emmanuel,Musafiri Sanctus,Mpabuka Willy Gasafari
Abstract
Abstract
Background
Extensive research on infant mortality (IM) exists in developing countries; however, most of the methods applied thus far relied on conventional regression analyses with limited prediction capability. Advanced of Machine Learning (AML) methods provide accurate prediction of IM; however, there is no study conducted using ML methods in Rwanda. This study, therefore, applied Machine Learning Methods for predicting infant mortality in Rwanda.
Methods
A cross-sectional study design was conducted using the 2014–15 Rwanda Demographic and Health Survey. Python software version 3.8 was employed to test and apply ML methods through Random Forest (RF), Decision Tree, Support Vector Machine and Logistic regression. STATA version 13 was used for analysing conventional methods. Evaluation metrics methods specifically confusion matrix, accuracy, precision, recall, F1 score, and Area under the Receiver Operating Characteristics (AUROC) were used to evaluate the performance of predictive models.
Results
Ability of prediction was between 68.6% and 61.5% for AML. We preferred with the RF model (61.5%) presenting the best performance. The RF model was the best predictive model of IM with accuracy (84.3%), recall (91.3%), precision (80.3%), F1 score (85.5%), and AUROC (84.2%); followed by decision tree model with model accuracy (83%), recall (91%), precision (79%), F1 score (84.67%) and AUROC(82.9%), followed by support vector machine with model accuracy (68.6%), recall (74.9%), precision(67%), F1 score (70.73%) and AUROC (68.6%) and last was a logistic regression with the low accuracy of prediction (61.5%), recall (61.1%), precision (62.2%), F1 score (61.6%) and AUROC (61.5%) compared to other predictive models. Our predictive models showed that marital status, children ever born, birth order and wealth index are the 4 top predictors of IM.
Conclusions
In developing a predictive model, ML methods are used to classify certain hidden information that could not be detected by traditional statistical methods. Random Forest was classified as the best classifier to be used for the predictive models of IM.
Publisher
Springer Science and Business Media LLC
Subject
Obstetrics and Gynecology
Reference47 articles.
1. Reidpath D, Allotey P. Infant mortality rate as an indicator of population health. J Epidemiol Community Heal [Internet]. 2003;57(5):344–346. https://doi.org/10.1136/jech.57.5.344
2. Blencowe H, Krasevec J, de Onis M, Black RE, An X, Stevens GA, et al. National, regional, and worldwide estimates of low birthweight in 2015, with trends from 2000: a systematic analysis. Lancet Glob Heal [Internet]. 2019;7(7):e849–60. https://doi.org/10.1016/S2214-109X(18)30565-5
3. UNICEF. UNICEF Annual Report 2015 [Internet]. New York, New York, United States; 2016. https://www.unicef.org/reports/unicef-annual-report-2015
4. Shukla V V., Eggleston B, Ambalavanan N, McClure EM, Mwenechanya M, Chomba E, et al. Predictive Modeling for Perinatal Mortality in Resource-Limited Settings. JAMA Netw Open [Internet]. 2020;3(11):1–13. https://doi.org/10.1001/jamanetworkopen.2020.26750
5. Ramakrishnan R, Rao S, He JR. Perinatal health predictors using artificial intelligence: A review. Women’s Heal. 2021;17:17455065211046132.
Cited by
15 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献