Accepted for/Published in: JMIR Medical Informatics
Date Submitted: Jan 21, 2025
Date Accepted: May 21, 2025
Interpretable Machine Learning for Predicting Adverse Pregnancy Outcomes in Gestational Diabetes: A Retrospective Cohort Study
ABSTRACT
Background:
Gestational diabetes mellitus (GDM) affects over 5% of pregnancies globally, elevating risks of type 2 diabetes postpartum and complications such as fetal death, miscarriage, and congenital abnormalities. Effective GDM management is essential to balance glycemic control and pregnancy outcomes.
Objective:
To develop interpretable machine learning models using GDM datasets for predicting adverse pregnancy outcomes and identifying key factors through the SHAP algorithm, thus supporting improved maternal and infant health.
Methods:
Data preprocessing and feature selection were performed, with ADASYN used to address class imbalance. Classification models, including logistic regression, random forest, SVM, and XGBoost, were built and enhanced through stacking method. Model interpretability was assessed with SHAP to quantify feature contributions.
Results:
Among 1,670 patients, 200 experienced adverse outcomes. The stacked model achieved 97.9% accuracy and an AUC of 0.96 in external validation. SHAP analysis highlighted key predictive factors, such as gestational age, fasting glucose, and blood pressure, supporting model reliability.
Conclusions:
This study underscores the potential of machine learning in predicting adverse outcomes in GDM, with interpretable features offering valuable clinical insights to enhance pregnancy management and maternal-infant health.
Citation
Request queued. Please wait while the file is being generated. It may take some time.
Copyright
© The authors. All rights reserved. This is a privileged document currently under peer-review/community review (or an accepted/rejected manuscript). Authors have provided JMIR Publications with an exclusive license to publish this preprint on it's website for review and ahead-of-print citation purposes only. While the final peer-reviewed paper may be licensed under a cc-by license on publication, at this stage authors and publisher expressively prohibit redistribution of this draft paper other than for review purposes.