Improved prediction of childhood anemia using hybrid ensemble learning and dual-level explainability

Ashima Kukkar1, Julius Moinget Loibor2, Mehmet Akif Cifci3,4

  • 1Institute of Engineering and Technology, Chitkara University, Punjab, India.

Insights

Early childhood anemia prediction is crucial for public health. An interpretable machine learning framework accurately detects anemia in children, aiding clinical decisions in underserved regions.

Area of Science:

  • Machine Learning in Public Health
  • Computational Epidemiology
  • Pediatric Health Informatics

Background:

  • Millions of children under five, especially in low- and middle-income countries, suffer from preventable anemia.
  • Early anemia detection is critical for improving public health outcomes and preventing long-term developmental issues.

Purpose of the Study:

  • To propose an interpretable machine learning framework for the early prediction of childhood anemia.
  • To utilize structured healthcare data for anemia risk assessment in pediatric populations.

Main Methods:

  • A stacked ensemble architecture integrating TabNet, XGBoost, and Multi-Layer Perceptron (MLP) with logistic regression as a meta-learner.
  • Hyperparameter optimization using GridSearchCV and Optuna, with feature importance analysis via SHAP and individual prediction explanations using LIME.
  • Training and testing on the Tanzania Demographic and Health Survey (DHS) dataset, with external validation using NFHS (India) data.

Main Results:

  • The ensemble model achieved 98.5% accuracy, outperforming individual models with high precision, recall, and F1-scores.
  • GridSearchCV and Optuna demonstrated optimal performance for hyperparameter tuning. Key predictors identified include child age, wealth index, and breastfeeding status.
  • External validation confirmed cross-regional generalizability, showing close agreement with observed anemia prevalence trends.

Conclusions:

  • The developed framework is accurate, transparent, and adaptable for early anemia detection.
  • It can significantly assist clinical decision-making, particularly in resource-limited settings.
  • The tool offers valuable support for global public health initiatives aimed at combating childhood anemia.
Abstract

Related Concept Videos