Federated Multi-centric Image Segmentation with Uneven Label Distribution
- Others:
- Eurecom [Sophia Antipolis]
- Università degli Studi di Siena = University of Siena (UNISI)
- University College of London [London] (UCL)
- Universitat Oberta de Catalunya [Barcelona] (UOC)
- E-Patient : Images, données & mOdèles pour la médeciNe numériquE (EPIONE) ; Inria Sophia Antipolis - Méditerranée (CRISAM) ; Institut National de Recherche en Informatique et en Automatique (Inria)-Institut National de Recherche en Informatique et en Automatique (Inria)
- Springer
- ANR-19-P3IA-0002,3IA@cote d'azur,3IA Côte d'Azur(2019)
- ANR-22-CE45-0015,I-VESSEG,Apprentissage Interactif et Collaboratif pour la Segmentation des Vaisseaux(2022)
Description
While federated learning is the state-of-the-art methodology for collaborative learning, its adoption for training segmentation models often relies on the assumption of uniform label distributions across participants, and is generally sensitive to the large variability of multi-centric imaging data. To overcome these issues, we propose a novel federated image segmentation approach adapted to complex non-iid setting typical of real-life conditions. We assume that labeled dataset is not available to all clients, and that clients data exhibit differences in distribution due to three factors: different scanners, imaging modalities and imaged organs. Our proposed framework collaboratively builds a multimodal data factory that embeds a shared, disentangled latent representation across participants. In a second asynchronous stage, this setup enables local domain adaptation without exchanging raw data or annotations, facilitating target segmentation. We evaluate our method across three distinct scenarios, including multi-scanner cardiac magnetic resonance segmentation, multi-modality skull stripping, and multi-organ vascular segmentation. The results obtained demonstrate the quality and robustness of our approach as compared to the state-of-the-art methods.
Abstract
International audience
Additional details
- URL
- https://hal.science/hal-04730834
- URN
- urn:oai:HAL:hal-04730834v1
- Origin repository
- UNICA