Background: The low live birth rate and difficult decision-making of the in vitro fertilization (IVF) treatment regimen bring great trouble to patients and clinicians. Based on the retrospective clinical data of patients undergoing the IVF cycle, this study aims to establish classification models for predicting live birth outcome (LBO) with machine learning methods.
Methods: The historical data of a total of 1405 patients undergoing IVF cycle were first collected and then analyzed by univariate and multivariate analysis. The statistically significant factors were identified and taken as input to build the artificial neural network (ANN) model and supporting vector machine (SVM) model for predicting the LBO. By comparing the model performance, the one with better results was selected as the final prediction model and applied in real clinical applications.
Results: Univariate and multivariate analysis shows that 7 factors were closely related to the LBO (with P < 0.05): Age, ovarian sensitivity index (OSI), controlled ovarian stimulation (COS) treatment regimen, Gn starting dose, endometrial thickness on human chorionic gonadotrophin (HCG) day, Progesterone (P) value on HCG day, and embryo transfer strategy. By taking the 7 factors as input, the ANN-based and SVM-based LBO models were established, yielding good prediction performance. Compared with the ANN model, the SVM model performs much better and was selected as the final model for the LBO prediction. In real clinical applications, the proposed ANN-based LBO model can predict the LBO with good performance and recommend the embryo transfer strategy of potential good LBO.
Conclusions: The proposed model involving all essential IVF treatment factors can accurately predict LBO. It can provide objective and scientific assistance to clinicians for customizing the IVF treatment strategy like the embryo transfer strategy.