Early risk stratification of gestational diabetes using interpretable machine learning with first-trimester screening parameters

Table 1 Comparison of previous machine learning studies for gestational diabetes mellitus prediction[17-28]

Ref.	Sample size	Key predictors	Class imbalance	ML algorithms	Best ML performance	ML explainability	External validation	Calibration analysis
Wu et al[17]	31811	Clinical, biochemical, lipid, thyroid, and obstetric variables; 73-variable model and simplified 7-variable LR model	Not reported	DNN, SVM, KNN, and LR	DNN using 73 variables, AUC: 80%; 7-variable LR, AUC: 77%	Not reported	No	No
Xiong et al[18]	490	Routine blood tests, hepatic and renal function markers, and coagulation markers, particularly PT and aPTT	Not reported	SVM and LightGBM	SVM using PT and aPTT, sensitivity: 88.3%, specificity: 99.47%, and AUC: 94.2%	Not reported	No	No
Kaya et al[19]	97	First-visit venous plasma glucose level, maternal BMI, family history of DM, smoking, and obstetric history	Not reported	Extra trees, average blender, LightGBM, XGBoost, LR, and RF	XGBoost, AUC: 55.0%, accuracy: 66.7%, sensitivity: 80.0%, and specificity: 50.0% in nulliparous women; AUC: 73.3%, accuracy: 72.7%, sensitivity: 40.0%, and specificity: 100.0% in primiparous women	SHAP	No	No
Li et al[20]	7594	Forty-five first-trimester features; top predictors included pre-pregnancy BMI and maternal abdominal circumference at pregnancy initiation, and FPG and HbA1c at the end of the first trimester	Not reported	LR, XGBoost, RF, and other ML algorithms	XGBoost, AUC: 75% at pregnancy initiation and 99% at the end of the first trimester in the XHCM cohort; external validation AUC: 83% in the SPNPH cohort	Feature importance analysis	Yes	No
Zorlu et al[21]	400	Maternal characteristics, BMI, PAPP-A, and free β-hCG	Not reported	RF, GradBoost, and LR	GradBoost, AUC: 71.5% and accuracy: 71.3%	Not reported	No	No
Ni et al[22]	956	Common first-trimester clinical and laboratory variables selected by Spearman correlation analysis and Boruta algorithm, including pre-pregnancy BMI, SBP, and HDL-C	Not reported	LR, RF, XGBoost, LightGBM, MLP, KNN, and SVM	LR, AUC: 78.7% (95%CI: 72.3%-85.0%); RF, AUC: 77.6% (95%CI: 71.1%-84.1%)	Not reported	No	Yes
Zaky et al[23]	138	History of high glucose/diabetes, insulin, HOMA-IR, uric acid, cholesterol, urea, PT, NT-proBNP, thyroid markers, and routine blood markers	Not reported	RF, GradBoost, AdaBoost, DT, LR, SVM, Gaussian NB, KNN, CatBoost, XGBoost, LightGBM, and stacking ensemble	Stacking ensemble, accuracy: 88.8%, precision: 87.3%, sensitivity: 92.1%, and F1 score: 89.6%	SHAP	No	No
Bigdeli et al[24]	106	Age, BMI, previous abortion history, FPG, demographic variables, medical history, and clinical findings	SMOTE	DT, MLP, KNN, NB, RF, and XGBoost	RF, accuracy: 89%, precision: 86%, sensitivity: 92%, and AUC: 94%	Not reported	No	No
Pazaras et al[25]	797	Maternal demographics, obstetric history, lifestyle factors, and FFQ-derived dietary micronutrient intake	SMOTE, borderline SMOTE, adaptive synthetic sampling, and SMOTE-Tomek	LR, RF, extra trees, XGBoost, LightGBM, CatBoost, GradBoost, AdaBoost, and MLP	LR without resampling, AUC: 66.4% (95%CI: 54.2%-77.7%), sensitivity: 78.3%, and NPV: 93.2%; reduced 9-feature model, AUC: 71.2% (95%CI: 58.9%-82.5%)	SHAP	No	Yes
Prashanthan and Prashanthan[26]	10000	Synthetic demographic characteristics, clinical risk factors, and first-trimester laboratory parameters including random blood sugar, post-prandial blood sugar, HbA1c, and OGTT values	SMOTE	Multiple ML algorithms	Best model, accuracy: 71.7% and AUC: 76.9%	SHAP	No	No
Zhai et al[27]	534	BMI, SAT, VAT, maternal clinical characteristics, and first-trimester ultrasonographic indicators	IPW	XGBoost, ANN, SVM, MLR, and RF	XGBoost with GA-selected features, internal test AUC: 96.2%; external validation AUC: 87.8%, sensitivity: 70.0%, and specificity: 93.5%	GA-based feature selection and heatmaps	Yes	No
Louzoun et al[28]	596 twin pregnancies	WBC count, platelet levels, BMI, and previous GDM	Not reported	LightGBM, XGBoost, and LR	LightGBM, AUC: 72% (95%CI: 69%-75%); detection rates: 28% and 42% at false-positive rates of 10% and 20%, respectively	Not reported	No	No
Present study	2756	Maternal age, MAP, pregestational weight, PAPP-A, and PlGF	SMOTE, ROS, and RUS	LR, GradBoost, XGBoost, AdaBoost, LightGBM, SVM, RF, and MLP	GradBoost + ROS, AUC: 76.8%, sensitivity: 64.7%, specificity: 77.7%, PPV: 28.9%, NPV: 94.0%, and F1 score: 40.0%	SHAP and patient-level heatmaps	No	Yes

ML: Machine learning; LR: Logistic regression; PT: Prothrombin time; aPTT: Activated partial thromboplastin time; BMI: Body mass index; DM: Diabetes mellitus; FPG: Fasting plasma glucose; HbA1c: Glycated hemoglobin; PAPP-A: Pregnancy-associated plasma protein A; β-hCG: β-human chorionic gonadotropin; SBP: Systolic blood pressure; HDL-C: High-density lipoprotein cholesterol; HOMA-IR: Homeostatic model assessment for insulin resistance; NT-proBNP: N-terminal pro-B-type natriuretic peptide; FFQ: Food frequency questionnaire; OGTT: Oral glucose tolerance test; SAT: Subcutaneous adipose tissue; VAT: Visceral adipose tissue; WBC: White blood cell count; GDM: Gestational diabetes mellitus; MAP: Mean arterial pressure; PlGF: Placental growth factor; IPW: Inverse probability weighting; SMOTE: Synthetic minority oversampling technique; RUS: Random undersampling; ROS: Random oversampling; DNN: Deep neural network; SVM: Support vector machine; KNN: K-nearest neighbors; LightGBM: Light gradient boosting machine; XGBoost: Extreme gradient boosting; RF: Random forest; GradBoost: Gradient boosting; MLP: Multilayer perceptron; AdaBoost: Adaptive boosting; DT: Decision tree; Gaussian NB: Gaussian naïve Bayes; CatBoost: Categorical boosting; NB: Naïve Bayes; ANN: Artificial neural network; MLR: Multivariate logistic regression; CI: Confidence interval; AUC: Area under the receiver operating characteristic curve; NPV: Negative predictive value; GA: Genetic algorithm; SHAP: SHapley Additive exPlanations; PPV: Positive predictive value.