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

Zohar Barnett-Itzhaki, Miriam Elbaz, Rachely Butterman, Devora Amar, Moshe Amitay, Catherine Racowsky, Raoul Orvieto, Russ Hauser, Andrea A. Baccarelli, Ronit Machtinger

Research output: Contribution to journalArticlepeer-review

4 Scopus citations

Abstract

Purpose: To assess whether machine learning methods provide advantage over classic statistical modeling for the prediction of IVF outcomes. Methods: 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. Results: 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. Conclusions: 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.

Original languageEnglish
Pages (from-to)2405-2412
Number of pages8
JournalJournal of Assisted Reproduction and Genetics
Volume37
Issue number10
DOIs
StatePublished - 1 Oct 2020
Externally publishedYes

Keywords

  • IVF
  • Implantation
  • Machine learning
  • Oocytes
  • Prediction models

Fingerprint

Dive into the research topics of 'Machine learning vs. classic statistics for the prediction of IVF outcomes'. Together they form a unique fingerprint.

Cite this