Distance-Aware Non-IID Federated Learning for Generalization and Personalization in Medical Imaging Segmentation - Archive ouverte HAL Accéder directement au contenu
Communication Dans Un Congrès Année : 2024

Distance-Aware Non-IID Federated Learning for Generalization and Personalization in Medical Imaging Segmentation

Résumé

Federated learning (FL) in healthcare suffers from non-identically distributed (non-IID) data, impacting model convergence and performance. While existing solutions for the non-IID problem often do not quantify the degree of non-IID nature between clients in the federation, assessing it can improve training experiences and outcomes, particularly in real-world scenarios with unfamiliar datasets. The paper presents a practical non-IID assessment methodology for a medical segmentation problem, highlighting its significance in medical FL. We propose a simple yet effective solution that utilizes distance measurements in the embedding space of medical images and statistical measurements calculated over their metadata. Our method, designed for medical imaging and integrated into federated averaging, improves model generalization by downgrading the contribution from the most distant client, treating it as an outlier. Additionally, it enhances model personalization by introducing distance-based clustering of clients. To the best of our knowledge, this method is the first to use distance-based techniques for providing a practical solution to the non-IID problem within the medical imaging FL domain. Furthermore, we validate our approach on three public FL imaging radiology datasets (FeTS, Prostate, and Fed-KITS2019) to demonstrate its effectiveness across various radiology imaging scenarios.
Fichier principal
Vignette du fichier
islandora_171528.pdf (692.1 Ko) Télécharger le fichier
Origine Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-04673203 , version 1 (19-08-2024)

Identifiants

  • HAL Id : hal-04673203 , version 1

Citer

Iuliia Alekseenko, Alexandros Karargyris, Nicolas Padoy. Distance-Aware Non-IID Federated Learning for Generalization and Personalization in Medical Imaging Segmentation. MIDL 2024, Medical Imaging with Deep Learning, Paris, France, 03-05 juillet 2024, Jul 2024, Paris, France. ⟨hal-04673203⟩
0 Consultations
0 Téléchargements

Partager

Gmail Mastodon Facebook X LinkedIn More