Purpose <p>Can machine learning effectively predict the number of mature oocytes retrieved after controlled ovarian stimulation in patients undergoing in vitro fertilization or intracytoplasmic sperm injection?</p> Design <p>This retrospective study included data from 24,976 infertility patients treated between 2018 and 2022 at a single reproductive center. After data preprocessing, feature selection was performed using correlation analysis, LASSO regression, and recursive elimination based on random forest. Eight machine learning models were developed to predict the number of mature oocytes retrieved. Model performance was evaluated using root mean square error (RMSE), mean absolute error (MAE), and <i>R</i>-squared (<i>R</i><sup>2</sup>). Calibration was assessed using intercept and slope on the testing set. Clinical validation was conducted using an independent 2022 dataset. The best-performing model was interpreted using SHAP values. A web-based calculator was developed to support clinical application.</p> Results <p>Six predictors were identified: antral follicle count; follicle-stimulating hormone and estradiol at stimulation start; estradiol on the trigger day; number of large follicles on the trigger day; and age. The multilayer perceptron model achieved the highest accuracy, with RMSE of 3.675, MAE of 2.702, and <i>R</i><sup>2</sup> of 0.714 in clinical validation. Calibration analysis showed good agreement between predicted and observed values. SHAP interpretation revealed estradiol level and number of large follicles on the trigger day as the strongest predictors.</p> Conclusions <p>We developed a machine learning model that accurately predicts mature oocyte yield. A web-based calculator was also created to enable individualized prediction and enhance clinical usability.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Interpretable machine learning model for predicting MII oocyte retrieved following controlled ovarian stimulation: a retrospective cohort study of 24,976 IVF/ICSI cycles

  • Lu Zhang,
  • Li-Juan Fu,
  • Yu-Ling Hu,
  • Yu Hao,
  • Rui-Ling Liu,
  • Mei-Hua Bao,
  • Zhao-Hui Zhong,
  • Xing-Yu Lv,
  • Yu-Bin Ding

摘要

Purpose

Can machine learning effectively predict the number of mature oocytes retrieved after controlled ovarian stimulation in patients undergoing in vitro fertilization or intracytoplasmic sperm injection?

Design

This retrospective study included data from 24,976 infertility patients treated between 2018 and 2022 at a single reproductive center. After data preprocessing, feature selection was performed using correlation analysis, LASSO regression, and recursive elimination based on random forest. Eight machine learning models were developed to predict the number of mature oocytes retrieved. Model performance was evaluated using root mean square error (RMSE), mean absolute error (MAE), and R-squared (R2). Calibration was assessed using intercept and slope on the testing set. Clinical validation was conducted using an independent 2022 dataset. The best-performing model was interpreted using SHAP values. A web-based calculator was developed to support clinical application.

Results

Six predictors were identified: antral follicle count; follicle-stimulating hormone and estradiol at stimulation start; estradiol on the trigger day; number of large follicles on the trigger day; and age. The multilayer perceptron model achieved the highest accuracy, with RMSE of 3.675, MAE of 2.702, and R2 of 0.714 in clinical validation. Calibration analysis showed good agreement between predicted and observed values. SHAP interpretation revealed estradiol level and number of large follicles on the trigger day as the strongest predictors.

Conclusions

We developed a machine learning model that accurately predicts mature oocyte yield. A web-based calculator was also created to enable individualized prediction and enhance clinical usability.