Abstract:BACKGROUND: Low birthweight (LBW) is a leading cause of neonatal mortality in the United States and a major causative factor of adverse health effects in newborns. Identifying high-risk patients early in prenatal care is crucial to preventing adverse outcomes. Previous studies have proposed various machine learning (ML) models for LBW prediction task, but they were limited by small and imbalanced data sets. Some authors attempted to address this through different data rebalancing methods. However, most of their reported performances did not reflect the models' actual performance in real-life scenarios. To date, few studies have successfully benchmarked the performance of ML models in maternal health; thus, it is critical to establish benchmarks to advance ML use to subsequently improve birth outcomes.OBJECTIVE: This study aimed to establish several key benchmarking ML models to predict LBW and systematically apply different rebalancing optimization methods to a large-scale and extremely imbalanced all-payer hospital record data set that connects mother and baby data at a state level in the United States. We also performed feature importance analysis to identify the most contributing features in the LBW classification task, which can aid in targeted intervention.METHODS: Our large data set consisted of 266,687 birth records across 6 years, and 8.63% (n=23,019) of records were labeled as LBW. To set up benchmarking ML models to predict LBW, we applied 7 classic ML models (ie, logistic regression, naive Bayes, random forest, extreme gradient boosting, adaptive boosting, multilayer perceptron, and sequential artificial neural network) while using 4 different data rebalancing methods: random undersampling, random oversampling, synthetic minority oversampling technique, and weight rebalancing. Owing to ethical considerations, in addition to ML evaluation metrics, we primarily used recall to evaluate model performance, indicating the number of correctly predicted LBW cases out of all actual LBW cases, as false negative health care outcomes could be fatal. We further analyzed feature importance to explore the degree to which each feature contributed to ML model prediction among our best-performing models.RESULTS: We found that extreme gradient boosting achieved the highest recall score-0.70-using the weight rebalancing method. Our results showed that various data rebalancing methods improved the prediction performance of the LBW group substantially. From the feature importance analysis, maternal race, age, payment source, sum of predelivery emergency department and inpatient hospitalizations, predelivery disease profile, and different social vulnerability index components were important risk factors associated with LBW.CONCLUSIONS: Our findings establish useful ML benchmarks to improve birth outcomes in the maternal health domain. They are informative to identify the minority class (ie, LBW) based on an extremely imbalanced data set, which may guide the development of personalized LBW early prevention, clinical interventions, and statewide maternal and infant health policy changes.

Exploring parental factors influencing low birth weight on the 2022 CDC natality dataset

Analysis of trends of birth weight, and incidence of adverse birth outcomes and relevant factors from 2002 to 2015

Analysis of affect factors of low birth weight

Weighting of risk factors for low birth weight: a linked routine data cohort study in Wales, UK

Factors Associated with Low Birthweight in Low-and-Middle Income Countries in South Asia

Sociodemographic, behavioral, obstetric, and healthcare factors associated with low weight at birth: a case-control study

Prediction and feature selection of low birth weight using machine learning algorithms

The very low-birth-weight rate: Principal predictor of neonatal mortality in industrialized populations

Benchmarking Machine Learning Models to Predict Low Birthweight Baby Outcomes and Identify Associated Risk Factors from an Extremely Unbalanced Large-Scale Dataset (Preprint)

Maternal Factors and Utilization of the Antenatal Care Services during Pregnancy Associated with Low Birth Weight in Rural Nepal: Analyses of the Antenatal Care and Birth Weight Records of the MATRI-SUMAN Trial

Determinants of low birth weight among newborns delivered in public hospitals of North Shewa Zone, Amhara region, Ethiopia: A case-control study (2023)

Multilevel multinomial regression analysis of factors associated with birth weight in sub-Saharan Africa

Factors associated with Low birth weight in Cambodia: A secondary analysis of Cambodia Demographic and Health Survey, 2021-2022

Contribution of low birth weight to childhood undernutrition in India: evidence from the national family health survey 2019-2021

Association of Maternal Risk Factors with Birth Weight of Newborn in a Tertiary Level Hospital in Dhaka City

Determinants of low birth weight and its effect on childhood health and nutritional outcomes in Bangladesh

Low birth weight trends in Organisation for Economic Co-operation and Development countries, 2000–2015: economic, health system and demographic conditionings

Spatial analysis and factors associated with low birth weight in Ghana using data from the 2017 Ghana Maternal Health Survey: spatial and multilevel analysis

Geographic and Racial Disparities in the Incidence of Low Birthweight in Pennsylvania

Influence of perinatal factors on full-term low-birth-weight infants and construction of a predictive model

National, regional, and global estimates of low birthweight in 2020, with trends from 2000: a systematic analysis