Please use this identifier to cite or link to this item:
https://doi.org/10.1186/s12884-023-05766-4
Title: | Development of machine learning models to predict gestational diabetes risk in the first half of pregnancy | Authors: | Cubillos, Gabriel Monckeberg, Max Plaza, Alejandra Morgan, Maria Estevez, Pablo A Choolani, Mahesh Kemp, Matthew W Illanes, Sebastian E Perez, Claudio A |
Keywords: | Science & Technology Life Sciences & Biomedicine Obstetrics & Gynecology Gestational diabetes mellitus (GDM) GDM risk prediction Machine learning models Data augmentation Widely available variables INTERNATIONAL ASSOCIATION MELLITUS CLASSIFICATION HYPERGLYCEMIA ALGORITHM GLUCOSE TERM |
Issue Date: | 23-Jun-2023 | Publisher: | BMC | Citation: | Cubillos, Gabriel, Monckeberg, Max, Plaza, Alejandra, Morgan, Maria, Estevez, Pablo A, Choolani, Mahesh, Kemp, Matthew W, Illanes, Sebastian E, Perez, Claudio A (2023-06-23). Development of machine learning models to predict gestational diabetes risk in the first half of pregnancy. BMC PREGNANCY AND CHILDBIRTH 23 (1). ScholarBank@NUS Repository. https://doi.org/10.1186/s12884-023-05766-4 | Abstract: | Background: Early prediction of Gestational Diabetes Mellitus (GDM) risk is of particular importance as it may enable more efficacious interventions and reduce cumulative injury to mother and fetus. The aim of this study is to develop machine learning (ML) models, for the early prediction of GDM using widely available variables, facilitating early intervention, and making possible to apply the prediction models in places where there is no access to more complex examinations. Methods: The dataset used in this study includes registries from 1,611 pregnancies. Twelve different ML models and their hyperparameters were optimized to achieve early and high prediction performance of GDM. A data augmentation method was used in training to improve prediction results. Three methods were used to select the most relevant variables for GDM prediction. After training, the models ranked with the highest Area under the Receiver Operating Characteristic Curve (AUCROC), were assessed on the validation set. Models with the best results were assessed in the test set as a measure of generalization performance. Results: Our method allows identifying many possible models for various levels of sensitivity and specificity. Four models achieved a high sensitivity of 0.82, a specificity in the range 0.72–0.74, accuracy between 0.73–0.75, and AUCROC of 0.81. These models required between 7 and 12 input variables. Another possible choice could be a model with sensitivity of 0.89 that requires just 5 variables reaching an accuracy of 0.65, a specificity of 0.62, and AUCROC of 0.82. Conclusions: The principal findings of our study are: Early prediction of GDM within early stages of pregnancy using regular examinations/exams; the development and optimization of twelve different ML models and their hyperparameters to achieve the highest prediction performance; a novel data augmentation method is proposed to allow reaching excellent GDM prediction results with various models. | Source Title: | BMC PREGNANCY AND CHILDBIRTH | URI: | https://scholarbank.nus.edu.sg/handle/10635/246505 | ISSN: | 1471-2393 | DOI: | 10.1186/s12884-023-05766-4 |
Appears in Collections: | Staff Publications Elements |
Show full item record
Files in This Item:
File | Description | Size | Format | Access Settings | Version | |
---|---|---|---|---|---|---|
Development of machine learning models to predict gestational diabetes risk in the first half of pregnancy.pdf | 1.83 MB | Adobe PDF | OPEN | None | View/Download |
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.