Hostname: page-component-76d6cb85b7-jhrpq Total loading time: 0 Render date: 2026-07-15T02:58:17.444Z Has data issue: false hasContentIssue false

Machine learning algorithms for predicting undernutrition among under-five children in Ethiopia

Published online by Cambridge University Press:  08 October 2021

Fikrewold H Bitew*
Affiliation:
Department of Demography, College for Health, Community and Policy, The University of Texas at San Antonio, 9947 Bricewood Hill, San Antonio, TX 78254, USA
Corey S Sparks
Affiliation:
Department of Demography, College for Health, Community and Policy, The University of Texas at San Antonio, 9947 Bricewood Hill, San Antonio, TX 78254, USA
Samuel H Nyarko
Affiliation:
Department of Demography, College for Health, Community and Policy, The University of Texas at San Antonio, 9947 Bricewood Hill, San Antonio, TX 78254, USA
*
*Corresponding author: Email fikre.wold@gmail.com
Rights & Permissions [Opens in a new window]

Abstract

Objective:

Child undernutrition is a global public health problem with serious implications. In this study, we estimate predictive algorithms for the determinants of childhood stunting by using various machine learning (ML) algorithms.

Design:

This study draws on data from the Ethiopian Demographic and Health Survey of 2016. Five ML algorithms including eXtreme gradient boosting, k-nearest neighbours (k-NN), random forest, neural network and the generalised linear models were considered to predict the socio-demographic risk factors for undernutrition in Ethiopia.

Setting:

Households in Ethiopia.

Participants:

A total of 9471 children below 5 years of age participated in this study.

Results:

The descriptive results show substantial regional variations in child stunting, wasting and underweight in Ethiopia. Also, among the five ML algorithms, xgbTree algorithm shows a better prediction ability than the generalised linear mixed algorithm. The best predicting algorithm (xgbTree) shows diverse important predictors of undernutrition across the three outcomes which include time to water source, anaemia history, child age greater than 30 months, small birth size and maternal underweight, among others.

Conclusions:

The xgbTree algorithm was a reasonably superior ML algorithm for predicting childhood undernutrition in Ethiopia compared to other ML algorithms considered in this study. The findings support improvement in access to water supply, food security and fertility regulation, among others, in the quest to considerably improve childhood nutrition in Ethiopia.

Information

Type
Research paper
Creative Commons
Creative Common License - CCCreative Common License - BY
This is an Open Access article, distributed under the terms of the Creative Commons Attribution licence (https://creativecommons.org/licenses/by/4.0/), which permits unrestricted re-use, distribution, and reproduction in any medium, provided the original work is properly cited.
Copyright
© The Author(s), 2021. Published by Cambridge University Press on behalf of The Nutrition Society
Figure 0

Fig. 1 Spatial variations in undernutrition indicators by administrative regions in Ethiopia, EDHS, 2016. Source: Created by the authors based on 2016 EDHS

Figure 1

Fig. 2 Spatial variations in severe undernutrition indicators by administrative region in Ethiopia, EDHS, 2016. Source: Created by the authors based on 2016 EDHS

Figure 2

Table 1 Performance indicators of all the five machine learning algorithms

Figure 3

Fig. 3 Stunting: comparison of sub-algorithms for stacking ensemble in R

Figure 4

Fig. 4 Wasting: comparison of sub-algorithms for stacking ensemble in R

Figure 5

Fig. 5 Underweight: comparison of sub-algorithms for stacking ensemble in R

Figure 6

Fig. 6 Top 20 most important variables from the xgbTree algorithm based on mean decrease Gini for stunting

Figure 7

Fig. 7 Top 20 most important variables from the xgbTree algorithm based on mean decrease Gini for wasting

Figure 8

Fig. 8 Top 20 most important variables from the xgbTree algorithm based on mean decrease Gini for underweight

Supplementary material: File

Bitew et al. supplementary material

Bitew et al. supplementary material

Download Bitew et al. supplementary material(File)
File 17.4 KB