Predicting clinical pregnancy using clinical features and machine learning algorithms in in vitro fertilization

Abstract

Introduction Assisted reproductive technology has been proposed for women with infertility. Moreover, in vitro fertilization (IVF) cycles are increasing. Factors contributing to successful pregnancy have been widely explored. In this study, we used machine learning algorithms to construct prediction models for clinical pregnancies in IVF. Materials and methods A total of 24,730 patients entered IVF and intracytoplasmic sperm injection cycles with clinical pregnancy outcomes at Taipei Medical University Hospital. Data used included patient characteristics and treatment. We used machine learning methods to develop prediction models for clinical pregnancy and explored how each variable affects the outcome of interest using partial dependence plots. Results Experimental results showed that the random forest algorithm outperforms logistic regression in terms of areas under the receiver operating characteristics curve. The ovarian stimulation protocol is the most important factor affecting pregnancy outcomes. Long and ultra-long protocols have shown positive effects on clinical pregnancy among all protocols. Furthermore, total frozen and transferred embryos are positive for a clinical pregnancy, but female age and duration of infertility have negative effects on clinical pregnancy. Conclusion Our findings show the importance of variables and propensity of each variable by random forest algorithm for clinical pregnancy in the assisted reproductive technology cycle. This study provides a ranking of variables affecting clinical pregnancy and explores the effects of each treatment on successful pregnancy. Our study has the potential to help clinicians evaluate the success of IVF in patients.

Machine Learning to Predict Pregnancy in Dairy Cows: An Approach Integrating Automated Activity Monitoring and On-Farm Data

“…Statistical models have been used to predict lifetime production [ 12 ], fertility [ 13 , 14 ], health [ 15 ], and genomic selection [ 16 ]. Machine learning models are an alternative approach to classical statistical models for developing predictive models in large datasets, such as livestock-related studies [ 17 ], allowing for proactive management decisions and the customization of approaches to suit specific farm conditions [ 18 ].…”

A novel machine learning model for predicting clinical pregnancy after laparoscopic tubal anastomosis

“…Machine learning has been increasingly used in medical practice due to its potential to improve clinical decision-making and patient outcomes. The application of machine learning methods has improved artificial intelligence and has been used in clinical prediction [ 8 – 10 ]. LASSO regression is a machine learning technique for linear regression that employs L1 regularization to build models and select variables.…”