Multifactor Prediction of Embryo Transfer Outcomes Based on a Machine Learning Algorithm

Abstract

In vitro fertilization-embryo transfer (IVF-ET) technology make it possible for infertile couples to conceive a baby successfully. Nevertheless, IVF-ET does not guarantee success. Frozen embryo transfer (FET) is an important supplement to IVF-ET. Many factors are correlated with the outcome of FET which is unpredictable. Machine learning is a field of study that predict various outcomes by defining data attributes and using relevant data and calculation algorithms. Machine learning algorithm has been widely used in clinical research. The present study focuses on making predictions of early pregnancy outcomes in FET through clinical characters, including age, body mass index (BMI), endometrial thickness (EMT) on the day of progesterone treatment, good-quality embryo rate (GQR), and type of infertility (primary or secondary), serum estradiol level (E2) on the day of embryo transfer, and serum progesterone level (P) on the day of embryo transfer. We applied four representative machine learning algorithms, including logistic regression (LR), conditional inference tree, random forest (RF) and support vector machine (SVM) to build prediction models and identify the predictive factors. We found no significant difference among the models in the sensitivity, specificity, positive predictive rate, negative predictive rate or accuracy in predicting the pregnancy outcome of FET. For example, the positive/negative predictive rate of the SVM (gamma = 1, cost = 100, 10-fold cross validation) is 0.56 and 0.55. This approach could provide a reference for couples considering FET. The prediction accuracy of the present study is limited, which suggests that there may be some other more effective predictors to be developed in future work.