Machine learning for predicting elective fertility preservation outcomes

Abstract

This retrospective study applied machine-learning models to predict treatment outcomes of women undergoing elective fertility preservation. Two-hundred-fifty women who underwent elective fertility preservation at a tertiary center, 2019–2022 were included. Primary outcome was the number of metaphase II oocytes retrieved. Outcome class was based on oocyte count (OC): Low (≤ 8), Medium (9–15) or High (≥ 16). Machine-learning models and statistical regression were used to predict outcome class, first based on pre-treatment parameters, and then using post-treatment data from ovulation-triggering day. OC was 136 Low, 80 Medium, and 34 High. Random Forest Classifier (RFC) was the most accurate model (pre-treatment receiver operating characteristic (ROC) area under the curve (AUC) was 77%, and post-treatment ROC AUC was 87%), followed by XGBoost Classifier (pre-treatment ROC AUC 74%, post-treatment ROC AUC 86%). The most important pre-treatment parameters for RFC were basal FSH (22.6%), basal LH (19.1%), AFC (18.2%), and basal estradiol (15.6%). Post-treatment parameters were estradiol levels on trigger-day (17.7%), basal FSH (11%), basal LH (9%), and AFC (8%). Machine-learning models trained with clinical data appear to predict fertility preservation treatment outcomes with relatively high accuracy.