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).