Related Experiment Video
Updated: May 10, 2025
Author Spotlight: Developing a Point-of-Care Hemoglobin Estimation Method for Anemia Management
Published on: January 19, 2024
Optimizing machine learning models for predicting anemia among under-five children in Ethiopia: insights from
Ali Yimer1, Hassen Ahmed Yesuf2, Sada Ahmed3
1Department of Public Health, College of Health Sciences, Woldia University, Po. Box: 400, Woldia, Ethiopia. ali24yimer@gmail.com.
Insights
The random forest machine learning model accurately predicts anemia in Ethiopian children under five. This AI approach offers a promising tool for early disease detection and improved child health outcomes.
Area of Science:
- Pediatric Health
- Medical Informatics
- Machine Learning in Healthcare
Background:
- Accurate disease prediction is crucial for pediatric healthcare, especially for conditions like anemia in young children.
- Delayed diagnosis of childhood anemia can lead to severe complications and increased mortality.
- Machine learning (ML) offers a powerful approach to analyze health data and generate insights for disease prediction.
Purpose of the Study:
- To determine the most effective machine learning algorithm for predicting anemia in Ethiopian children under five.
- To compare the performance of various ML models in identifying anemia risk factors.
Main Methods:
- Utilized data from the 2016 Ethiopian Demographic and Health Survey.
- Evaluated six ML models: logistic regression, random forest, decision tree, support vector machine, Naïve Bayes, and K-nearest neighbors.
- Assessed model performance using receiver operating characteristic curves and accuracy metrics.
Main Results:
- The random forest model achieved the highest prediction accuracy at 81.16%.
- Other models showed varying accuracies: Decision Tree (68.40%), SVM (59.94%), Naïve Bayes (53.06%), KNN (69.96%), and Logistic Regression (54.79%).
Conclusions:
- The random forest algorithm is the most effective model for predicting anemia in children under five.
- The random forest model demonstrated strong performance with 83.07% sensitivity and 79.26% specificity.
Background:
Healthcare practitioners require a robust predictive system to accurately diagnose diseases, especially in young children with conditions such as anemia. Delays in diagnosis and treatment can have severe consequences, potentially leading to serious complications and childhood mortality. By leveraging machine learning methods with extensive datasets, valuable and scientifically sound insights can be generated to address pressing health and healthcare-related challenges.
Objectives:
The primary objective of this study was to identify the most effective machine-learning algorithm for predicting anemia among under five children in Ethiopia.
Methods:
The data utilized in this study were sourced from the 2016 Ethiopian Demographic and Health Survey. Six machine-learning models, comprising a classic logistic regression model along with random forest, decision tree, support vector machine, Naïve Bayes, and K-nearest neighbors, were employed to predict factors influencing anemia in children under five. The predictive capacities of each machine-learning model were evaluated using receiver operating characteristic curves and various measures of model accuracy.
Results:
The random forest model demonstrated the highest accuracy among the algorithms tested, achieving an overall accuracy of 81.16%. The accuracy rates for the decision tree, support vector machines, Naïve Bayes, K-nearest neighbors, and classical logistic regression models were 68.40%, 59.94%, 53.06%, 69.96%, and 54.79%, respectively.
Conclusion:
In general, the random forest algorithm emerged as the preferred model for predicting anemia in children under five. The model exhibited a specificity of 79.26%, sensitivity of 83.07%, positive predictive value of 80.02%, negative predictive value of 82.40%, and an area under the curve of 81.80%.
