Use of federated learning to develop an artificial intelligence model predicting usable blastocyst formation from pre-ICSI oocyte images

Abstract

Research question: Can federated learning be used to develop an artificial intelligence (AI) model for evaluating oocyte competence using two-dimensional images of denuded oocytes in metaphase II prior to intracytoplasmic sperm injection (ICSI)?

Results: The oocyte AI model demonstrated area under the curve (AUC) up to 0.65 on two blind test datasets. High sensitivity for predicting competent oocytes (83-88%) was offset by lower specificity (26-36%). Exclusion of confounding biological variables (male factor infertility and maternal age ≥35 years) improved AUC up to 14%, primarily due to increased specificity. AI score correlated with size of the zona pellucida and perivitelline space, and ooplasm appearance. AI score also correlated with blastocyst expansion grade and morphological quality. The sum of AI scores from oocytes in group culture images predicted the formation of two or more usable blastocysts (AUC 0.77).

Conclusion: An AI model to evaluate oocyte competence was developed using federated learning, representing an essential step in protecting patient data. The AI model was significantly predictive of oocyte competence, as defined by usable blastocyst formation, which is a critical factor for IVF success. Potential clinical utility ranges from selective oocyte fertilization to guiding treatment decisions regarding additional rounds of oocyte retrieval.

Design: In total, 10,677 oocyte images with associated metadata were collected prospectively by eight IVF clinics across six countries. AI training used federated learning, where data were retained on regional servers to comply with data privacy laws. The final AI model required a single image as input to evaluate oocyte competence, which was defined by the formation of a usable blastocyst (≥expansion grade 3 by day 5 or 6 post ICSI).

Keywords: Artificial intelligence; Computer vision; Embryo development; IVF; Medical imaging; Oocyte assessment.