Machine Learning-based Classifiers for the Prediction of Low Birth Weight

Healthc Inform Res. 2023 Jan;29(1):54-63. doi: 10.4258/hir.2023.29.1.54. Epub 2023 Jan 31.

Abstract

Objectives: Low birth weight (LBW) is a global concern associated with fetal and neonatal mortality as well as adverse consequences such as intellectual disability, impaired cognitive development, and chronic diseases in adulthood. Numerous factors contribute to LBW and vary based on the region. The main objectives of this study were to compare four machine learning classifiers in the prediction of LBW and to determine the most important factors related to this phenomenon in Hamadan, Iran.

Methods: We carried out a retrospective cross-sectional study on a dataset collected from Fatemieh Hospital in 2017 that included 741 mother-newborn pairs and 13 potential factors. Decision tree, random forest, artificial neural network, support vector machine, and logistic regression (LR) methods were used to predict LBW, with five evaluation criteria utilized to compare performance.

Results: Our findings revealed a 7% prevalence of LBW. The average accuracy of all models was 87% or higher. The LR method provided a sensitivity, specificity, positive likelihood ratio, negative likelihood ratio, and accuracy of 74%, 89%, 7.04%, 29%, and 88%, respectively. Using LR, gestational age, number of abortions, gravida, consanguinity, maternal age at delivery, and neonatal sex were determined to be the six most important variables associated with LBW.

Conclusions: Our findings underscore the importance of facilitating timely diagnosis of causes of abortion, providing genetic counseling to consanguineous couples, and strengthening care before and during pregnancy (particularly for young mothers) to reduce LBW.

Keywords: Abortion; Gestational Age; Induced; Infant; Logistic Models; Low Birth Weight; Machine Learning.