Related Experiment Video
Updated: May 15, 2025
A Novel Use of Three-dimensional High-frequency Ultrasonography for Early Pregnancy Characterization in the Mouse
Published on: October 24, 2017
An explainable ultrasound-based machine learning model for predicting reproductive outcomes after frozen embryo
Fangfang Xu1, Qianqing Ma2, Penghao Lai3
1Department of Ultrasound, The First Affiliated Hospital of Anhui Medical University, Anhui, P. R. China.
Research Question:
Can an optimal machine learning model be developed to predict reproductive outcomes following frozen embryo transfer (FET)?
Design:
This prospective study included 787 infertile females who underwent FET. The participants were split into a training cohort (n = 550) and a test cohort (n = 237) at a ratio of seven to three. Radiomics features were extracted from ultrasound images of the endometrium and junctional zone. A radiomics model was developed to generate the radiomics score (rad score). Logistic regression was applied to process the clinical data and create a clinical model. A fusion machine learning model was developed by integrating the rad score with independent clinical data using the XGboost algorithm. The performance of the models was compared using the area under the receiver operating characteristic curve (AUC). The SHapley Additive exPlanations (SHAP) method was used to interpret and visualize the contributions of features to the outcomes of FET.
Results:
The fusion model demonstrated superior performance, as indicated by an AUC of 0.861 (95% CI 0.829-0.890), in the training cohort, surpassing both the clinical model (AUC 0.680, 95% CI 0.635-0.722; P < 0.001) and the radiomics model (AUC 0.814, 95% CI 0.777-0.848; P < 0.001). The SHAP summary plot reveals the impacts of each feature on the predictive model, and the rad score was found to be the main feature. SHAP force plots provided explanations at the individual level.
Conclusion:
An explainable machine learning model was established utilizing clinical data and ultrasound images to forecast the outcomes of FET. By utilizing the SHAP method, clinicians may better comprehend the contributors to the outcomes of FET in individual patients, and make better decisions before FET.
