Hybrid Deep Learning Model for IVF Outcome Prediction from Time-Series Hormonal Data

Document Type : Research Paper

Authors

1 Research scholar, School of Computer Science and Engineering, REVA University, Bengaluru, Karnataka, India.

2 Associate Professor, School of Computer Science and Engineering, REVA University, Bengaluru, Karnataka, India.

10.22059/jitm.2026.106258

Abstract

Optimizing results in assisted reproduction requires tailoring the dosage of follicle-stimulating hormone (FSH) during controlled ovarian stimulation (COS), but this is still challenging because of patient variability. Real-time modifications during stimulation are not supported by the majority of current models, which are restricted to static forecasts of starting dosages. This study proposes an advanced AI-driven framework for forecasting hormone dynamics and improving reproductive outcome prediction in IVF cycles. The methodology integrates multi-source clinical data with high-resolution time-series hormone profiles collected from Day 1 to Day 30 of ovarian stimulation. Data pre-processing includes normalization of hormonal values, alignment of temporal and static clinical attributes, and creation of patient-level merged datasets. Feature engineering incorporates daily hormone variations, moving averages, peak detection, and wavelet-based temporal pattern extraction, alongside encoded and normalized clinical parameters. For hormone trend forecasting, a hybrid deep learning model is developed that combines Wavelet Transform for noise reduction with LSTM and Transformer layers for sequential representation learning. The architecture captures short-term hormone fluctuations and long-range temporal dependencies, enabling accurate next-day hormone prediction. Model performance is optimized using RMSE = 0.31, MAE = 0.22, and MAPE loss metrics. This integrated approach enhances predictive accuracy = 88.9%, facilitates early-cycle monitoring, and supports clinical decision-making in assisted reproductive treatments.

Keywords


Bai, R., Li, J. W., Hong, X., Xuan, X. Y., Li, X. H., & Tuo, Y. (2025). Predictive modeling of pregnancy outcomes utilizing multiple machine learning techniques for in vitro fertilization-embryo transfer. BMC Pregnancy and Childbirth25(1), 316.
Biacchiardi, C. P., Revelli, A., Gennarelli, G., Rustichelli, S., Moffa, F., & Massobrio, M. (2004). Fallopian tube sperm perfusion versus intrauterine insemination in unexplained infertility: a randomized, prospective, cross-over trial. Fertility and sterility81(2), 448-451.
Blank, C., Wildeboer, R. R., DeCroo, I., Tilleman, K., Weyers, B., De Sutter, P., ... & Schoot, B. C. (2019). Prediction of implantation after blastocyst transfer in in vitro fertilization: a machine-learning perspective. Fertility and sterility111(2), 318-326.
CN Barreto, N., Castro, G. Z., Pereira, R. G., Pereira, F. A., Reis, F. M., Junior, W. M., ... & Gomes, K. B. (2022). Predicting in vitro fertilization success in the Brazilian public health system: a machine learning approach. Medical & Biological Engineering & Computing60(7), 1851-1861.
De Geyter, C., Calhaz-Jorge, C., Kupka, M. S., Wyns, C., Mocanu, E., Motrenko, T., ... & Goossens, V. (2018). ART in Europe, 2014: results generated from European registries by ESHRE: The European IVF-monitoring Consortium (EIM) for the European Society of Human Reproduction and Embryology (ESHRE). Human reproduction33(9), 1586-1601.
Deshpande, P. S., & Gupta, A. S. (2019). Causes and prevalence of factors causing infertility in a public health facility. Journal of human reproductive sciences12(4), 287-293.
Goyal, A., Kuchana, M., & Ayyagari, K. P. R. (2020). Machine learning predicts live-birth occurrence before in-vitro fertilization treatment. Scientific reports10(1), 20925.
Graham, M. E., Jelin, A., Hoon Jr, A. H., Wilms Floet, A. M., Levey, E., & Graham, E. M. (2023). Assisted reproductive technology: Short‐and long‐term outcomes. Developmental Medicine & Child Neurology65(1), 38-49.
Güvenir, H. A., Misirli, G., Dilbaz, S., Ozdegirmenci, O., Demir, B., & Dilbaz, B. (2015). Estimating the chance of success in IVF treatment using a ranking algorithm. Medical & biological engineering & computing53(9), 911-920.
Handayani, N., Louis, C. M., Erwin, A., Aprilliana, T., Polim, A. A., Sirait, B., ... & Sini, I. (2022). Machine learning approach to predict clinical pregnancy potential in women undergoing IVF program. Fertility & Reproduction4(02), 77-87.
Jiahui Qiu et al (2019), Personalized prediction of live birth prior to the first in vitro fertilization treatment: A machine learning method, Qiu et al. J Transl Med https://doi.org/10.1186/s12967-019-2062-5
Khosravi, P., Kazemi, E., Zhan, Q., Malmsten, J. E., Toschi, M., Zisimopoulos, P., ... & Hajirasouliha, I. (2019). Deep learning enables robust assessment and selection of human blastocysts after in vitro fertilization. NPJ digital medicine2(1), 21.
Klitzman, R. (2017). Unconventional combinations of prospective parents: ethical challenges faced by IVF providers. BMC Medical Ethics18(1), 18.
Mohammad Reza Nateghi (2024) The use of Machine Learning for Human Oocyte selection and Success Rate in IVF Methods, Sarem Journal of Medical Research. Volume 9, Issue 2, Summer 2024: 93-104.
Qiu, J., Li, P., Dong, M., Xin, X., & Tan, J. (2019). Personalized prediction of live birth prior to the first in vitro fertilization treatment: a machine learning method. Journal of translational medicine17(1), 317.
Sarkar, S., & Gupta, P. (2016). Socio-demographic correlates of women’s infertility and treatment seeking behavior in India. Journal of reproduction & infertility17(2), 123.
Schlegel, P. N., Sigman, M., Collura, B., De Jonge, C. J., Eisenberg, M. L., & Lamb, D. J. (2020). Diagnosis and treatment of infertility in men: AUA/ASRM guideline. Linthicum (MD): American Urology Association.
Siristatidis, C., Stavros, S., Drakeley, A., Bettocchi, S., Pouliakis, A., Drakakis, P., ... & Vlahos, N. (2021). Omics and artificial intelligence to improve in vitro fertilization (IVF) success: a proposed protocol. Diagnostics11(5), 743.
Wang, C. W., Kuo, C. Y., Chen, C. H., Hsieh, Y. H., & Su, E. C. Y. (2022). Predicting clinical pregnancy using clinical features and machine learning algorithms in in vitro fertilization. Plos one17(6), e0267554.
World Health Organization. Available at: https://www.who.int/news-room/fact-sheets/detail/infertility. Access Date: May, 2023
Yao, M. W., Nguyen, E. T., Retzloff, M. G., Gago, L. A., Copland, S., Nichols, J. E., ... & Walmer, D. K. (2024). Improving IVF utilization with patient-centric artificial intelligence-machine learning (AI/ML): a retrospective multicenter experience. Journal of clinical medicine13(12), 3560.
Zhu, S., Huang, Z., Chen, X., Jiang, W., Zhou, Y., Zheng, B., & Sun, Y. (2025). Construction and evaluation of machine learning-based prediction model for live birth following fresh embryo transfer in IVF/ICSI patients with polycystic ovary syndrome. Journal of Ovarian Research18(1), 70.