Predicting low birth weight risks in pregnant women in Brazil using machine learning algorithms: data from the Araraquara cohort study

Autor(es)

Victor, Audencio ; Almeida, Francielly ; Xavier, Sancho Pedro ; Rondo, Patricia H. C.

• Tipo de documento: Artigo Científico
• Fonte: BMC PREGNANCY AND CHILDBIRTH; v. 25, n. 1, p. 9-pg., 2025-03-19.

Resumo

BackgroundLow birth weight (LBW) is a critical factor linked to neonatal morbidity and mortality. Early prediction is essential for timely interventions. This study aimed to develop and evaluate predictive models for LBW using machine learning algorithms, including Random Forest, XGBoost, Catboost, and LightGBM.MethodsWe analyzed data from 1,579 pregnant women enrolled in the Araraquara Cohort, a population-based longitudinal study. Predictor variables included maternal sociodemographic, clinical, and behavioral factors. Four ML algorithms Random Forest, XGBoost, CatBoost, and LightGBM, were trained using an 80/20 train-test split and 10-fold cross-validation. To address class imbalance, the Synthetic Minority Over-sampling Technique (SMOTE) was applied. Model performance was assessed using metrics such as area under the receiver operating characteristic curve (AUROC), F1-score, and precision-recall. Variable importance was evaluated using Shapley values.ResultsXGBoost demonstrated the best performance, achieving an AUROC of 0.94, followed by CatBoost (0.94), Random Forest (0.94), and LightGBM (0.94). Maternal gestational age was the most influential predictor, followed by marital status and prenatal care frequency. Behavioral factors, such as physical activity, also contributed to LBW risk. Shapley analysis provided interpretable insights into variable contributions, supporting the clinical applicability of the models.ConclusionMachine learning, combined with SMOTE, proved to be an effective approach for predicting LBW. XGBoost stood out as the most accurate model, but Catboost and Random Forest also provided solid results. These models can be applied to identify high-risk pregnancies, improving perinatal outcomes through early interventions. (AU)

• Processo FAPESP: 15/03333-6 - Relação entre adiposidade materna e adiposidade do concepto nos períodos fetal, neonatal e no primeiro ano de vida: estudo prospectivo de base populacional
• Beneficiário: Patricia Helen de Carvalho Rondó
• Modalidade de apoio: Auxílio à Pesquisa - Temático
• Processo FAPESP: 23/07936-3 - A influência do ganho de peso gestacional no crescimento fetal e nos desfechos neonatais: Estudo Coorte Araraquara
• Beneficiário: Audencio Victor
• Modalidade de apoio: Bolsas no Brasil - Doutorado