Leveraging federated learning for boosting data privacy and performance in IVF embryo selection

Chun I. Lee, Chii Ruey Tzeng, Monty Li, Hsing Hua Lai, Chi Huang Chen, Yulun Huang, T. Arthur Chang, Chien Hong Chen, Chun Chia Huang, Maw Sheng Lee, Mark Liu

Research output: Contribution to journalArticlepeer-review

Abstract

Purpose: To study the effectiveness of federated learning in in vitro fertilization on embryo evaluation tasks. Methods: This is a retrospective cohort analysis. Two datasets were used in this study. The ploidy status dataset consisted of 10,065 embryo records, 3760 treatments, and 2479 infertile couples from 5 hospitals. The clinical pregnancy dataset consisted of 4495 embryo records, 4495 treatments, and 3704 infertile couples from 4 hospitals. Federated learning and the gradient boosting decision tree algorithm were utilized for modeling. Results: On the ploidy status dataset, the areas under the receiver operating characteristic curves of our model trained with federated learning were 71.78%, 73.10%, 69.39%, 69.72%, and 73.46% for 5 hospitals respectively, showing an average increase of 2.5% compared to those of our model trained without federated learning. On the clinical pregnancy dataset, the areas under the receiver operating characteristic curves of our model trained with federated learning were 72.03%, 56.77%, 61.63%, and 58.58% for 4 hospitals respectively, showing an average increase of 3.08%. Conclusions: Federated learning can improve data privacy and data security and meanwhile improve the performance of embryo selection tasks by leveraging data from multiple sources. This study demonstrates the effectiveness of federated learning in embryo evaluation, and the results show the promise for future application.

Original languageEnglish (US)
JournalJournal of Assisted Reproduction and Genetics
DOIs
StateAccepted/In press - 2024
Externally publishedYes

Keywords

  • Clinical pregnancy
  • Federated learning
  • Gradient boosting decision tree
  • In vitro fertilization
  • Ploidy status

ASJC Scopus subject areas

  • Reproductive Medicine
  • Genetics
  • Obstetrics and Gynecology
  • Developmental Biology
  • Genetics(clinical)

Fingerprint

Dive into the research topics of 'Leveraging federated learning for boosting data privacy and performance in IVF embryo selection'. Together they form a unique fingerprint.

Cite this