Application of artificial intelligence algorithms to estimate the success rate in medically assisted procreation

Abstract

The aim of this study was to build an Artificial Neural Network (ANN) complemented by a decision tree to predict the chance of live birth after an In Vitro Fertilization (IVF)/Intracytoplasmic Sperm Injection (ICSI) treatment, before the first embryo transfer, using demographic and clinical data. Overall, 26 demographic and clinical data from 1193 cycles who underwent an IVF/ICSI treatment at Centro de Infertilidade e Reprodução Medicamente Assistida, between 2012 and 2019, were analyzed. An ANN was constructed by selecting experimentally the input variables which most correlated to the target through Pearson correlation. The final used variables were: woman’s age, total dose of gonadotropin, number of eggs, number of embryos and Antral Follicle Count (AFC). A decision tree was developed considering as an initial set the input variables integrated in the previous model. The ANN model was validated by the holdout method and the decision tree model by the 10-fold cross method. The ANN accuracy was 75.0% and the Area Under the Receiver Operating Characteristic (AUROC) curve was 75.2% (95% Confidence Interval (CI): 72.5–77.5%), whereas the decision tree model reached 75.0% and 74.9% (95% CI: 72.3–77.5%). These results demonstrated that both ANN and decision tree methods are fair for prediction the chance of conceive after an IVF/ICSI cycle.