A stepwise prediction and interpretation of gestational diabetes mellitus: Foster the practical application of machine learning in clinical decision

Zhou，Fang2; Ran，Xiao4,5; Song，Fangliang4; Wu，Qinglan2; Jia，Yuan2; Liang，Ying4; Chen，Suichen2; Zhang，Guojun2; Dong，Jie1; Wang，Yukun2,3

2024-06-30

10.1016/j.heliyon.2024.e32709

Heliyon   影响因子和分区

2405-8440

Background: Machine learning has shown to be an effective method for early prediction and intervention of Gestational diabetes mellitus (GDM), which greatly decreases GDM incidence, reduces maternal and infant complications and improves the prognosis. However, there is still much room for improvement in data quality, feature dimension, and accuracy. The contributions and mechanism explanations of clinical data at different pregnancy stages to the prediction accuracy are still lacking. More importantly, current models still face notable obstacles in practical applications due to the complex and diverse input features and difficulties in redeployment. As a result, a simple, practical but accurate enough model is urgently needed. Design and methods: In this study, 2309 samples from two public hospitals in Shenzhen, China were collected for analysis. Different algorithms were systematically compared to build a robust and stepwise prediction system (level A to C) based on advanced machine learning, and models under different levels were interpreted. Results: XGBoost reported the best performance with ACC of 0.922, 0.859 and 0.850, AUC of 0.974, 0.924 and 0.913 for the selected level A to C models in the test set, respectively. Tree-based feature importance and SHAP method successfully identified the commonly recognized risk factors, while indicated new inconsistent impact trends for GDM in different stages of pregnancy. Conclusion: A stepwise prediction system was successfully established. A practical tool that enables a quick prediction of GDM was released at https://github.com/ifyoungnet/MedGDM.This study is expected to provide a more detailed profiling of GDM risk and lay the foundation for the application of the model in practice.

Computational medicine Gestational diabetes mellitus Machine learning Model explanation Risk prediction

SCI

英语

第一

2-s2.0-85195876178

Scopus

1.Xiangya School of Pharmaceutical Sciences,Central South University,Changsha,410083,China
2.Department of Pharmacy,Southern University of Science and Technology Hospital,Shenzhen,Guangdong,518055,China
3.Department of Pharmacology,School of Medicine,Southern University of Science and Technology,Shenzhen,Guangdong,518055,China
4.School of Food Science and Engineering,Central South University of Forestry and Technology,Changsha,410004,China
5.SINOCARE Inc.,Changsha,410004,China

Zhou，Fang,Ran，Xiao,Song，Fangliang,et al. A stepwise prediction and interpretation of gestational diabetes mellitus: Foster the practical application of machine learning in clinical decision[J]. Heliyon,2024,10(12).

Zhou，Fang.,Ran，Xiao.,Song，Fangliang.,Wu，Qinglan.,Jia，Yuan.,...&Wang，Yukun.(2024).A stepwise prediction and interpretation of gestational diabetes mellitus: Foster the practical application of machine learning in clinical decision.Heliyon,10(12).

Zhou，Fang,et al."A stepwise prediction and interpretation of gestational diabetes mellitus: Foster the practical application of machine learning in clinical decision".Heliyon 10.12(2024).