Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS)

preprint OA: closed
📄 Open PDF Full text JSON View at publisher

Abstract

ABSTRACT Background Accurate prediction of the risk of anemia in under five children using ML can help reduce the burden of anemia in Zambia. This study applied ML models to predict the risk of anemia in under five children in Zambia. Methods This cross sectional study utilized data from the 2018 ZDHS. Feature selection was performed using the Boruta algorithm. Several ML models were trained on 80% of the data set. The best fit model was selected by comparing the model accuracy and area under the receiver operating characteristic curve (AUC -ROC). The 5-fold cross validation was used with feature importance performed using Shapley additive explanation (SHAP). Logistic regression was later performed in STATA. Python was used for ML modelling. Results Out of 7743 under five children, 58.5% had anemia. The top five most influential predictors of under-five anemia were current age of a child, maternal anemia, currently breastfeeding, region and stunting. The main ML model that was used to determine these factors was the XGB that gave an accuracy of 0.6432 and AUC-ROC of 0.6729. Two years of age or below OR=2.17, p<0.0001, currently breastfeeding OR=1.66, p<0.0001, stunting OR=1.27, p<0.0001, being a protestant OR=0.77, p=0.001, being a Muslim OR=0.33, p=0.019, living on the Copperbelt province OR=1.37, p=0.014), Luapula OR=1.91, p<0.0001), North western OR=1.66, p<0.0001, Southern OR=1.35, p=0.019, Western OR=1.47, p=0.005, having mild anemia OR=1.25, p=0.006), and moderate anemia OR=1.67, p<0.0001) significantly increased the odds of having anemia in under five children. Conclusion The extreme gradient boosting was the best performing ML model for predicting under-five mortality in Zambia and it showed that current age of a child, maternal anemia, currently breastfeeding, region and stunting are the top five most influential predictors. Therefore, this model could be useful in addressing anemia in Zambia as it can be used as a screening tool for early detection of children with anemia.
Full text 40,562 characters · extracted from preprint-html · click to expand
Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS) | medRxiv /* */ /* */ <!-- <!-- /*! * yepnope1.5.4 * (c) WTFPL, GPLv2 */ (function(a,b,c){function d(a){return"[object Function]"==o.call(a)}function e(a){return"string"==typeof a}function f(){}function g(a){return!a||"loaded"==a||"complete"==a||"uninitialized"==a}function h(){var a=p.shift();q=1,a?a.t?m(function(){("c"==a.t?B.injectCss:B.injectJs)(a.s,0,a.a,a.x,a.e,1)},0):(a(),h()):q=0}function i(a,c,d,e,f,i,j){function k(b){if(!o&&g(l.readyState)&&(u.r=o=1,!q&&h(),l.onload=l.onreadystatechange=null,b)){"img"!=a&&m(function(){t.removeChild(l)},50);for(var d in y[c])y[c].hasOwnProperty(d)&&y[c][d].onload()}}var j=j||B.errorTimeout,l=b.createElement(a),o=0,r=0,u={t:d,s:c,e:f,a:i,x:j};1===y[c]&&(r=1,y[c]=[]),"object"==a?l.data=c:(l.src=c,l.type=a),l.width=l.height="0",l.onerror=l.onload=l.onreadystatechange=function(){k.call(this,r)},p.splice(e,0,u),"img"!=a&&(r||2===y[c]?(t.insertBefore(l,s?null:n),m(k,j)):y[c].push(l))}function j(a,b,c,d,f){return q=0,b=b||"j",e(a)?i("c"==b?v:u,a,b,this.i++,c,d,f):(p.splice(this.i++,0,a),1==p.length&&h()),this}function k(){var a=B;return a.loader={load:j,i:0},a}var l=b.documentElement,m=a.setTimeout,n=b.getElementsByTagName("script")[0],o={}.toString,p=[],q=0,r="MozAppearance"in l.style,s=r&&!!b.createRange().compareNode,t=s?l:n.parentNode,l=a.opera&&"[object Opera]"==o.call(a.opera),l=!!b.attachEvent&&!l,u=r?"object":l?"script":"img",v=l?"script":u,w=Array.isArray||function(a){return"[object Array]"==o.call(a)},x=[],y={},z={timeout:function(a,b){return b.length&&(a.timeout=b[0]),a}},A,B;B=function(a){function b(a){var a=a.split("!"),b=x.length,c=a.pop(),d=a.length,c={url:c,origUrl:c,prefixes:a},e,f,g;for(f=0;f<d;f++)g=a[f].split("="),(e=z[g.shift()])&&(c=e(c,g));for(f=0;f<b;f++)c=x[f](c);return c}function g(a,e,f,g,h){var i=b(a),j=i.autoCallback;i.url.split(".").pop().split("?").shift(),i.bypass||(e&&(e=d(e)?e:e[a]||e[g]||e[a.split("/").pop().split("?")[0]]),i.instead?i.instead(a,e,f,g,h):(y[i.url]?i.noexec=!0:y[i.url]=1,f.load(i.url,i.forceCSS||!i.forceJS&&"css"==i.url.split(".").pop().split("?").shift()?"c":c,i.noexec,i.attrs,i.timeout),(d(e)||d(j))&&f.load(function(){k(),e&&e(i.origUrl,h,g),j&&j(i.origUrl,h,g),y[i.url]=2})))}function h(a,b){function c(a,c){if(a){if(e(a))c||(j=function(){var a=[].slice.call(arguments);k.apply(this,a),l()}),g(a,j,b,0,h);else if(Object(a)===a)for(n in m=function(){var b=0,c;for(c in a)a.hasOwnProperty(c)&&b++;return b}(),a)a.hasOwnProperty(n)&&(!c&&!--m&&(d(j)?j=function(){var a=[].slice.call(arguments);k.apply(this,a),l()}:j[n]=function(a){return function(){var b=[].slice.call(arguments);a&&a.apply(this,b),l()}}(k[n])),g(a[n],j,b,n,h))}else!c&&l()}var h=!!a.test,i=a.load||a.both,j=a.callback||f,k=j,l=a.complete||f,m,n;c(h?a.yep:a.nope,!!i),i&&c(i)}var i,j,l=this.yepnope.loader;if(e(a))g(a,0,l,0);else if(w(a))for(i=0;i (function(w,d,s,l,i){w[l]=w[l]||[];w[l].push({'gtm.start':new Date().getTime(),event:'gtm.js'});var f=d.getElementsByTagName(s)[0];var j=d.createElement(s);var dl=l!='dataLayer'?'&l='+l:'';j.src='//www.googletagmanager.com/gtm.js?id='+i+dl;j.type='text/javascript';j.async=true;f.parentNode.insertBefore(j,f);})(window,document,'script','dataLayer','GTM-P4HH5NV'); Skip to main content Home About Submit ALERTS / RSS Search for this keyword Advanced Search Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS) View ORCID Profile Teebeny Zulu , Leah Kamulaza , Atupele Chisiza , Nasilele Amatende , View ORCID Profile Nathan Nasson Tembo , Mutale Sampa , Hope Sabao , Wilbroad Mutale doi: https://doi.org/10.1101/2025.08.01.25332670 Teebeny Zulu 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia 2 The University of Zambia, School of Public Health, Department of Epidemiology and Biostatistics , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site ORCID record for Teebeny Zulu For correspondence: teebenyzulu{at}gmail.com Leah Kamulaza 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site Atupele Chisiza 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site Nasilele Amatende 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site Nathan Nasson Tembo 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia 2 The University of Zambia, School of Public Health, Department of Epidemiology and Biostatistics , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site ORCID record for Nathan Nasson Tembo Mutale Sampa 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia 2 The University of Zambia, School of Public Health, Department of Epidemiology and Biostatistics , Lusaka, Zambia 3 University of Witwatersrand, Division of Epidemiology and Biostatistics Find this author on Google Scholar Find this author on PubMed Search for this author on this site Hope Sabao 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia 4 The University of Zambia, School of Natural and Applied Sciences, Department of Mathematics, Statistics and Actuarial Science Find this author on Google Scholar Find this author on PubMed Search for this author on this site Wilbroad Mutale 1 Southern African Institute for Collaborative Research and Innovation Organization (SAICRIO) , Lusaka, Zambia 5 The university of Zambia, School of Public Health, Department of Health Policy and Systems , Lusaka, Zambia Find this author on Google Scholar Find this author on PubMed Search for this author on this site Abstract Full Text Info/History Metrics Data/Code Preview PDF ABSTRACT Background Accurate prediction of the risk of anemia in under five children using ML can help reduce the burden of anemia in Zambia. This study applied ML models to predict the risk of anemia in under five children in Zambia. Methods This cross sectional study utilized data from the 2018 ZDHS. Feature selection was performed using the Boruta algorithm. Several ML models were trained on 80% of the data set. The best fit model was selected by comparing the model accuracy and area under the receiver operating characteristic curve (AUC -ROC). The 5-fold cross validation was used with feature importance performed using Shapley additive explanation (SHAP). Logistic regression was later performed in STATA. Python was used for ML modelling. Results Out of 7743 under five children, 58.5% had anemia. The top five most influential predictors of under-five anemia were current age of a child, maternal anemia, currently breastfeeding, region and stunting. The main ML model that was used to determine these factors was the XGB that gave an accuracy of 0.6432 and AUC-ROC of 0.6729. Two years of age or below OR=2.17, p<0.0001, currently breastfeeding OR=1.66, p<0.0001, stunting OR=1.27, p<0.0001, being a protestant OR=0.77, p=0.001, being a Muslim OR=0.33, p=0.019, living on the Copperbelt province OR=1.37, p=0.014), Luapula OR=1.91, p<0.0001), North western OR=1.66, p<0.0001, Southern OR=1.35, p=0.019, Western OR=1.47, p=0.005, having mild anemia OR=1.25, p=0.006), and moderate anemia OR=1.67, p<0.0001) significantly increased the odds of having anemia in under five children. Conclusion The extreme gradient boosting was the best performing ML model for predicting under-five mortality in Zambia and it showed that current age of a child, maternal anemia, currently breastfeeding, region and stunting are the top five most influential predictors. Therefore, this model could be useful in addressing anemia in Zambia as it can be used as a screening tool for early detection of children with anemia. 1. INTRODUCTION Anemia is a significant global health concern, with children under five facing heightened vulnerability ( 1 ). Globally, an estimated 269 million children aged 6–59 months are affected by anemia ( 2 ). In Sub-Saharan Africa, the prevalence of anemia is alarming, with 67% of children affected ( 3 , 4 ). The repercussions of anemia in under-five children extend far beyond immediate health implications, impacting cognitive and behavioural development, physical growth, and, subsequently, affecting the socio-economic landscape of communities ( 2 , 4 ). While iron deficiency remains a predominant factor contributing to anemia in various regions, the causes of this condition are intricate and context-dependent ( 3 ). Demographic characteristics, socio-economic factors, as well as maternal health factors, are also associated with the risk of anemia among children ( 5 , 6 ). Previous studies have found iron-deficiency anemia to be the predominant type due to diets that are low in iron and other key micronutrients such as vitamin A and folate ( 7 – 10 ). However, other studies have found that exclusive( 11 ) breastfeeding beyond six months ( 11 ), younger children (6–23 months) ( 12 ), recent episodes of fever ( 13 , 14 ), maternal anemia ( 15 , 16 ), lower maternal education ( 17 ), living in poorer households ( 12 , 14 , 18 , 19 ) and living in households using unimproved water sources and sanitation facilities ( 20 , 21 ) are associated with an increased risk of anemia among under-five children. Identifying the most influential predictors of anemia in under-five children is key in developing targeted and cost-effective interventions. In addition, developing a screening tool for predicting under-five anemia based on maternal and child characteristics could supplement current diagnostic tools, especially in resource-limited settings. Machine learning models have been applied in medical research to make accurate predictions of several health conditions ( 6 , 22 – 24 ). These methods have been proven to make robust predictions compared to conventional statistical methods ( 25 ). Despite the wide application of ML models to predict the risk of anemia in under-five children in other settings, this modelling approach has not yet been implemented in LMICs like Zambia. Many studies have identified predictors of under-five anemia in Zambia using conventional statistical approaches, which are not able to identify rank the predictors in order of their influence on anemia ( 10 , 14 , 16 , 20 ).This study is aimed at identifying the best ML model for predicting anemia in under-five children and its risk factors so as to inform policy and ensure targeted and cost-effective interventions aimed at reducing the burden of anemia in Zambia. 2. Materials and Methods 2.1 Study Design This study used an analytical cross-sectional study design. A detailed conceptual framework of the ML modelling steps implemented in this study was adapted from a study by Mulenga and others ( 27 ) and is presented in Figure 1 Download figure Open in new tab Figure 1: Conceptual framework of the machine learning modelling steps 2.2 Study population and setting This study was conducted in Zambia which is estimated to have a population of about 21.7 million of which 14% of this population are children under the age of five ( 12 , 28 ). Data for this study was extracted from the children component ( kids recode ) in the 2018 ZDHS. The ZDHS is a national representative five-year periodic survey used to collect information from women aged 15–49 years and men aged 15–59 years about demographic and health status. Children aged 6-59 months from the selected household in the ZDHS sample were tested for blood hemoglobin level. In this study 7743 children aged 6-59 months with complete information of blood hemoglobin test results (anemia status) were included for analysis. Details of the DHS are published in ( 29 ). 2.3 Outcome variable In this study anemia status (“ anemic” versus “not anemic” ) was the outcome variable. The original variable ( hw57 ) in ZDHS was measured on an ordinal scale ( not anemic, mild, moderate and severe ). Children who had mild, moderate or severe anemia were coded 1 for “anemic” otherwise 0 for “not anemic” . 2.4 Predictor variables We selected 30 variables associated with the risk for childhood anemia based on previous studies ( 9 , 14 , 18 ) and these include; maternal age, maternal education, employment status, child’s breastfeeding status, sex of a child, current age of a child, disposal of child stools when not using a toilet, type of toilet facility, source of drinking water, household wealth status, size of a child at birth, child received vitamin A in the last 6 months, distance to seek medical care at a health facility, place of antenatal care visit, under five children who slept under the mosquito bed net, food, child had fever, type of place of residence, maternal anemia, wasting, stunting, religion, region, household type of cooking fuel used, number of children ever born, number of children under five years, maternal amenorrhea, gave a child juice, number of births in the last three years and underweight. 2.5 Handling of Missing Data To start with missing values in the outcome (anemia) were dropped and this reduced the sample size to 7743. Missing values in the features were imputed using multiple imputations by chained equations (MICE) in python ( 30 ). 2.6 Descriptive statistics and statistical software Frequencies and percentages were presented for descriptive statistics to measure the prevalence of anemia in under-five children. Associations between under-five anemia and categorical predictors were explored using the chi-squared test and the T-test was used for quantitative predictors. Data cleaning and exploratory analysis such as checking for missing values, feature engineering, univariate analysis and LR were all performed in STATA version 17. Feature selection was performed in R version 4.4.1. The Python version 3.12.8 was using for ML model building ( 31 ). 2.7 Validation Strategy This study used the 5-fold cross validation strategy where the dataset was split into the training and test sets in the ratio 80:20. This strategy has been shown to be sufficient in assessing the generalizability of ML models ( 35 ). The ML models used were optimized for performance using the GridSearchCV hyper parameter tunings ( 36 ). 2.8 Machine Learning Algorithms Seven machine learning algorithms were performed in this study and these included the LR, DT, RF, SVM, GB, XGB and NB. These algorithms are discussed in detail by ( 6 , 27 ). 2.9 Performance Evaluation Metrics The metrics used to evaluate the performance of models in this study were accuracy, precision, recall, F1-score and the AUC-ROC. The accuracy and AUC-ROC were prioritized. 2.10 Feature importance analysis Feature importance was performed using the (SHAP) values after performing the XGB to rank the predictors according to their influence on anemia prediction. Logistic regression was later performed to quantify the influence that each important predictor has on anemia. 2.11 Ethical Approval The ICF Institutional Review Board approved the 2018 ZDHS data survey protocols with the ICF Project Number: 132989.0.000.ZM. DHS.02. Permission to use the dataset was acquired from ICF Macro, and the dataset named ZMKR71DTA is available for download upon registration at https://www.dhsprogram.com/data . The user carefully adhered to the given guidelines, highlighting the sensitive nature of the information and the significance of not trying to uncover the identity of any household or individual surveyed for the study (maintaining anonymity). 3. RESULTS 3.1 Sample characteristics This study involved 7743 under five children in Zambia of whom 58.5% had anemia ( Table 1 ). Majority of children were below two years (58.2%), underweight (36.7%) and not breastfeeding (54.5%). Most of these children came from Eastern province (11.5%) with Western contributing the least (08.2%). The Mothers’ overall mean age and standard deviation was 30.0 (7.16) with most of them acquiring primary education 52.8%. Furthermore, 27.9% of these mothers had either mild, moderate or severe anemia ( Table 2 ). View this table: View inline View popup Download powerpoint Table 1: Socio-demographic characteristics of a child View this table: View inline View popup Table 2: socio-demographic characteristics of the mother 3.2 Feature Selection Thirty features known to be associated with under-five anemia from literature were included in a Boruta algorithm ( Figure 2 ) and 17 of these which included; ‘age of a child’, ‘place of antenatal care visit’, ‘disposal of child stools when not using the toilet’, gave a child juicy’, ‘food made from beans, peas, lentils and nuts’, births in the last three years’, ‘maternal anemia’, currently breastfeeding’, ‘region’, ‘stunting’, ‘underweight’, ‘residence’, ‘amenorrhea’, ‘type of fuel used for cooking’, religion’, maternal age’, having fever’ were classified as being important in predicting anemia. Of these, three features were removed as being redundant using the Cramer’s V Heatmap, leaving 14 features for model training. Download figure Open in new tab Figure 2: Boruta algorithm for important feature selection 3.3 Performance of Classification Models Seven machine learning models were fitted in this study and the XGB model outperformed all other models with an accuracy of 0.6432 and AUC-ROC metric of 0.6729 as shown in Table 3 and Figure 3 . Download figure Open in new tab Figure 3: ROC curves for all ML models View this table: View inline View popup Table 3: results from the seven ML models 3.4 Feature Importance Analysis Results from feature importance analysis are presented in figure 4 . The first part of the plot show that the current age of a child was the most influential predictor of anemia followed by maternal anemia, currently breastfeeding, region, stunting, child fever, place of antenatal care visit, underweight, maternal amenorrhea, religion, number of births in the last three years, type of place of residence, type of fuel used for cooking, and the least was the number of children ever born. The second part of the plot shows that older age, high levels of maternal anemia, prolonged breastfeeding, stunting, child fever, antenatal care visits, and underweight increase the risk of anemia in under-five children. Download figure Open in new tab Figure 4: SHAP values showing feature importance analysis 3.5 Adjusted and unadjusted odds ratios for important predictors of under-five anemia Results from multiple logistic regression have been grouped into child related factors ( Table 4 ) and maternal related factors ( Table 5 ). In this study, two years of age or below OR=2.17, (95% CI=1.6 3.05; p<0.0001), currently breastfeeding OR=1.66, (95% CI=1.44 1.92; p<0.0001), stunting OR=1.27, (95% CI=1.12 1.44; p<0.0001), being a protestant OR=0.77, (95% CI=0.65 0.90; p=0.001), being a Muslim OR=0.33, (95% CI=0.13 0.83; p=0.019), living on the Copperbelt province OR=1.37, (95% CI=1.07 1.77; p=0.014), Luapula OR=1.91, (95% CI=1.47 2.49; p<0.0001), North western OR=1.66, (95% CI=1.27 2.16; p<0.0001), Southern OR=1.35, (95% CI=1.05 1.72; p=0.019), Western OR=1.47, (95% CI=1.12 1.93; p=0.005), mild maternal anemia OR=1.25, (95% CI=1.07 1.47; p=0.006), and moderate maternal anemia OR=1.67, (95% CI=1.38 2.02; p<0.0001) significantly increased the odds of having anemia in under five children. The rest of the variables were not statically significant. View this table: View inline View popup Table 4: Child related important predictors of anemia View this table: View inline View popup Table 5: Maternal related important predictors of anemia 4. DISCUSSION 4.1 Machine learning model performance This study found XGB to be the best performing model with an accuracy of 0.6432 and a ROC-AUC of 0.6729. The performance of the XGB model in our study was slightly lower than the performance of a RF model in a study conducted in Ethiopia using the EDHS which had an ROC-AUC of 0.69 ( 22 ). Another study conducted in Bangladesh using the BDHS also found RF to be the best performing model with an ROC-AUC of 0.6857 ( 6 ). The variation in findings may be attributed to differences in data characteristics, such as sample size, population structure, anemia prevalence, and health system differences between countries. Additionally, models like XGB may perform better in datasets with complex feature interactions and less noise, as was potentially the case with the Zambian data ( 38 ). 4.2 Important predictors of anemia in under five children The current age of a child was the most influential predictor of anemia in under-five children. Under-five children who were below two years of age were more likely to have anemia compared to those above two years. This relationship is due to elevated iron requirements because of increased growth velocity during this period which depletes iron stores acquired during pregnancy, and without appropriate iron-rich complementary foods, breast milk alone cannot meet this demand thereby increasing their susceptibility to iron deficiency. These factors highlight the importance of early interventions through iron supplementation and proper complementary feeding to prevent anemia in young children ( 9 , 39 ). Maternal anemia was the second most influential predictor of anemia in this study. Logistic regression revealed that children born from mothers with mild or moderate anemia had an increased susceptibility to anemia. This relationship has been found in other studies ( 7 , 8 ). Maternal anemia limits the transfer of iron to the fetus, resulting in impaired fetal iron stores, which increases the likelihood of anemia after birth, especially if iron-rich complementary foods are not introduced in a timely manner ( 39 ). These findings underscore the importance of addressing maternal anemia during pregnancy to reduce the risk of anemia in children and highlight the need for early iron supplementation and appropriate feeding practices after birth. Furthermore, breastfeeding was the third most influential predictor of anemia. Children who were currently breastfeeding were more likely to have anemia compared to those who were not breastfeeding. As infants grow, their iron needs increase, and exclusive breastfeeding beyond 6 months without the introduction of complementary iron-rich foods may lead to iron deficiency anemia ( 11 ). These results highlight the importance of timely introduction of iron-rich complementary foods to prevent anemia in breastfeeding children. Region was ranked fourth in the top five most influential predictors of under-five anemia. Children who resided in Copperbelt, Eastern, Luapula, Northen and North western provinces of Zambia were at higher risk of anemia compared to those in Central province. This regional disparity could be explained by several contextual factors. Provinces such as Luapula, Northern, and North Western are characterized by higher poverty rates and food insecurity, which limit access to iron-rich and diverse diets essential for preventing anemia ( 12 ). The fifth most influential predictor of under-five anemia in Zambia was stunting. This study has revealed that stunted children were more likely to be anemic as compared to those who were not stunted. This finding can be explained by the fact that both stunting and anemia often share common underlying causes, such as chronic malnutrition and micronutrient deficiencies. Stunting is a marker of prolonged under-nutrition, which is typically associated with inadequate intake of essential nutrients, including iron, folate, and vitamin A, all of which play critical roles in hematopoiesis ( 2 ). This study brings to light the most influential predictors of under-five anemia in Zambia which is key in developing targeted interventions aimed at these predictors thereby reducing the burden of anemia in under-five children. Moreover, the identification of XGBoost as a best performing model in predicting anemia in under-five is the initial step in deploying this model to build a screening tool for under-five anemia from maternal and children characteristics. This tool could be used in resource limited settings like Zambia reducing the cost of anemia screening. 4.3 Strengths and limitations This study has a number of strengths. Firstly, the use of a nationally representative sample means that these findings could be applied to all under five children in Zambia. Secondly, the use of LR through conventional statistical modelling goes beyond ranking the predictors to quantifying and explaining the influence that each predictor has on under five anemia. Despite these strengths, the notable limitation of this study is the use of the 2018 ZDHS which was taken Seven ( 7 ) years ago and may not reflect the current trends and interactions among variables. We therefore recommend that this model should be updated with the latest information to improve its prediction accuracy. 5. CONCLUSION We compared seven machine learning prediction models for predicting under-five anemia. Among the models considered, the extreme gradient boosting performed the best with the highest classification accuracy and the area under the receiver operating curve. This study goes beyond identifying the best performing machine learning model and ranking predictors from the most influential to the least, to quantifying their influence with a logistic regression from the conventional statistical modelling approach. Further research on this topic should focus on deploying the machine learning model proposed in this study to build a screening tool for under-five anemia in Zambia. Data Availability Permission to use the dataset was acquired from ICF Macro, and the dataset named ZMKR71DTA is available for download upon registration at https://www.dhsprogram.com/data. https://www.dhsprogram.com/data . REFERENCES 1. ↵ Safiri S , Kolahi AA , Noori M , Nejadghaderi SA , Karamzad N , Bragazzi NL , et al. Burden of anemia and its underlying causes in 204 countries and territories, 1990-2019: results from the Global Burden of Disease Study 2019 . J Hematol OncolJ Hematol Oncol . 2021 Nov 4; 14 ( 1 ): 185 . OpenUrl 2. ↵ Bruno de Benoist , Erin McLean World Health Organization . Worldwide prevalence of anaemia 1993-2005 : WHO global database on anaemia . / Edited by Bruno de Benoist , Erin McLean , Ines Egli and Mary Cogswell . 2008 ; 40 . 3. ↵ Chaparro CM , Suchdev PS . Anemia epidemiology, pathophysiology, and etiology in low- and middle-income countries . Ann N Y Acad Sci . 2019 Aug; 1450 ( 1 ): 15 – 31 . OpenUrl CrossRef PubMed 4. ↵ Ngesa O , Mwambi H . Prevalence and risk factors of anaemia among children aged between 6 months and 14 years in Kenya . PloS One . 2014 ; 9 ( 11 ): e113756 . OpenUrl CrossRef PubMed 5. ↵ Ayoya MA , Ngnie-Teta I , Séraphin MN , Mamadoultaibou A , Boldon E , Saint-Fleur JE , et al. Prevalence and Risk Factors of Anemia among Children 6-59 Months Old in Haiti . Anemia . 2013 ; 2013 : 502968 . OpenUrl CrossRef PubMed 6. ↵ Khan JR , Awan N , Misu F . Determinants of anemia among 6-59 months aged children in Bangladesh: evidence from nationally representative data . BMC Pediatr . 2016 Jan 11; 16 : 3 . OpenUrl CrossRef PubMed 7. ↵ Bain LE , Awah PK , Geraldine N , Kindong NP , Sigal Y , Bernard N , et al. Malnutrition in Sub-Saharan Africa: burden, causes and prospects . Pan Afr Med J . 2013 ; 15 : 120 . OpenUrl PubMed 8. ↵ Kassebaum NJ , Jasrasaria R , Naghavi M , Wulf SK , Johns N , Lozano R , et al. A systematic analysis of global anemia burden from 1990 to 2010 . Blood . 2014 Jan 30; 123 ( 5 ): 615 – 24 . OpenUrl Abstract / FREE Full Text 9. ↵ Adu-Afarwuah S Gebreweld A , Ali N , Ali R , Fisha T. Prevalence of anemia and its associated factors among children under five years of age attending at Guguftu health center, South Wollo, Northeast Ethiopia . Adu-Afarwuah S , editor. PLOS ONE . 2019 Jul 5; 14 ( 7 ): e0218961 . OpenUrl CrossRef PubMed 10. ↵ Kobayashi E , Negi B , Nakazawa M . The association between food groups and childhood anemia in Zambia, based on the analysis of Zambia Demographic and Health Survey 2018 . J Public Health Afr . 2022 May 24; 13 ( 1 ): 2199 . OpenUrl PubMed 11. ↵ Aguayo VM. Complementary feeding practices for infants and young children in South Asia. A review of evidence for action post-2015 . Matern Child Nutr. 2017 ; 13 ( S2 ): e12439 . OpenUrl PubMed 12. ↵ Zambia Statistics Agency [Internet] . [cited 2025 Jun 18]. Available from: https://www.zamstats.gov.zm/ 13. ↵ Ewusie JE , Ahiadeke C , Beyene J , Hamid JS . Prevalence of anemia among under-5 children in the Ghanaian population: estimates from the Ghana demographic and health survey . BMC Public Health . 2014 Jun 19; 14 : 626 . OpenUrl PubMed 14. ↵ Nawa M , Konkola G , Sialubanje C . Factors Associated with Anaemia in Under-five Children in Zambia; Secondary Analysis of the 2021 Malaria Indicator Survey Data . Med J Zambia . 2024 Mar 30; 50 ( 4 ): 307 – 19 . OpenUrl 15. ↵ Okube OT , Mirie W , Odhiambo E , Sabina W , Habtu M . Prevalence and Factors Associated with Anaemia among Pregnant Women Attending Antenatal Clinic in the Second and Third Trimesters at Pumwani Maternity Hospital, Kenya . Open J Obstet Gynecol . 2016 ; 06 ( 01 ): 16 – 27 . OpenUrl 16. ↵ Rahman MM , Abe SK , Rahman MS , Kanda M , Narita S , Bilano V , et al. Maternal anemia and risk of adverse birth and health outcomes in low- and middle-income countries: systematic review and meta-analysis . Am J Clin Nutr . 2016 Feb; 103 ( 2 ): 495 – 504 . OpenUrl Abstract / FREE Full Text 17. ↵ Balarajan Y , Ramakrishnan U , Ozaltin E , Shankar AH , Subramanian SV . Anaemia in low-income and middle-income countries . Lancet Lond Engl . 2011 Dec 17; 378 ( 9809 ): 2123 – 35 . OpenUrl 18. ↵ Malako BG , Teshome MS , Belachew T . Anemia and associated factors among children aged 6–23 months in Damot Sore District, Wolaita Zone, South Ethiopia . BMC Hematol . 2018 Jul 3; 18 ( 1 ): 14 . OpenUrl PubMed 19. ↵ Assaf S , Juan C . Stunting and Anemia in Children from Urban Poor Environments in 28 Low and Middle-income Countries: A Meta-analysis of Demographic and Health Survey Data . Nutrients . 2020 Nov; 12 ( 11 ): 3539 . OpenUrl PubMed 20. ↵ Nkhoma P , Loti P , Sinkala M , Mantina H , Mwaba F , Kafita D , et al. Anaemia Types and Severity in Patients aged 1 to 14 years at the Children’s Hospital of the University Teaching Hospitals in Zambia [Internet] . medRxiv ; 2020 [cited 2025 Jun 18]. p. 2020.09.17.20194779 . Available from: https://www.medrxiv.org/content/10.1101/2020.09.17.20194779v1 21. ↵ Kwami CS , Godfrey S , Gavilan H , Lakhanpaul M , Parikh P. Water, Sanitation, and Hygiene: Linkages with Stunting in Rural Ethiopia . Int J Environ Res Public Health . 2019 Oct 9; 16 ( 20 ): 3793 . OpenUrl PubMed 22. ↵ Tesfaye SH , Seboka BT , Sisay D . Application of machine learning methods for predicting childhood anaemia: Analysis of Ethiopian Demographic Health Survey of 2016 . PloS One . 2024 ; 19 ( 4 ): e0300172 . OpenUrl PubMed 23. Kebede Kassaw A , Yimer A , Abey W , Molla TL , Zemariam AB. The application of machine learning approaches to determine the predictors of anemia among under five children in Ethiopia . Sci Rep . 2023 Dec 21; 13 ( 1 ): 22919 . OpenUrl PubMed 24. ↵ A Machine Learning Study to Classify the Type of Anemia in Children Under 5 Years of Age | SpringerLink [Internet] . [cited 2025 Jun 18]. Available from: https://link.springer.com/chapter/10.1007/978-3-031-31007-2_46 25. ↵ Rajula HSR , Verlato G , Manchia M , Antonucci N , Fanos V . Comparison of Conventional Statistical Methods with Machine Learning in Medicine: Diagnosis, Drug Development, and Treatment . Med Kaunas Lith . 2020 Sep 8; 56 ( 9 ): 455 . OpenUrl 26. Luo W , Phung D , Tran T , Gupta S , Rana S , Karmakar C , et al. Guidelines for Developing and Reporting Machine Learning Predictive Models in Biomedical Research: A Multidisciplinary View . J Med Internet Res . 2016 Dec 16; 18 ( 12 ): e323 . OpenUrl CrossRef PubMed 27. ↵ Mulenga C , Kaonga P , Hamoonga R , Mazaba ML , Chabala F , Musonda P . Predicting Mortality in Hospitalized COVID-19 Patients in Zambia: An Application of Machine Learning . Glob Health Epidemiol Genomics . 2023 ; 2023 : 8921220 . OpenUrl 28. ↵ Zambia Population ( 2025 ) - Worldometer [Internet] . [cited 2025 Jun 14]. Available from: https://www.worldometers.info/world-population/zambia-population 29. ↵ Corsi DJ , Neuman M , Finlay JE , Subramanian SV . Demographic and health surveys: a profile . Int J Epidemiol . 2012 Dec; 41 ( 6 ): 1602 – 13 . OpenUrl CrossRef PubMed Web of Science 30. ↵ Azur MJ , Stuart EA , Frangakis C , Leaf PJ . Multiple imputation by chained equations: what is it and how does it work? Int J Methods Psychiatr Res . 2011 Mar; 20 ( 1 ): 40 – 9 . OpenUrl CrossRef PubMed 31. ↵ Pedregosa F , Pedregosa F , Varoquaux G , Varoquaux G , Org N , Gramfort A , et al. Scikit-learn: Machine Learning in Python . Mach Learn PYTHON . 32. Chen T , Guestrin C. XGBoost: A Scalable Tree Boosting System . In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining [Internet] . 2016 [cited 2025 Jun 18]. p. 785 – 94 . Available from: http://arxiv.org/abs/1603.02754 33. Loizides F , Scmidt B Kluyver T , Ragan-Kelley B , Pérez F , Granger B , Bussonnier M , Frederic J , et al. Jupyter Notebooks – a publishing format for reproducible computational workflows . In: Loizides F , Scmidt B , editors. IOS Press ; 2016 [cited 2025 Jun 14]. p. 87 – 90 . Available from: https://eprints.soton.ac.uk/403913/ 34. Microsoft . Visual Studio Code Source Codes (2019 - 2022) [Internet] . 2023 [cited 2025 Jun 18]. Available from: http://archive.org/details/vscode-release-archive-2019-2022 35. ↵ Berrar D. Cross-Validation . In: Encyclopedia of Bioinformatics and Computational Biology [Internet] . Elsevier ; 2019 [cited 2025 Jun 15]. p. 542 – 5 . Available from: https://linkinghub.elsevier.com/retrieve/pii/B978012809633820349X 36. ↵ Bergstra J , Bengio Y . Random search for hyper-parameter optimization . J Mach Learn Res . 2012 Feb 1; 13 ( null ): 281 – 305 . OpenUrl 37. Iano Y , Saotome O , Kemper Vásquez GL , de Moraes Gomes Rosa MT , Arthur R , Gomes de Oliveira G Peña-Cáceres OJM , Chunga-Zapata EA , Basurco-Chambilla TR , Cayo-Cabrera GH , Villegas-Paz AD. A Machine Learning Study to Classify the Type of Anemia in Children Under 5 Years of Age . In: Iano Y , Saotome O , Kemper Vásquez GL , de Moraes Gomes Rosa MT , Arthur R , Gomes de Oliveira G , editors. Proceedings of the 8th Brazilian Technology Symposium (BTSym’22) . Cham : Springer International Publishing ; 2023 . p. 488 – 97 . 38. ↵ Chakraborty C , Joseph A. Machine Learning at Central Banks . SSRN Electron J [Internet] . 2017 [cited 2025 Jun 18]; Available from: https://www.ssrn.com/abstract=3031796 39. ↵ Kiserud T , Piaggio G , Carroli G , Widmer M , Carvalho J , Neerup Jensen L , et al. The World Health Organization Fetal Growth Charts: A Multinational Longitudinal Study of Ultrasound Biometric Measurements and Estimated Fetal Weight . PLoS Med . 2017 Jan; 14 ( 1 ): e1002220 . OpenUrl CrossRef PubMed 40. Lozoff B , Georgieff MK . Iron deficiency and brain development . Semin Pediatr Neurol . 2006 Sep; 13 ( 3 ): 158 – 65 . OpenUrl CrossRef PubMed 41. Atzatzev . The State of the World’s Children 2019: Children, food and nutrition [Internet] . UNICEF DATA . 2019 [cited 2025 Jun 18]. Available from: https://data.unicef.org/resources/state-of-the-worlds-children-2019/ 42. Black RE , Victora CG , Walker SP , Bhutta ZA , Christian P , de Onis M , et al. Maternal and child undernutrition and overweight in low-income and middle-income countries . Lancet Lond Engl . 2013 Aug 3; 382 ( 9890 ): 427 – 51 . OpenUrl 43. Weinberg ED. Iron loading and disease surveillance . Emerg Infect Dis. 1999 ; 5 ( 3 ): 346 – 52 OpenUrl CrossRef PubMed Web of Science View the discussion thread. Back to top Previous Next Posted August 03, 2025. Download PDF Data/Code Email Thank you for your interest in spreading the word about medRxiv. NOTE: Your email address is requested solely to identify you as the sender of this article. Your Email * Your Name * Send To * Enter multiple addresses on separate lines or separate them with commas. You are going to email the following Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS) Message Subject (Your Name) has forwarded a page to you from medRxiv Message Body (Your Name) thought you would like to see this page from the medRxiv website. Your Personal Message CAPTCHA This question is for testing whether or not you are a human visitor and to prevent automated spam submissions. Share Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS) Teebeny Zulu , Leah Kamulaza , Atupele Chisiza , Nasilele Amatende , Nathan Nasson Tembo , Mutale Sampa , Hope Sabao , Wilbroad Mutale medRxiv 2025.08.01.25332670; doi: https://doi.org/10.1101/2025.08.01.25332670 Share This Article: Copy Citation Tools Application of Machine Learning (ML) to Predict Under-Five Anemia using the 2018 Zambia Demographic and Health Survey (ZDHS) Teebeny Zulu , Leah Kamulaza , Atupele Chisiza , Nasilele Amatende , Nathan Nasson Tembo , Mutale Sampa , Hope Sabao , Wilbroad Mutale medRxiv 2025.08.01.25332670; doi: https://doi.org/10.1101/2025.08.01.25332670 Citation Manager Formats BibTeX Bookends EasyBib EndNote (tagged) EndNote 8 (xml) Medlars Mendeley Papers RefWorks Tagged Ref Manager RIS Zotero Tweet Widget Facebook Like Google Plus One Subject Area Public and Global Health Subject Areas All Articles Addiction Medicine (568) Allergy and Immunology (863) Anesthesia (299) Cardiovascular Medicine (4425) Dentistry and Oral Medicine (443) Dermatology (382) Emergency Medicine (607) Endocrinology (including Diabetes Mellitus and Metabolic Disease) (1507) Epidemiology (15221) Forensic Medicine (30) Gastroenterology (1123) Genetic and Genomic Medicine (6588) Geriatric Medicine (667) Health Economics (997) Health Informatics (4524) Health Policy (1368) Health Systems and Quality Improvement (1612) Hematology (540) HIV/AIDS (1264) Infectious Diseases (except HIV/AIDS) (15910) Intensive Care and Critical Care Medicine (1103) Medical Education (623) Medical Ethics (145) Nephrology (667) Neurology (6588) Nursing (346) Nutrition (998) Obstetrics and Gynecology (1143) Occupational and Environmental Health (956) Oncology (3331) Ophthalmology (970) Orthopedics (369) Otolaryngology (420) Pain Medicine (435) Palliative Medicine (129) Pathology (663) Pediatrics (1690) Pharmacology and Therapeutics (691) Primary Care Research (710) Psychiatry and Clinical Psychology (5440) Public and Global Health (9220) Radiology and Imaging (2195) Rehabilitation Medicine and Physical Therapy (1369) Respiratory Medicine (1196) Rheumatology (593) Sexual and Reproductive Health (710) Sports Medicine (529) Surgery (710) Toxicology (99) Transplantation (289) Urology (265) (function(){function c(){var b=a.contentDocument||a.contentWindow.document;if(b){var d=b.createElement('script');d.innerHTML="window.__CF$cv$params={r:'9ffd04730aa21b23',t:'MTc3OTQ2NTQ1NA=='};var a=document.createElement('script');a.src='/cdn-cgi/challenge-platform/scripts/jsd/main.js';document.getElementsByTagName('head')[0].appendChild(a);";b.getElementsByTagName('head')[0].appendChild(d)}}if(document.body){var a=document.createElement('iframe');a.height=1;a.width=1;a.style.position='absolute';a.style.top=0;a.style.left=0;a.style.border='none';a.style.visibility='hidden';document.body.appendChild(a);if('loading'!==document.readyState)c();else if(window.addEventListener)document.addEventListener('DOMContentLoaded',c);else{var e=document.onreadystatechange||function(){};document.onreadystatechange=function(b){e(b);'loading'!==document.readyState&&(document.onreadystatechange=e,c())}}}})();

Text is read by the "Ask this paper" AI Q&A widget below. Extraction quality varies by source — PMC NXML preserves structure cleanly, OA-HTML may include some navigation residue, and OA-PDF can have broken hyphenation. The publisher copy (via DOI) is the canonical version.

My notes (saved in your browser only)

Ask this paper AI returns verbatim quotes from the full text · source: preprint-html

Answers must be backed by verbatim quotes from this paper's full text. Hallucinated quotes are dropped automatically; if no verbatim passage answers the question, we say so. How this works

Citation neighborhood (no data yet)

We don't have any in-corpus citations linked to this paper yet. This is a recent paper (2025) — citers typically take a year or two to land, and the OpenAlex reference graph may still be filling in.

Source provenance

europepmc
last seen: 2026-05-20T01:45:00.602351+00:00