Federated task-adaptive learning for personalized selection of human IVF-derived embryos

Tianrun Gao, Yuning Yang, Kai Wang, Yuanxu Gao, Lishuang Ma, Lei Chen, Guangdong Liu, Ping Zhang, Xiaohong Liu, Guangyu Wang

Abstract: In-vitro fertilization (IVF) provides an effective infertility treatment. However, the success of IVF heavily depends on manual morphological assessment of embryos, a process that is both time-consuming and labor-intensive. While artificial intelligence (AI) enables automated assessment, its reliance on centralized large-scale data training raises privacy concerns. Here, we develop a distributed AI system, termed ‘FedEmbryo’, tailored for personalized embryo selection while preserving data privacy. Within FedEmbryo, we introduce a federated task-adaptive learning (FTAL) approach with a hierarchical dynamic weighting adaptation (HDWA) mechanism. The FTAL integrates multitask learning (MTL) with federated learning (FL) by proposing a unified multitask architecture that consists of shared layers and task-specific layers to accommodate the single- and multi-task learning within each client. The HDWA mechanism considers the learning feedback (loss ratios) from the tasks and clients, facilitating a dynamic balance between task attention and client aggregation. We conduct extensive experiments in different scenarios of the IVF cycle to evaluate the effectiveness of FedEmbryo for personalized embryo selection. The observer study validates that FedEmbryo achieves superior performance in both morphological valuation and prediction of live-birth outcomes compared to the locally trained model, as well as state-of-the-art FL methods. We present FedEmbryo, an AI-powered system designed to improve IVF outcomes through privacy-preserving, decentralized training across multiple clinical sites. FedEmbryo demonstrates superior performance in capturing stage-specific morphological features of embryos and achieves accurate predictions for key IVF-related tasks. We hope that FedEmbryo will serve as a practical tool for enhancing clinical decision-making in IVF. In-vitro fertilization (IVF) is one of the most widely used methods in assisted reproductive technology (ART). However, IVF success remains highly dependent on manual embryo assessment, a process that is time-consuming and labor-intensive. In this study, we introduce FedEmbryo, an AI system designed to improve IVF outcomes. Our experiments show that FedEmbryo provides more accurate evaluations of embryo quality, including morphological evaluation and live-birth outcomes prediction.This work highlights the potential of FedEmbryo to assist in reproductive healthcare. Gao et al. propose FedEmbryo, a federated task-adaptive learning (FTAL) approach that dynamically adjusts task and client contributions based on learning feedback, enabling privacy-preserving, decentralized training across multiple clinical sites to improve IVF outcomes. FedEmbryo demonstrates superior performance in both morphological evaluation and live-birth outcome prediction, thereby enhancing the accuracy of high-quality embryo selection.

External IDs:doi:10.1038/s43856-025-01182-1