Machine learning algorithms for predicting low birth weight in Ethiopia

Wondesen Teshome Bekele1

  • 1Department of Statistics, College of Natural and Computational Sciences, Dire Dawa University, Dire Dawa, Ethiopia. wondesen52@gmail.com.

Insights

Predicting low birth weight (LBW) is crucial for infant survival. The Random Forest model accurately identified LBW in Ethiopia, highlighting key predictors like maternal age and child

Area of Science:

  • Public Health
  • Medical Informatics
  • Machine Learning

Background:

  • Low birth weight (LBW) significantly increases infant mortality risk, affecting 14.6% of newborns globally.
  • Prevalence of LBW varies regionally, with Africa experiencing higher rates (13.7%) compared to developed regions (7.2%).
  • Ethiopia faces a substantial LBW burden, contributing significantly to Africa's statistics, with associated long-term health implications for affected infants.

Purpose of the Study:

  • To implement and compare various machine learning models for predicting low birth weight (LBW) in Ethiopia.
  • To identify the most effective classifier for accurate LBW prediction using the Ethiopia Demographic and Health Survey 2016 data.

Main Methods:

  • Utilized data from the Ethiopia Demographic and Health Survey 2016 for model development.
  • Compared eight machine learning classifiers: Logistic Regression, Decision Tree, Naive Bayes, K-Nearest Neighbor, Random Forest (RF), Support Vector Machine, Gradient Boosting, and Extreme Gradient Boosting.
  • Performed data preprocessing, including data cleaning, with a binary target category for Normal and LBW infants.

Main Results:

  • The Random Forest (RF) model demonstrated superior performance in predicting LBW.
  • RF achieved 91.60% accuracy, 91.60% Recall, 96.80% ROC-AUC, and 91.60% F1 Score.
  • Key predictors for LBW in Ethiopia included child's gender, birth interval, mother's occupation, and mother's age.

Conclusions:

  • The Random Forest model is highly effective for predicting low birth weight in the Ethiopian context.
  • Accurate LBW prediction can serve as a vital preventive measure and indicator of infant health risks.
  • Identifying critical predictors like maternal and demographic factors can inform targeted interventions to reduce LBW rates.
Abstract