AI models’ performance in predicting live-birth occurrence
(A) ROC curves showing performance of live-birth occurrence prediction on internal test set. The green, orange, and blue ROC curves represent using the metadata-only model, the embryo image-only model, and the combined model, respectively.
(B) Illustration of features contributing to progression to live-birth occurrence by SHAP values. Features on the right of the risk explanation bar pushed the risk higher, and features on the left pushed the risk lower.
(C and D) ROC curves showing performance of live-birth occurrence prediction on (C) external validation set 1 (double embryo transfer pregnancy) and (D) external validation set 2 (single embryo transfer pregnancy).