Advancing public health initiatives: a comprehensive analysis of infant mortality predictors in India using the robust XGBoost algorithm
摘要
Infant mortality is a significant public health challenge in developing nations, particularly in India, where effective predictive models are critical for targeted interventions. This study utilizes the XGBoost algorithm, a robust machine learning technique, to predict infant mortality using socio-demographic, maternal, and child health variables from the National Family Health Survey-4 (2015–2016). The dataset, derived from a stratified multistage sampling method, includes comprehensive information on key factors influencing infant mortality. Key predictors, including maternal age, birth weight, and access to healthcare, were identified through feature importance analysis. The XGBoost model was trained and evaluated using a split dataset, demonstrating exceptional performance metrics: 97.6% accuracy, 96.8% precision, 97.2% recall, 96.9% F1-score, and an AUC-ROC of 0.974. The findings reveal that the XGBoost model outperforms traditional approaches, effectively addressing challenges such as data sparsity and class imbalance. This study highlights the utility of advanced machine learning methods in predicting infant mortality, providing actionable insights for healthcare professionals and policymakers.