TY - JOUR
T1 - FedCDC
T2 - Efficient Similarity Identification in Clustered Federated Learning via Community Detection on Non-IID Data
AU - Sun, Bingli
AU - Song, Xiao
AU - Tu, Yuchun
N1 - Publisher Copyright:
© 2014 IEEE.
PY - 2025
Y1 - 2025
N2 - federated learning (FL) enables decentralized devices to collaboratively train models without sharing raw data. However, when client data is highly heterogeneous, conventional FL often suffers from poor model performance and personalization. To address this, clustered FL (CFL) has emerged as a promising solution, grouping clients with similar data distributions to jointly learn better personalized models. Yet, most existing CFL methods rely on predefined thresholds or fixed numbers of clusters, which limits their adaptability to real-world, dynamic environments with diverse and evolving client data. This study introduces FedCDC, a novel CFL framework that leverages graph clustering to dynamically identify client communities without prior knowledge of clustering structure. Specifically, we propose a client similarity identification algorithm based on Louvain community detection community detection clustering (CDC), which constructs a similarity graph using model inference results and performs modularity-optimizing clustering. Furthermore, the graph-based approach captures high-order structural relationships among clients, enabling more precise and stable clustering even under severe data heterogeneity conditions. Extensive experiments across some benchmark datasets demonstrate that FedCDC consistently outperforms state-of-the-art (SOTA) baselines. In the challenging Dir(0.1) setting on CIFAR-100, FedCDC achieves accuracy gains of 12.75% over FedAvg, 23.33% over PerFedAvg, and 3.42% over FLIS(DC). More broadly, this work bridges graph theory with FL, introducing a scalable and interpretable way to form client communities. It provides a solution for real-world deployments of FL systems that are both accurate and adaptive, particularly in complex environments.
AB - federated learning (FL) enables decentralized devices to collaboratively train models without sharing raw data. However, when client data is highly heterogeneous, conventional FL often suffers from poor model performance and personalization. To address this, clustered FL (CFL) has emerged as a promising solution, grouping clients with similar data distributions to jointly learn better personalized models. Yet, most existing CFL methods rely on predefined thresholds or fixed numbers of clusters, which limits their adaptability to real-world, dynamic environments with diverse and evolving client data. This study introduces FedCDC, a novel CFL framework that leverages graph clustering to dynamically identify client communities without prior knowledge of clustering structure. Specifically, we propose a client similarity identification algorithm based on Louvain community detection community detection clustering (CDC), which constructs a similarity graph using model inference results and performs modularity-optimizing clustering. Furthermore, the graph-based approach captures high-order structural relationships among clients, enabling more precise and stable clustering even under severe data heterogeneity conditions. Extensive experiments across some benchmark datasets demonstrate that FedCDC consistently outperforms state-of-the-art (SOTA) baselines. In the challenging Dir(0.1) setting on CIFAR-100, FedCDC achieves accuracy gains of 12.75% over FedAvg, 23.33% over PerFedAvg, and 3.42% over FLIS(DC). More broadly, this work bridges graph theory with FL, introducing a scalable and interpretable way to form client communities. It provides a solution for real-world deployments of FL systems that are both accurate and adaptive, particularly in complex environments.
KW - clustered federated learning (CFL)
KW - community detection
KW - graph clustering
KW - non-IID data
UR - https://www.scopus.com/pages/publications/105013253926
U2 - 10.1109/JIOT.2025.3598149
DO - 10.1109/JIOT.2025.3598149
M3 - 文章
AN - SCOPUS:105013253926
SN - 2327-4662
VL - 12
SP - 43666
EP - 43680
JO - IEEE Internet of Things Journal
JF - IEEE Internet of Things Journal
IS - 20
ER -