Machine learning vs. classic statistics for the prediction of IVF outcomes.

Abstract

To assess whether machine learning methods provide advantage over classic statistical modeling for the prediction of IVF outcomes. The study population consisted of 136 women undergoing a fresh IVF cycle from January 2014 to August 2016 at a tertiary, university-affiliated medical center. We tested the ability of two machine learning algorithms, support vector machine (SVM) and artificial neural network (NN), vs. classic statistics (logistic regression) to predict IVF outcomes (number of oocytes retrieved, mature oocytes, top-quality embryos, positive beta-hCG, clinical pregnancies, and live births) based on age and BMI, with or without clinical data. Machine learning algorithms (SVM and NN) based on age, BMI, and clinical features yielded better performances in predicting number of oocytes retrieved, mature oocytes, fertilized oocytes, top-quality embryos, positive beta-hCG, clinical pregnancies, and live births, compared with logistic regression models. While accuracies were 0.69 to 0.9 and 0.45 to 0.77 for NN and SVM, respectively, they were 0.34 to 0.74 using logistic regression models. Our findings suggest that machine learning algorithms based on age, BMI, and clinical data have an advantage over logistic regression for the prediction of IVF outcomes and therefore can assist fertility specialists' counselling and their patients in adjusting the appropriate treatment strategy.