FedPC: An Efficient Prototype-Based Clustered Federated Learning on Medical Imaging

• Published in: IEEE journal of biomedical and health informatics Vol. 29; no. 10; pp. 7396 - 7408
• Main Authors: Gao, Tianrun, Liu, Keyan, Yang, Yuning, Liu, Xiaohong, Zhang, Ping, Wang, Guangyu
• Format: Journal Article
• Language: English
• Published: United States IEEE 01.10.2025
• Subjects: Algorithms; Artificial intelligence; Biomedical imaging; Cluster Analysis; clustered federated learning (CFL); Clustering algorithms; Computational modeling; Contrastive learning; Data heterogeneity; Data models; Databases, Factual; Diagnostic Imaging - methods; disease diagnosis; Federated Learning; Humans; Image Processing, Computer-Assisted - methods; Machine Learning; prototype learning; Prototypes; Servers; Training
• ISSN: 2168-2194; 2168-2208; 2168-2208
• DOI: 10.1109/JBHI.2025.3567055

Federated learning (FL) has emerged as a promising distributed paradigm that enables collaborative model training while preserving data privacy, but it suffers from performance degradation due to data heterogeneity. Although clustered federated learning (CFL) attempts to address this challenge by grouping clients with similar data distributions, existing methods are inefficient in capturing client data representations, leading to incorrect cluster identities and inferior cluster performance. To overcome these limitations, we propose an efficient prototype-based CFL framework (FedPC). Specifically, we introduce a dual-prototype strategy combining specific prototypes and generalized prototypes to capture class representations for cluster identities, along with a prototype-contrastive training mechanism that maximizes intra-cluster prototype consistency to improve cluster performance. Extensive experiments on medical imaging datasets (BloodMNIST and DermaMNIST) demonstrate that the FedPC outperforms nine state-of-the-art (SOTA) approaches, achieving average improvements of 2.17% and 3.47%, respectively. Furthermore, the FedPC reduces communication overhead by 3.33 to 5.68 times compared to SOTA methods, showcasing its efficiency in real-world FL scenarios.