A hybrid feature selection algorithm to determine effective factors in predictive model of success rate for in vitro fertilization/intracytoplasmic sperm injection treatment: A cross-sectional study

Background: Previous research has identified key factors affecting in vitro fertilization or intracytoplasmic sperm injection success, yet the lack of a standardized approach for various treatments remains a challenge. Objective: The objective of this study is to utilize a machine learning approach to identify the principal predictors of success in in vitro fertilization and intracytoplasmic sperm injection treatments. Materials and Methods: We collected data from ۷۳۴ individuals at ۲ infertility centers in Mashhad, Iran between November ۲۰۱۶ and March ۲۰۱۷. We employed feature selection methods to reduce dimensionality in a random forest model, guided by hesitant fuzzy sets (HFSs). A hybrid approach enhanced predictor identification and accuracy (ACC), as assessed using machine learning metrics such as matthew’s correlation coefficient, runtime, ACC, area under the receiver operating characteristic curve, precision or positive predictive value, recall, and F-Score, demonstrating the effectiveness of combining feature selection methods. Results: Our hybrid feature selection method excelled with the highest ACC (۰.۷۹۵), area under the receiver operating characteristic curve (۰.۷۲), and F-Score (۰.۸), while selecting only ۷ features. These included follicle-stimulation hormone (FSH), ۱۶Cells, FAge, oocytes, quality of transferred embryos (GIII), compact, and unsuccessful. Conclusion: We introduced HFSs in our novel method to select influential features for predicting infertility success rates. Using a multi-center dataset, HFSs improved feature selection by reducing the number of features based on standard deviation among criteria. Results showed significant differences between pregnant and non-pregnant groups for selected features, including FSH, FAge, ۱۶Cells, oocytes, GIII, and compact. We also found a significant correlation between FAge and fetal heart rate and clinical pregnancy rate, with the highest FSH level (۳۱.۸۷%) observed for doses ranging from ۱۰-۱۳ (mIU/ml).

کلیدواژه ها:

Machine learning ، Feature selection ، Infertility treatment ، Hesitant fuzzy set. ، یادگیری ماشین ، انتخاب ویژگی ، درمان ناباروری ، مجموعه های فازی مردد.