Machine learning based prediction of low birth weight and its associated risk factors: Insights from the Bangladesh Demographic and Health Survey 2022

Nourin Sultana, Zeba Afia, Isteaq Kabir Sifat, Shamsuz Zoha, Tajin Ahmed Jisa and Md Kaderi Kibria

PLOS Global Public Health, 2025, vol. 5, issue 9, 1-17

Abstract: Low birth weight (LBW) is a major public health concern particularly in low and middle-income countries as it contributes to increased infant mortality and long-term health complications. This study applies and evaluates machine learning (ML) algorithms to predict LBW and identify its key risk factors in Bangladesh. Data were collected from 3,192 complete records of ever-married women aged 15–49 years from the Bangladesh Demographic and Health Survey, 2022. Risk factors for LBW were identified by four feature selection techniques including Boruta-based selection (BFS), LASSO regression, Elastic Net and Random Forest (RF). Six ML algorithms, including Logistic Regression (LR), RF, Decision Tree (DT), Artificial Neural Networks (ANN), Extreme Gradient Boosting (XGB), and Light Gradient Boosting Machine (LGBM) were performed to predict LBW. Model performance was evaluated using accuracy, precision, recall, F1-score, AUC, and ROC analysis. SHAP values were utilized to examine the influence of individual features on the model’s prediction. The prevalence of LBW in Bangladesh was 27.8%. Twelve features were identified and the XGB model outperformed the other models by achieving the highest performance in predicting LBW with an accuracy of 80% and area under the curve of 0.761 in holdout (90:10) cross-validation. SHAP analysis revealed that ‘pregnancy duration’ and ‘division’ were the strongest predictors of LBW risk followed by ‘marriage to first birth interval’ ‘ANC visits’ ‘C-section’ and ‘place of delivery’. These findings demonstrate that XGB can serve as an effective tool for predicting LBW and identifying important risk factors that may guide targeted interventions. The insights generated from this study can support public health strategies aimed at reducing LBW prevalence in Bangladesh.

Date: 2025
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/globalpublichealth/artic ... journal.pgph.0005187 (text/html)
https://journals.plos.org/globalpublichealth/artic ... 05187&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pgph00:0005187

DOI: 10.1371/journal.pgph.0005187

Access Statistics for this article

More articles in PLOS Global Public Health from Public Library of Science
Bibliographic data for series maintained by globalpubhealth ().