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.

Learning from imbalanced fetal outcomes of systemic lupus erythematosus in artificial neural networks

Automated 3D Fetal Brain Segmentation Using an Optimized Deep Learning Approach

Adverse pregnancy outcomes in women with systemic lupus erythematosus: can we improve predictions with machine learning?

AI driven interpretable deep learning based fetal health classification

Using Deep Learning in a Monocentric Study to Characterize Maternal Immune Environment for Predicting Pregnancy Outcomes in the Recurrent Reproductive Failure Patients

Comparison of machine learning and logistic regression as predictive models for adverse maternal and neonatal outcomes of preeclampsia: A retrospective study

Assessing Performance Across Various Machine Learning Algorithms with Integrated Feature Selection for Fetal Heart Classification

Construction and application of fetal loss risk model in systemic lupus erythematosus patients with mild disease severity

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

Prediction of neonatal death in pregnant women in an intensive care unit: Application of machine learning models

Fetal weight estimation based on deep neural network: a retrospective observational study

Prediction of pre-eclampsia complicated by fetal growth restriction and its perinatal outcome based on an artificial neural network model

Deep Learning with Information Fusion and Model Interpretation for Health Monitoring of Fetus based on Long-term Prenatal Electronic Fetal Heart Rate Monitoring Data

Artificial intelligence-driven predictive framework for early detection of still birth

An innovative supervised longitudinal learning procedure of recurrent neural networks with temporal data augmentation: Insights from predicting fetal macrosomia and large-for-gestational age

Prenatal Prediction of Neonatal Respiratory Morbidity: a Radiomics Method Based on Imbalanced Few-Shot Fetal Lung Ultrasound Images

Unveiling the Unborn: Advancing Fetal Health Classification through Machine Learning

Clinical data-based modeling of IVF live birth outcome and its application

Closing the Gap in High-Risk Pregnancy Care Using Machine Learning and Human-AI Collaboration

Automatic classification of fetal heart rate based on a multi-scale LSTM network

Fetal biometric based abnormality detection during prenatal development using deep learning techniques