Predicting stunting status among under-5 children in Rwanda using neural network model: Evidence from 2020 Rwanda demographic and health survey.

Abstract

Background: Stunting is a serious public health concern in Rwanda, affecting around 33.3% of children under five in 2020. The researchers have employed machine learning algorithms to predict stunting in Rwanda; however, few studies used ANNs, despite their strong capacity to predict stunting. The purpose of this study was to predict stunting in Rwanda using ANNs and the most recent DHS data from 2020.

Methods: DHS 2020 dataset was used to train and test an ANN model for predicting stunting in children. The dataset, which included various child, parental, and socio-demographic characteristics, was split into 80% training data and 20% testing and validation data. The model utilised a multilayer perceptron (MLP). Model performance was assessed using accuracy, precision, recall, and AUC-ROC. Feature importances were determined and highlighted the most critical predictors of stunting.

Results: An overall accuracy of 72.0% on the test set was observed, with an AUC-ROC of 0.84, indicating the model's good performance. Factors appear to contribute to stunting among the negative value aspects. First and foremost, the mother's height is important, as a lower height suggests an increased risk of stunting in children. Positive value characteristics, on the other hand, emphasise elements that reduce the likelihood of stunting. The timing of the initiation of breastfeeding stands out as a crucial factor, showing that early breastfeeding initiation has been linked with a decreased risk of stunting.

Conclusions: These findings suggest that ANNs can be a useful tool for predicting stunting in Rwanda and identifying the most important associated factors for stunting. These insights can inform targeted interventions to reduce the burden of stunting in Rwanda and other low- and middle-income countries. Potential targeted interventions include nutritional support programs for pregnant and lactating mothers, and providing educational programs for parents on nutrition and hygiene.