Comparative study of machine learning approaches integrated with genetic algorithm for IVF success prediction

Shirin Dehghan, Reza Rabiei, Hamid Choobineh, Keivan Maghooli, Mozhdeh Nazari and Mojtaba Vahidi-Asl

PLOS ONE, 2024, vol. 19, issue 10, 1-19

Abstract: Introduction: IVF is a widely-used assisted reproductive technology with a consistent success rate of around 30%, and improving this rate is crucial due to emotional, financial, and health-related implications for infertile couples. This study aimed to develop a model for predicting IVF outcome by comparing five machine-learning techniques. Method: The research approached five prominent machine learning algorithms, including Random Forest, Artificial Neural Network (ANN), Support Vector Machine (SVM), Recursive Partitioning and Regression Trees (RPART), and AdaBoost, in the context of IVF success prediction. The study also incorporated GA as a feature selection method to enhance the predictive models’ robustness. Results: Findings demonstrate that AdaBoost, particularly when combined with GA feature selection, achieved the highest accuracy rate of 89.8%. Using GA, Random Forest also demonstrated strong performance, achieving an accuracy rate of 87.4%. Genetic Algorithm significantly improved the performance of all classifiers, emphasizing the importance of feature selection. Ten crucial features, including female age, AMH, endometrial thickness, sperm count, and various indicators of oocyte and embryo quality, were identified as key determinants of IVF success. Conclusion: These findings underscore the potential of machine learning and feature selection techniques to assist IVF clinicians in providing more accurate predictions, enabling tailored treatment plans for each patient. Future research and validation can further enhance the practicality and reliability of these predictive models in clinical IVF practice.

Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0310829 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 10829&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0310829

DOI: 10.1371/journal.pone.0310829

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().