FL-MedSegBench: A Comprehensive Benchmark for Federated Learning on Medical Image Segmentation
Organizations: Department of Electrical and Computer Engineering, The University of Hong Kong, Hong Kong, China · School of Communication Engineering, Hangzhou Dianzi University, Hang Zhou, China · Department of Radiation Oncology, Stanford University, California, USA · Department of Electronic Engineering, Chinese University of Hong Kong, Hong Kong, China
Abstract
Federated learning (FL) offers a privacy-preserving paradigm for collaborative medical image analysis without sharing raw data. However, the absence of standardized benchmarks for medical image segmentation hinders fair and comprehensive evaluation of FL methods. To address this gap, we introduce FL-MedSegBench, the first comprehensive benchmark for federated learning on medical image segmentation. Our benchmark encompasses nine segmentation tasks across ten imaging modalities, covering both 2D and 3D formats with realistic clinical heterogeneity. We systematically evaluate eight generic FL (gFL) and five personalized FL (pFL) methods across multiple dimensions: segmentation accuracy, fairness, communication efficiency, convergence behavior, and generalization to unseen domains. Extensive experiments reveal several key insights: (i) pFL methods, particularly those with client-specific batch normalization (e.g., FedBN), consistently outperform generic approaches; (ii) No single method universally dominates, with performance being dataset-dependent; (iii) Communication frequency analysis shows normalization-based personalization methods exhibit remarkable robustness to reduced communication frequency; (iv) Fairness evaluation identifies methods like Ditto and FedRDN that protect underperforming clients; (v) A method's generalization to unseen domains is strongly tied to its ability to perform well across participating clients. We will release an open-source toolkit to foster reproducible research and accelerate clinically applicable FL solutions, providing empirically grounded guidelines for real-world clinical deployment. The source code is available at https://github.com/meiluzhu/FL-MedSegBench.
Figures & tables
| Dataset | Index | Client Name | Sample size | Devices/Institutions |
| Fed-Vessel (Fundus) | C1 | chaseDB [ 16 ] | 28 | Nidek NM-200-D |
| C2 | DR-Hagis [ 22 ] | 40 | Canon CR DGi/Topcon TRC-NW6s/Topcon TRC-NW8 | |
| C3 | DRIVE [ 51 ] | 40 | Canon CR5 nonmydriatic 3CCD/TopCon TRV-50 | |
| C4 | HRF [ 42 ] | 45 | Canon CF–60 UVi | |
| C5 | LES-AV [ 44 ] | 22 | N/A | |
| C6 | ORVS [ 47 ] | 49 | Zeiss, Viscuam 200 |
| Dataset | Index | Client name | Sample size | Devices/Institutions |
| Fed-Pancreas (MRI-T1) | C1 | NYU | 162 | New York University Medical Center |
| C2 | MCF | 151 | Mayo Clinic Florida | |
| C3 | NU | 30 | Northwestern University | |
| C4 | AHN | 17 | Allegheny Health Network | |
| C5 | MCA | 25 | Mayo Clinic Arizona | |
| Fed-M&Ms (CMR) | C1 | Centre 1 | 95 | Vall d’Hebron Hospital, Spain / Siemens |
| Methods | C1 | C2 | C3 | C4 | C5 | C6 |
| Local | ||||||
| Generic FL methods | ||||||
| FedAvg | ||||||
| FedAWA | ||||||
| FedLWS | ||||||
| FedNova | ||||||
| Methods | C1 | C2 | C3 | C4 | C5 | C6 |
| Local | ||||||
| Generic FL methods | ||||||
| FedAvg | ||||||
| FedAWA | ||||||
| FedLWS | ||||||
| FedNova | ||||||
| Methods | C1 | C2 | C3 | C4 | C5 |
| Local | |||||
| Generic FL methods | |||||
| FedAvg | |||||
| FedAWA | |||||
| FedLWS | |||||
| FedNova | |||||
| Methods | C1 | C2 | C3 | C4 | C5 |
| Local | |||||
| Generic FL methods | |||||
| FedAvg | |||||
| FedAWA | |||||
| FedLWS | |||||
| FedNova | |||||
| Methods | C1 | C2 | C3 | C4 | C5 | C6 | C7 | C8 | C9 |
| Local | |||||||||
| Generic FL methods | |||||||||
| FedAvg | |||||||||
| FedAWA | |||||||||
| FedLWS | |||||||||
| FedNova | |||||||||
| Methods | ARIA | IOSTAR | MSD | Kvasir-SEG | BUS |
| FedAvg | |||||
| FedAWA | |||||
| FedLWS | |||||
| FedNova | |||||
| FedProx | |||||
| MOON |
| Methods | Fed-Vessel | Fed-Prostate | Fed-Polyp | Fed-COSAS | ||||||||
| aDice | Std | wDice | aDice | Std | wDice | aDice | Std | wDice | aDice | Std | wDice | |
| Generic FL methods | ||||||||||||
| FedAvg | 73.01 | 3.28 | 68.50 | 84.05 | 4.80 | 76.35 | 74.36 | 14.23 | 49.68 | 66.76 | 4.84 | 60.14 |
| FedAWA | 73.79 | 3.73 | 68.45 | 84.01 | 5.26 | 74.84 | 74.49 | 12.32 | 53.55 | 67.39 | 3.96 | 61.82 |
| FedLWS | 75.28 | 4.01 | 70.24 | 84.92 | 5.56 | 74.49 | 73.58 | 13.32 | 49.38 | 65.90 | 6.38 | 57.11 |
| FedNova | 73.98 | 3.68 | 68.86 | 84.05 | 5.46 | 74.94 | 75.92 | 10.60 | 58.79 | 66.76 | 4.84 | 60.14 |
| Methods | Fed-BUS | Fed-Pancreas | Fed-M&Ms | FeTS2022 | ||||||||
| aDice | Std | wDice | aDice | Std | wDice | aDice | Std | wDice | aDice | Std | wDice | |
| Generic FL methods | ||||||||||||
| FedAvg | 80.62 | 7.98 | 71.90 | 60.98 | 15.10 | 33.97 | 82.42 | 3.54 | 79.00 | 85.59 | 4.07 | 78.40 |
| FedAWA | 81.81 | 6.02 | 74.89 | 61.42 | 13.81 | 36.68 | 82.25 | 3.60 | 78.75 | 85.38 | 4.15 | 77.60 |
| FedLWS | 77.80 | 8.55 | 68.59 | 56.80 | 16.65 | 26.44 | 82.32 | 3.41 | 79.52 | 85.40 | 4.07 | 77.68 |
| FedNova | 81.66 | 6.53 | 73.44 | 62.55 | 12.73 | 41.83 | 82.54 | 3.37 | 79.01 | 85.65 | 4.10 | 78.11 |