Federated brain tumor segmentation: an extensive benchmark
Résumé
Recently, federated learning has raised increasing interest in the medical image analysis field due to its ability to aggregate multi-center data with privacy-preserving properties. A large amount of federated training schemes have been published, which we categorize into global (one final model), personalized (one model per institution) or hybrid (one model per cluster of institutions) methods. However, their applicability on the recently published Federated Brain Tumor Segmentation 2022 dataset has not been explored yet. We propose an extensive benchmark of federated learning algorithms from all three classes on this task. While standard FedAvg already performs very well, we show that some methods from each category can bring a slight performance improvement and potentially limit the final model(s) bias toward the predominant data distribution of the federation. Moreover, we provide a deeper understanding of the behaviour of federated learning on this task through alternative ways of distributing the pooled dataset among institutions, namely an Independent and Identical Distributed (IID) setup, and a limited data setup.
Mots clés
Deep learning Federated learning Medical image segmentation BraTS Personalized federated learning Clustered federated learning
Deep learning
Federated learning
Medical image segmentation
BraTS
Personalized federated learning
Clustered federated learning
Deep learning Federated learning Medical image segmentation BraTS Personalized federated learning Clustered federated learning
Deep learning Federated learning Medical image segmentation BraTS Personalized federated learning Clustered federated learning
Origine | Fichiers produits par l'(les) auteur(s) |
---|