Abstract:Background: This study sought to provide machine learning-based classification models to predict the success of intrauterine insemination (IUI) therapy. Additionally, we sought to illustrate the effect of models fitting with balanced data vs original data with imbalanced data labels using two different types of resampling methods. Finally, we fit models with all features against optimized feature sets using various feature selection techniques. Methods: The data for the cross-sectional study were collected from 546 infertile couples with IUI at the Fatemehzahra Infertility Research Center, Babol, North of Iran. Logistic regression (LR), support vector classification, random forest, Extreme Gradient Boosting (XGBoost) and, Stacking generalization (Stack) as the machine learning classifiers were used to predict IUI success by Python v3.7. We employed the Smote-Tomek (Stomek) and Smote-ENN (SENN) resampling methods to address the imbalance problem in the original dataset. Furthermore, to increase the performance of the models, mutual information classification (MIC-FS), genetic algorithm (GA-FS), and random forest (RF-FS) were used to select the ideal feature sets for model development. Results: In this study, 28% of patients undergoing IUI treatment obtained a successful pregnancy. Also, the average age of women and men was 24.98 and 29.85 years, respectively. The calibration plot in this study for IUI success prediction by machine learning models showed that between feature selection methods, the RF-FS, and among the datasets used to fit the models, the balanced dataset with the Stomek method had well-calibrating predictions than other methods. Finally, the brier scores for the LR, SVC, RF, XGBoost, and Stack models that were fitted utilizing the Stomek dataset and the chosen feature set using the Random Forest technique obtained equal to 0.202, 0.183, 0.158, 0.129, and 0.134, respectively. It showed duration of infertility, male and female age, sperm concentration, and sperm motility grading score as the most predictable factors in IUI success. Conclusion: The results of this study with the XGBoost prediction model can be used to foretell the individual success of IUI for each couple before initiating therapy.

Explainable AI to Predict Male Fertility Using Extreme Gradient Boosting Algorithm with SMOTE

Development of a Model Predicting the Outcome of in Vitro Fertilization Cycles by a Robust Decision Tree Method

Explainable artificial intelligence models for predicting pregnancy termination among reproductive-aged women in six east African countries: machine learning approach

Using feature optimization and LightGBM algorithm to predict the clinical pregnancy outcomes after in vitro fertilization

Beyond black-box models: explainable AI for embryo ploidy prediction and patient-centric consultation

Interpretable machine learning models for predicting clinical pregnancies associated with surgical sperm retrieval from testes of different etiologies: a retrospective study

Predicting Male Infertility Using Artificial Neural Networks: A Review of the Literature

Protein deep profile and model predictions for identifying the causal genes of male infertility based on deep learning

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

Fertility-LightGBM: A fertility-related protein prediction model by multi-information fusion and light gradient boosting machine

Discovering mechanisms underlying medical AI prediction of protected attributes

Developing machine learning-based models to predict intrauterine insemination (IUI) success by address modeling challenges in imbalanced data and providing modification solutions for them

Machine Learning-Based Analysis of Sperm Videos and Participant Data for Male Fertility Prediction

Interpretable Predictive Models to Understand Risk Factors for Maternal and Fetal Outcomes

Machine learning predicts live-birth occurrence before in-vitro fertilization treatment

The Application of Artificial Intelligence in Predicting Embryo Transfer Outcome of Recurrent Implantation Failure

Diabetes prediction using machine learning and explainable AI techniques

Artificial intelligence model to predict pregnancy and multiple pregnancy risk following in vitro fertilization-embryo transfer (IVF-ET)

Predictive models for live birth outcomes of FET: Improving clinical decision-making based on machine learning analysis

Using Explainable AI to Cross-Validate Socio-economic Disparities Among Covid-19 Patient Mortality

AI-Based Optimal Treatment Strategy Selection for Female Infertility for First and Subsequent IVF-ET Cycles