Comparing machine learning approaches for predicting the success of ICSI treatment: A study on clinical applications

Abstract

Intracytoplasmic Sperm Injection (ICSI) is widely used to treat almost all forms of male infertility and to overcome fertilization failure. While ICSI is a powerful procedure, it's also considered quite expensive, which means couples and clinicians have to make informed decisions about whether or not to proceed with this treatment. About 10,036 patient records, 46 attribute sets, and one label column that indicates the success or failure of pregnancy after the ICSI treatment were used to conduct this research. The data were gathered from Razan infertility center in Palestine. The ICSI dataset contains only clinical features that are known prior to deciding on ICSI treatment. The dataset contains 46 features, 5 of the independent features have categorical values, 12 are numerical, 3 are string, and 26 are binary. Based on the results, RF algorithm achieved the highest AUC score of 0.97, followed by the NN with a score of 0.95, and the RIMARC algorithm with a score of 0.92. AUC is a widely used metric for evaluating the performance of binary classification models. Therefore, judging by the AUC scores, it appears that RF algorithm outperformed the other two algorithms in terms of the evaluated metric. The method employed in our analysis demonstrates considerable promise, practicality, and generalizability, driving advancements in fertility treatments and ultimately improving the chances of couples achieving their desired family goals.