Development and external validation of a three-stage model to predict live birth after fresh IVF/ICSI embryo transfer

OA: gold CC-BY-4.0
📄 Open PDF Full text JSON
⚙ AI-generated summary by qwen3.7-flash+body, 2026-10-11 ⓘ

A three-stage machine learning pipeline integrating baseline, stimulation, and embryo transfer data accurately predicts live birth after fresh IVF/ICSI in women aged 20–45 years with no stated link to endometriosis or adenomyosis.

One-sentence paraphrase of the abstract; not a substitute for reading it. No clinical advice. How this works

⚙ AI-generated deep summary by qwen3.7-flash, 2026-10-11 · read from full text ⓘ

This retrospective study developed and validated a three-stage machine learning pipeline to predict live birth following fresh IVF or ICSI embryo transfers. The framework integrated clinical data from baseline, ovarian stimulation, and embryo development phases using ten base models and ensemble strategies within cohorts of over 10,000 patients across two Chinese centers. The authors demonstrated that this dynamic, multi-stage approach outperformed traditional single-time-point tools in accuracy and calibration while providing interpretable insights via SHAP analysis. The paper does not explicitly discuss endometriosis or adenomyosis; it was included in the corpus via a keyword match in the upstream search index.

Read from the paper's body, not the abstract. Not a substitute for reading the paper. No clinical advice. How this works

Full text 31,945 characters · extracted from pmc-nxml · 5 sections · click to expand

Intro

Globally, infertility affects approximately 17.5% of individuals during their lifetime, representing a serious medical and psychosocial burden ( 1 ). Assisted reproductive technologies, particularly in vitro fertilization (IVF) and intracytoplasmic sperm injection (ICSI), have revolutionized the treatment of infertility, offering hope to millions of couples worldwide. Despite continuous advances in laboratory techniques and clinical protocols, the live birth rates per cycle remain relatively modest at 30–40% and decline significantly with age. The complexity of IVF/ICSI outcomes arises from multiple interacting factors across the treatment continuum, including patient demographics, ovarian reserve, hormonal dynamics, stimulation protocols, embryological characteristics, and endometrial receptivity ( 2 , 3 ). Predicting which patients will achieve a live birth remains a clinical challenge. Traditional prediction tools, such as morphological embryo grading and logistic regression models, are limited by subjectivity, linear assumptions, and an inability to capture complex interactions ( 4 ). Most existing models operate at a single time point—either before treatment or after embryo transfer, and fail to integrate information as it accumulates along the clinical pathway ( 5 ). While some studies have developed two-stage or multi-phase models, their performance has been modest, and few focus specifically on live birth ( 6 , 7 ). There is a clear need for a dynamic, multi-stage framework that reflects how clinical decisions are actually made: initial assessment, stimulation monitoring, and embryo transfer decision-making. Machine learning offers new tools for integrating high-dimensional clinical data. Recent studies have shown promise: Borji et al. developed a TabTransformer-based deep learning pipeline on large-scale Human Fertilization and Embryology Authority dataset ( 8 ); Zhang et al. applied artificial neural networks to predict live birth in natural-cycle IVF ( 9 ); and Sadegh-Zadeh et al. achieved 96.35% accuracy using ensemble methods ( 10 ). However, most existing models are single-staged, lack rigorous external validation, or treat machine learning as a black box. Few studies have systematically compared basic learners with advanced ensemble strategies or validated models across independent cohorts. Equally important, model interpretability and clinically actionable threshold identification have rarely been explored, which substantially limits clinical translation and real-world application. In this study, we developed a three-stage machine learning pipeline for live birth prediction at key clinical decision points: baseline, ovarian stimulation, and embryo transfer. Ten base models and three ensemble strategies were evaluated using internal and external validation. We used SHapley Additive exPlanations (SHAP), restricted cubic splines (RCS), and decision curve analysis (DCA) to build a transparent, dynamic, and clinically applicable tool for personalized IVF/ICSI assessment and patient counseling.

Methods

This retrospective study included 8,389 women undergoing their first fresh IVF/ICSI cycles at a university-affiliated reproductive center in Shenzhen, China from January 2019 to December 2024. A temporal external validation cohort comprising 2,058 patients undergoing their first fresh cycles was enrolled from another independent tertiary hospital in Wuhan, China between June 2024 and June 2025. Data were retrieved from standardized electronic medical records at both centers. Inclusion criteria were: female age 20–45 years; fresh embryo transfer; available follow-up data for the primary outcome. Exclusion criteria were: cycles cancellation before oocyte retrieval or embryo transfer; use of donor gametes; missing live birth outcome; preimplantation genetic testing (PGT) cycles. The primary outcome was live birth, defined as delivery of at least one live infant after 24 weeks of gestation. Multiple pregnancies resulting in at least one live birth were counted as a single live-birth event. The study was approved by the Research Ethics Committee of each hospital. Due to the retrospective, anonymized nature of the study, informed consent was waived. The dataset was split into three mutually exclusive subsets by stratified random sampling to preserve the live birth distribution. The training set (70%, n=5871) was used for model training, feature engineering, and internal cross-validation. The internal validation set (15%, n=1259) was used for monitoring training, early stopping, hyperparameter optimization, Platt scaling, and dynamic pruning. The test set (15%, n=1259) was a fully locked unseen dataset for final model evaluation. Flowchart of this study is shown in Figure 1 . Study flowchart and three-stage sequential machine learning framework. To mimic real-world clinical workflow and prevent look-ahead information leakage, a cumulative staging strategy was applied. Three sequential clinical stages were defined: Stage 1 (baseline stage): Variables available at initial consultation, including demographics, infertility causes and type, baseline hormones on day 2–3 of the menstrual cycle, and semen parameters. Ovulatory disorder (female factor 3) was defined according to the Rotterdam criteria, including polycystic ovary syndrome (PCOS), hypothalamic hypogonadism, hyperprolactinemia, and other causes of oligo-ovulation or anovulation. Diminished ovarian reserve (DOR, female factor 4) was defined per the Bologna criteria; advanced maternal age in this study referred to female age ≥38 years at oocyte retrieval. Stage 2 (stimulation stage): All Stage 1 variables plus ovarian stimulation protocol, gonadotropin dosage, trigger day hormone levels, and endometrial thickness (EMT). Stage 3 (embryo development stage): All Stage 2 variables plus oocyte yield, fertilization, embryo development metrics, and embryo characteristics. Variables acquired in later stages were not introduced into earlier-stage models. Fresh transfer included Day-3 cleavage embryos or Day-5 blastocysts. High-quality Day 3 embryos were defined as 7–9 blastomeres, ≤10% fragmentation, and no multinucleation. High-quality blastocysts were defined as ≥3BB by Gardner grading. A rigorous leakage-proof preprocessing framework was established. All continuous variables were winsorized at the 1st and 99th percentiles to reduce outlier effects. Overall missingness was low across all predictors, and all missing values were imputed using MissForest trained exclusively on the training set. Missing data were imputed after dataset splitting under a strict stage-isolation strategy. Continuous missing values were imputed by the MissForest algorithm (n_estimators=30, max_iter=5); categorical variables were imputed by mode. The imputation model was trained only on the training set and applied to validation and test sets with a fixed random seed (42). All preprocessing steps were fitted exclusively on the training set to prevent data leakage. Variables with variance inflation factor (VIF) > 10 were removed to address multicollinearity. A dual preprocessing pipeline was used: For tree-based models (RF, XGBoost, LightGBM, CatBoost), variables were retained in their original scales, and class imbalance was handled via class weight balancing. For linear models and neural networks, continuous variables were normalized using the RobustScaler and transformed with the Yeo-Johnson transformation to reduce skewness, and categorical variables were one-hot encoded. Given the near-balanced outcome distribution, no oversampling was applied. Feature selection was performed exclusively on the training set using a multi-algorithm voting framework: 1. LASSO regression with 5-fold cross-validation, selecting variables with non-zero coefficients using the minimum λ criterion. 2. Random forest feature importance (n_estimators = 100) to rank variables by mean decrease in impurity. 3. Recursive feature elimination (RFE) with a logistic regression estimator to iteratively remove low-impact features. 4. Boruta algorithm (max_depth = 7) to identify features significantly more important than random shadow variables. Features supported by at least two algorithms, or included in the predefined clinical whitelist (clinical consensus), were retained for subsequent modeling. Selection results of key variables were shown in Supplementary Table 2 . Ten machine learning models were trained: Logistic Regression (LR), Random Forest (RF), XGBoost, LightGBM, CatBoost, Support Vector Machine (SVM), Multilayer Perceptron (MLP), Naive Bayes (NB), K-Nearest Neighbors (KNN), and Decision Tree (DT). Hyperparameter optimization was performed in Optuna with the Tree-structured Parzen Estimator (TPE) and 30 trials, using 5-fold stratified cross validation with area under the receiver operating characteristic curve (AUC-ROC) as the objective. Early stopping was applied with a patience of 100 rounds. Platt scaling was conducted on the internal validation set to calibrate predicted probabilities. A three-dimensional ensemble strategy was implemented: 1. Static Ensemble Selection (SES): NSGAII algorithm (population=30, generations=20) selected a Pareto-optimal subset of base learners by minimizing prediction error and inter-model correlation. 2. Dynamic Ensemble Pruning (DEP): For each test sample, 15 nearest neighbors were identified in the internal validation set using Euclidean distance. The worst 50% of models were dynamically pruned, and average predictions from the top-performing subset were used. 3. Stacking: A cross-validated logistic regression served as the meta-learner to combine pruned base model predictions. All models were compared in the internal validation set. The best-performing base model and three ensemble strategies were evaluated in the test and external validation cohorts. Model performance was assessed using AUC-ROC, precision-recall curve (PR-AUC), accuracy, sensitivity, specificity, Brier score, positive predictive value (PPV), negative predictive value (NPV) and F1 score. Calibration was evaluated by calibration curves. Clinical utility was assessed by DCA across threshold probabilities of 1%–99% against treat-all and treat-none strategies. Bootstrapping with 1000 replicates was used to estimate 95% confidence intervals. The DeLong test was used for paired AUC comparisons. Net reclassification improvement (NRI) and integrated discrimination improvement (IDI) were used to quantify incremental gains across stages. Subgroup analyses were stratified by infertility cause, type, female age, Body Mass Index (BMI), and polycystic ovary syndrome (PCOS) status, with interaction P -values to test heterogeneity. To interpret the complex SES-DEP ensemble model, surrogate model distillation was first performed: a shallow XGBoost regressor (max_depth = 4, learning_rate = 0.05) was trained to approximate the ensemble’s predictions. A pre-specified threshold of surrogate-model R² > 0.8 was planned for proceeding to SHAP analysis and to quantify the contribution of each feature to individual predictions (Tree-SHAP). Mean absolute SHAP values were used to rank global feature importance. SHAP dependence plots were used to illustrate directional effects. RCS regression was used to evaluate the nonlinear association between continuous variables and live birth, with knots at the 5th, 35th, 65th, and 95th percentiles. All RCS models were strictly adjusted for major clinical confounders. Subgroup RCS curves were conducted stratified by female age at 35 years. P for overall association and P for non-linearity were calculated to evaluate linear and non-linear trends. Continuous variables are presented as median with interquartile range (IQR) and compared using the Wilcoxon rank-sum test. Categorical variables are summarized as counts and percentages and compared using the chi-squared test or Fisher’s exact test. All machine learning analyses were performed using Python version 3.9. Data preprocessing was implemented using scikit-learn and imbalanced-learn. Hyperparameter optimization was performed using Optuna with TPE, and SES was implemented using DEAP. Statistical analyses and visualizations were performed using SciPy, Statsmodels, SHAP, Matplotlib, and Seaborn. All statistical tests were two-sided, and a P value < 0.05 was considered statistically significant.

Results

Figure 1 showed the flowchart of this study. A total of 10,447 fresh IVF/ICSI cycles were included: 5,871 in training, 1,259 in internal validation, 1,259 in testing, and 2,058 in external validation. Live birth rates were 42.3%, 44.4%, 43.4%, and 44.7%, respectively. Baseline characteristics were shown in Table 1 . Most baseline characteristics were well balanced among the internal cohorts (all P > 0.05), confirming effective stratified randomization. Within the training cohort, women achieving live birth were significantly younger (median age 32 vs. 35 years, P < 0.001), with higher ovarian reserve indicated by higher anti-Müllerian hormone (AMH) and antral follicle count (AFC). Male partners in the live birth group were also younger with lower sperm DNA fragmentation index (DFI). The live birth group had a higher proportion of ovulatory disorders and a lower rate of DOR or advanced maternal age (both P < 0.001). Baseline demographic and clinical characteristics of the study population across four cohorts. Data are presented as median [interquartile range] for continuous variables and n (%) for categorical variables. *P-value comparing training, internal, external validation, and testing cohorts; †P-value comparing live birth and non-live birth groups in the training cohort. *DOR, diminished ovarian reserve; BMI, body mass index; FSH, follicle-stimulating hormone; LH, luteinizing hormone; E2, estradiol; PRL, prolactin; AMH, anti-Müllerian hormone; AFC, antral follicle count; DFI, DNA fragmentation index. HDS, high DNA stainability. The bold values in Table 1 indicate statistically significant differences (P < 0.05). Ovarian stimulation, embryology, and transfer parameters are presented in Supplementary Table 1 . The live birth group more commonly underwent the long GnRH agonist protocol, had thicker trigger-day endometrium, more oocytes retrieved, more MII oocytes, and more 2PN zygotes (all P < 0.001). Blastocyst formation rate, blastocyst count, cleavage embryos, and frozen embryos were all significantly higher (all P < 0.001). Patients with live birth also had higher rates of day 5 blastocyst transfer and single high-quality embryo transfer (both P < 0.001). The external validation cohort showed expected differences in several baseline and treatment variables, providing a geographically and temporally independent dataset for robust external validation. We used a multi-algorithm voting framework including LASSO, RF, RFE, Boruta, and clinical priority evaluation to identify key predictors across the three sequential stages. For each stage, LASSO regression was used for dimensionality reduction, with optimal penalty parameters (α) validated via MSE paths, and coefficient shrinkage at the best α confirmed stable feature elimination. RF Gini importance plots verified feature relevance across stages ( Supplementary Figure 1 ). The algorithm intersection network ( Supplementary Figure 2 ) visualized the consensus features for each stage, including clinically meaningful predictors such as female age, trigger-day progesterone, high-quality embryos transferred, and EMT, confirming clinical plausibility. Detailed voting scores are shown in Supplementary Table 2 . Variables were retained if selected by at least two algorithms or judged to be clinically indispensable ( Supplementary Table 3 ). Stage-specific prediction models were developed and evaluated in internal validation and test cohort. In internal validation cohort, 10 base models (KNN, RF, CatBoost, LightGBM, XGBoost, NB, SVM, LR, MLP, DT) were compared ( Supplementary Figures 3A–D ). The best-performing model was then compared with three ensemble strategies (SES, Stacking, and DEP ensemble). The SES ensemble displayed progressive AUC improvement from Stage 1 (0.699) to Stage 2 (0.723) and Stage 3 (0.754). The ensemble methods outperformed the best base model, with detailed performance shown in Supplementary Figures 4A–D . In the test cohort ( Figure 2A , Table 2 ), stacking ensemble model yielded highest AUC of 0.678 (95% CI: 0.649-0.708) at the baseline stage. The stimulation stage model achieved an AUC of 0.725 (95% CI: 0.698–0.751), representing a significant improvement over stage 1 (DeLong test P<0.001; NRI = 0.630, IDI = 0.072; Supplementary Figure 5A ). The SES ensemble model reached the optimal AUC of 0.731 (95% CI: 0.704-0.757) at stage 3, while the improvement was not statistically significant from stage 2. Notably, stage 3 still demonstrated a significant predictive advantage relative to the baseline stage (NRI = 0.470, IDI = 0.083, P < 0.001). Performance comparison of the optimal base model and three ensemble strategies across three stages in test cohort. (A) ROC curves showing the discriminative performance at Stage 1, baseline stage; Stage 2, stimulation stage; and Stage 3 embryo development stages. (B) Calibration curves, with Brier scores noted. (C) Decision curve analysis. Performance of different machine learning models for predicting live birth outcome across three stages in test cohort. Stage 1, baseline stage, the pre-IVF assessment period; Stage 2, stimulation stage, the ovarian stimulation phase; Stage 3, embryo development stage, embryological data including characteristics of embryos selected for transfer; AUC, area under the receiver operating characteristic curve; AUPRC, area under the precision-recall curve; PPV, positive predictive value; NPV, negative predictive value; SES, Static Ensemble Selection; DEP, Dynamic Ensemble Pruning; Best Base, best-performing base model. PR-AUC increased sequentially from the baseline to the embryo development stage ( Supplementary Figure 5B ). Calibration curves showed good agreement between predicted and observed live birth rates ( Figure 2B ). DCA was conducted to quantify the clinical net benefit ( Figure 2C ). Probability-stratified live-birth rates based on prespecified clinical risk cutoffs are shown in Supplementary Table 4 , illustrating observed live-birth outcomes across different predicted-risk subgroups. Model performance was further verified in an independent external validation cohort of 2,058 patients. Baseline characteristics by live birth outcome are presented in Supplementary Table 5 . Patients with live birth were significantly younger, with higher AMH and AFC, lower BMI, and a lower rate of advanced age or DOR (all P < 0.05). Supplementary Table 6 demonstrated women with live birth had lower gonadotropin dosage, higher peak E2, thicker endometrium, more oocytes retrieved, and higher blastocyst formation rates (all P < 0.001). Supplementary Table 7 summarizes the model performance in the final stages. External validation shows slightly lower discrimination than internal testing, due to inter-center differences in patient profiles and clinical practice. Among the models, the SES ensemble performed optimally and consistently at all stages, achieving the highest AUC of 0.719 (95% CI: 0.696-0.741) at the embryo development stage ( Figures 3A–C ). Calibration was acceptable across all models, with Brier scores ranging from 0.213 to 0.230. The findings supported favorable model generalizability. Performance of machine learning models across three stages in the external validation cohort. ROC curves showing the discriminative performance of three stages. (A) Stage 1, baseline stage, (B) Stage 2, stimulation stage, (C) Stage 3, embryo development stages. To identify key predictors of live birth, we performed SHAP analysis across three sequential IVF/ICSI stages. Female age emerged as the most important predictor across all stages, exerting a strong negative impact on live birth likelihood. In the baseline stage ( Figure 4A ), AFC and male age were also key variables. In the stimulation stage ( Figure 4B ), down-regulation dose, trigger-day progesterone and EMT were the second important predictors. In the embryo development stage ( Figure 4C ), the number of high-quality embryos transferred and trigger-day progesterone were the next most influential predictors, followed by high-quality blastocysts and EMT. Bar plots showed the mean SHAP values for all features ( Supplementary Figures 6A–C ). These results confirmed the stepwise contributions of clinical, hormonal, and embryological data across the IVF/ICSI treatment pathway. SHAP analysis revealing feature importance and dependence in the ensemble model across three stages. SHAP summary plot (beeswarm) showing the relative importance of variables in predicting live birth (A) baseline stage, (B) stimulation stage, (C) embryo development stage). (D) SHAP dependence plot showing the effect of trigger-day endometrial thickness (EMT) on model-predicted live birth probability, stratified by female age. (E) SHAP dependence plot depicting the association between trigger-day progesterone (P) and live birth prediction, stratified by female age. (F) SHAP dependence plot of the number of oocytes retrieved, demonstrating its impact on live birth prediction across different female age groups. (G) SHAP dependence plot of high-quality (HQ) blastocyst count, visualizing its contribution to live birth prediction by female age. Female factor 4 indicated advanced age/DOR. SHAP dependence plots identified key effect modifiers. The negative impact of thin endometrium (10 mm was also less pronounced in this group ( Figure 4D ). Elevated trigger-day progesterone was more detrimental in younger women ( Figure 4E ). Oocyte yield and high-quality blastocysts conferred greater benefits to older women ( Figures 4F, G ). As shown in Supplementary Figure 6D , high AMH and AFC mitigated the adverse impact of advanced female age, while younger women were less affected by poor ovarian response. Down-regulation dose showed optimal outcomes at 1–3.75 mg. Critically, neither thin endometrium nor elevated progesterone could be rescued by increasing the number of transferred high-quality embryos. RCS analysis revealed significant non-linear associations between key continuous variables and live birth probability ( Supplementary Figure 7 , Supplementary Table 8 ). Female and male age exhibited curvilinear trends. AMH and AFC were also non-linearly positively associated with live birth probability. Down-regulation and antagonist doses exhibited age-dependent non-linear effects. For women under 35, optimal range of down-regulation dose is approximately 2.5–3.0 mg. Trigger-day hormone profiles were also stratified by female age: in women aged ≥35, FSH and LH showed significant negative linear trends, whereas E2 followed an inverted U-shaped curve. Progesterone demonstrated a linear negative association in women <35 years. EMT on trigger day plateaued at 12mm, with no additional benefit beyond this threshold. Oocytes retrieved, cleaved embryos, high-quality blastocysts, and frozen embryos all showed threshold-dependent benefits, peaking at counts of 15, 11, 4, and 6 respectively before declining. We further assessed model performance across clinically relevant subgroups to evaluate its robustness ( Supplementary Figure 8 ). While the overall AUC was 0.730 (95% CI: 0.704–0.757), significant heterogeneity was detected in subgroups stratified by infertility factors, female age, oocytes retrieved, infertility type, and PCOS status (all P for interaction < 0.05). Specifically, the model performed better in patients with endometriosis (AUC = 0.799) or DOR (AUC = 0.787), secondary infertility (AUC = 0.753), age ≥35 years (AUC = 0.751), and ≥10 oocytes retrieved (AUC = 0.751). Poorer performance was seen in those with ovulatory disorder (AUC = 0.631), age <35 years (AUC = 0.671), PCOS (AUC = 0.628), and primary infertility (AUC = 0.687). No significant interactions were found for other factors.

Discussion

This study developed and externally validated a three-stage sequential ensemble model for live-birth prediction in fresh IVF/ICSI cycles. Among ten base learners and three ensemble strategies, the SES ensemble demonstrated robust discrimination, calibration, and clinical net benefit across internal and external cohorts. The largest predictive improvement occurred after incorporating ovarian-stimulation-related variables at Stage 2, and SHAP analysis identified female age and transferred high-quality embryos as top predictive features. Live birth after IVF/ICSI depends on sustained implantation, factors including embryo viability, euploidy, placental function, endometrial receptivity, immune and obstetric factors, representing a more clinically meaningful endpoint than clinical pregnancy ( 11 , 12 ). Although many machine learning models predict clinical pregnancy ( 13 , 14 ), fewer focus on live birth. Live birth is a complex multifactorial long-term outcome affected by many maternal variables not recorded in routine clinical data, inherently limiting predictive accuracy and resulting in moderate AUC values. Even so, our staged ensemble model achieved comparable discrimination to previously reported live birth prediction models ( 7 , 15 ). Our three-stage framework mirrors real clinical decision-making: initial counseling based on baseline characteristics, mid-cycle decisions guided by ovarian response, and embryo transfer planning informed by embryological data. Adding stimulation parameters in Stage 2 significantly improved performance over Stage 1, reflecting the incremental value of ovarian response data. Adding embryological data in Stage 3 did not statistically improve prediction. One possible explanation is that ovarian response and endocrine profiles may already reflect key aspects of embryo competence ( 16 ). Another is that morphological grading is an imperfect surrogate for euploidy, such that high-quality embryos do not always result in live birth. Consequently, the limited AUC gain does not negate the clinical importance of embryological data, which remains valuable for post-transfer refinement rather than overall discrimination. Moreover, embryological features may yield individual-level risk re-stratification before embryo transfer within this modelling framework, though it was only observed in the test cohort and requires further external confirmation. Our three-stage framework is designed to deliver probability estimates at sequential clinical checkpoints, and Stage 3 corresponds specifically to the time point immediately prior to embryo transfer. Future integration of additional embryo-related biomarkers such as time-lapse morphokinetic parameters or preimplantation genetic testing for aneuploidy (PGT-A) may further boost predictive performance. Our model incorporates a comprehensive panel of demographic, clinical, endocrine, and embryological variables, including male factors that are frequently underreported in other prediction models ( 17 , 18 ). To minimize overfitting and ensure robust predictor selection, we applied a stringent multi-algorithm voting strategy combined with a clinical whitelist. Only variables supported by at least two algorithms or considered clinically indispensable were retained. This multi-modal selection approach differs from the single-method dimensionality-reduction strategies. Ensemble learning reduces prediction variance and improves stability by integrating complementary strengths of multiple base learners, thereby better capturing nonlinear biological interactions and enhancing external generalizability in IVF prediction ( 19 ). Compared with existing models, our study has notable strengths: a three-stage sequential design, two-center external validation, systematic ensemble optimization, and comprehensive interpretability analyses (SHAP and RCS). These features collectively enhance model transparency, generalizability, and clinical applicability. Among all ensemble strategies, SES yielded the highest AUC at the embryo development stage in both internal test and external validation cohorts while maintaining stable performance across all clinical stages. It also achieved the optimal PR-AUC and the lowest Brier score in external validation, indicating superior predictive precision and reliable probability calibration. SES also exhibits favorable feasibility for future translation given its relatively concise ensemble structure. SHAP analysis identified female age ranked highest across all stages, consistent with previous studies ( 20 , 21 ). At baseline, male factors including male age, sperm DFI and HDS also showed strong predictive importance, supporting the value of joint couple assessment before treatment ( 22 – 24 ). In the stimulation phase, down-regulation dose, trigger-day progesterone and EMT were key predictors, consistent with known drivers of implantation ( 25 , 26 ). RCS revealed the nonlinear relationships: EMT reached a plateau at 12 mm, with no additional benefit beyond this threshold. In the embryo development stage, number of transferred high-quality embryos and total high-quality blastocysts dominated prediction, confirming embryo quality as a strong predictor of live birth ( 27 ). SHAP dependence plots also reflected age-related heterogeneity: elevated progesterone exerted stronger adverse impacts on younger women, while higher EMT provided less benefit in older patients. Sufficient oocyte yield and good blastocyst quality benefited older women more, and intact ovarian reserve could partly offset the negative impact of advanced age. A key clinical implication is that poor endometrial condition or elevated progesterone cannot be compensated by transferring more high-quality embryos. Moderate oocyte and blastocyst numbers were sufficient for optimal outcomes, supporting a quality-over-quantity approach in clinical stimulation ( 28 ). Subgroup analysis defined the model’s applicable population, showing better performance in patients with advanced age, endometriosis and DOR, and relatively lower efficacy in those with PCOS. This study had several limitations. First, its retrospective design carried inherent risks of selection bias and unmeasured confounding. Although we used data from two independent centers, prospective and multi-center studies are needed to improve generalizability, its performance in other ethnic groups requires further evaluation. Clinically actionable probability thresholds for individual-patient management remain to be validated in prospective cohorts. Second, preimplantation genetic testing and frozen embryo transfer cycles were excluded, limiting applicability to these settings. Third, lifestyle, psychological factors, and reproductive immune indicators were not available and could not be included. Fourth, we only adopted binary live birth outcome as the endpoint, without distinguishing biochemical pregnancy, early miscarriage and clinical pregnancy as separate predictive endpoints. Fifth, obstetric factors affecting ongoing pregnancy were not included, and integrating such data may further improve predictive performance.

Conclusions

This validated three-stage prediction model enables dynamic, stage-specific live birth assessment in fresh IVF/ICSI cycles that proceed to embryo transfer. Female age and number of transferred high-quality embryos were the top-ranked predictors in our model, with embryo quality showing greater predictive weight for live-birth probability than oocyte or blastocyst yield. This work may inform personalized patient counselling, though prospective validation of clinical decision thresholds is required before real-world clinical application.

Text is read by the "Ask this paper" AI Q&A widget below. Extraction quality varies by source — PMC NXML preserves structure cleanly, OA-HTML may include some navigation residue, and OA-PDF can have broken hyphenation. The publisher copy is the canonical version.

My notes (saved in your browser only)

⚙ Ask this paper AI returns verbatim quotes from the full text · source: pmc-nxml ⓘ

Answers must be backed by verbatim quotes from this paper's full text. Hallucinated quotes are dropped automatically; if no verbatim passage answers the question, we say so. How this works

Citation neighborhood (no data yet)

We don't have any in-corpus citations linked to this paper yet. The paper's references may be in our DB but unresolved to ``paper_id`` (resolution happens at ingest when the cited DOI matches a row we already have). Run the cross-source citation reconcile pass to retry.

Source provenance

europepmc
last seen: 2026-10-11T09:27:45.537177+00:00
License: CC-BY-4.0 · commercial use OK · attribution required
Per Europe PMC