Efficient Distribution Similarity Identification in Clustered Federated Learning via Principal Angles Between Client Data Subspaces
Not on the map. Why?
Abstract (arXiv)
Clustered federated learning (FL) has been shown to produce promising results by grouping clients into clusters. This is especially effective in scenarios where separate groups of clients have significant differences in the distributions of their local data. Existing clustered FL algorithms are essentially trying to group together clients with similar distributions so that clients in the same cluster can leverage each other's data to better perform federated learning. However, prior clustered FL algorithms attempt to learn these distribution similarities indirectly during training, which can be quite time consuming as many rounds of federated learning may be required until the formation of clusters is stabilized. In this paper, we propose a new approach to federated learning that directly aims to efficiently identify distribution similarities among clients by analyzing the principal angles between the client data subspaces. Each client applies a truncated singular value decomposition (SVD) step on its local data in a single-shot manner to derive a small set of principal vectors, which provides a signature that succinctly captures the main characteristics of the underlying distribution. This small set of principal vectors is provided to the server so that the server can directly identify distribution similarities among the clients to form clusters. This is achieved by comparing the similarities of the principal angles between the client data subspaces spanned by those principal vectors. The approach provides a simple, yet effective clustered FL framework that addresses a broad range of data heterogeneity issues beyond simpler forms of Non-IIDness like label skews. Our clustered FL approach also enables convergence guarantees for non-convex objectives. Our code is available at https://github.com/MMorafah/PACFL.
Key facts. Cited by 36 papers in the map (data through 14 Oct 2026). Builds on 25 of its 50 references that are in the map. First posted on arXiv 21 Sep 2022.
Builds on
25 of 50 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2014Deep Learning Face Attributes in the Wildcited by 3,693
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,898
- 2018Federated Learning with Non-IID Datacited by 1,108
- 2020Tackling the Objective Inconsistency Problem in Heterogeneous Federated Optimizationcited by 906
- 2017Federated Multi-Task Learningcited by 672
- 2021Federated Learning on Non-IID Data Silos: An Experimental Studycited by 568
- 2020An Efficient Framework for Clustered Federated Learningcited by 511
- 2019Clustered Federated Learning: Model-Agnostic Distributed Multi-Task Optimization under Privacy Constraintscited by 503
13 more
- 2020Adaptive Personalized Federated Learningcited by 358
- 2020Think Locally, Act Globally: Federated Learning with Local and Global Representationscited by 325
- 2020Three Approaches for Personalization with Applications to Federated Learningcited by 325
- 2019Improving Federated Learning Personalization via Model Agnostic Meta Learningcited by 269
- 2019On the Convergence of Local Descent Methods in Federated Learningcited by 153
- 2021Local Learning Matters: Rethinking Data Heterogeneity in Federated Learningcited by 114
- 2015Network Lasso: Clustering and Optimization in Large Graphscited by 43
- 2021Personalized Federated Learning by Structured and Unstructured Pruning under Data Heterogeneitycited by 27
- 2022FLIS: Clustered Federated Learning via Inference Similarity for Non-IID Data Distributioncited by 22
- 2021Federated Learning with Taskonomy for Non-IID Datacited by 21
- 2020Communication-Efficient Distributed SVD via Local Power Iterationscited by 12
- 2022Vertical Federated Principal Component Analysis and Its Kernel Extension on Feature-wise Distributed Datacited by 9
- 2020Federated Learning From Big Data Over Networkscited by 4
Built on by
Citing papers per year: 2022: 3; 2023: 4; 2024: 10; 2025: 11; 2026: 8.
36 papers.
Sort:
By area:
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2024Global and Local Prompts Cooperation via Optimal Transport for Federated Learningcited by 46
- 2025RL Fine-Tuning Heals OOD Forgetting in SFTcited by 31
- 2022Rethinking Data Heterogeneity in Federated Learning: Introducing a New Notion and Standard Benchmarkscited by 24
- 2024Personalized Federated Continual Learning via Multi-granularity Promptcited by 20
- 2023Optimizing the Collaboration Structure in Cross-Silo Federated Learningcited by 19
- 2024Advances in Robust Federated Learning: A Survey with Heterogeneity Considerationscited by 12
- 2024Stable Diffusion-based Data Augmentation for Federated Learning with Non-IID Datacited by 11
- 2022When Do Curricula Work in Federated Learning?cited by 9
- 2024FedClust: Optimizing Federated Learning on Non-IID Data through Weight-Driven Client Clusteringcited by 9
- 2024FedClust: Tackling Data Heterogeneity in Federated Learning through Weight-Driven Client Clusteringcited by 9
- 2023A Practical Recipe for Federated Learning Under Statistical Heterogeneity Experimental Designcited by 5
24 more
- 2022Clustered Federated Learning based on Nonconvex Pairwise Fusioncited by 4
- 2023Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologiescited by 3
- 2025A Survey on Foundation Models for Personalized Federated Intelligencecited by 3
- 2024Geometrically Inspired Kernel Machines for Collaborative Learning Beyond Gradient Descentcited by 2
- 2025A survey on Clustered Federated Learning: Taxonomy, Analysis and Applicationscited by 2
- 2025Operator-Theoretic Framework for Gradient-Free Federated Learningcited by 2
- 2026Few-for-Many Personalized Federated Learningcited by 2
- 2024FedRAV: Hierarchically Federated Region-Learning for Traffic Object Classification of Autonomous Vehiclescited by 1
- 2025MMiC: Mitigating Modality Incompleteness in Clustered Federated Learningcited by 1
- 2025Distilling A Universal Expert from Clustered Federated Learningcited by 1
- 2025CLoVE: Personalized Federated Learning through Clustering of Loss Vector Embeddingscited by 1
- 2024DA-PFL: Dynamic Affinity Aggregation for Personalized Federated Learningcited by 0
- 2024Equitable Federated Learning with Activation Clusteringcited by 0
- 2025Stratify: Rethinking Federated Learning for Non-IID Data through Balanced Samplingcited by 0
- 2025Tackling Heterogeneity in Federated Learning via Variance-Reduced Boltzmann Sampling within Homogeneous Social Coalitionscited by 0
- 2025Clustered Federated Learning via Embedding Distributionscited by 0
- 2025EBS-CFL: Efficient and Byzantine-robust Secure Clustered Federated Learningcited by 0
- 2026FedBiCross: Personalized One-Shot Federated Learning on Medical Imagescited by 0
- 2026FedDAG: Clustered Federated Learning via Global Data and Gradient Integration for Heterogeneous Environmentscited by 0
- 2026Differentially Private Clustered Federated Learning with Privacy-Preserving Initialization and Normality-Driven Aggregationcited by 0
- 2026FMCL: Class-Aware Client Clustering with Foundation Model Representations for Heterogeneous Federated Learningcited by 0
- 2026Encryption-Compatible Clustered Federated Learning via Distributed Expectation-Maximization over Metadatacited by 0
- 2026FeDepth: Federated Learning for Depth Estimation under Robot Heterogeneitycited by 0
- 2026CF-LoRA: Decoupled Factor Aggregation and Adaptation-Aware Client Clustering for Federated LoRA Fine-Tuningcited by 0
- 2026CF-LoRA: Decoupled Factor Aggregation and Adaptation-Aware Client Clustering for Federated LoRA Fine-Tuning29 Sep 2026
- 2026FeDepth: Federated Learning for Depth Estimation under Robot Heterogeneity2 Aug 2026
- 2026Encryption-Compatible Clustered Federated Learning via Distributed Expectation-Maximization over Metadata30 Jul 2026
- 2026FMCL: Class-Aware Client Clustering with Foundation Model Representations for Heterogeneous Federated Learning30 Apr 2026
- 2026Differentially Private Clustered Federated Learning with Privacy-Preserving Initialization and Normality-Driven Aggregation22 Apr 2026
- 2026Few-for-Many Personalized Federated Learning12 Mar 2026
- 2026FedDAG: Clustered Federated Learning via Global Data and Gradient Integration for Heterogeneous Environments26 Feb 2026
- 2026FedBiCross: Personalized One-Shot Federated Learning on Medical Images5 Jan 2026
- 2025Operator-Theoretic Framework for Gradient-Free Federated Learning30 Nov 2025
- 2025RL Fine-Tuning Heals OOD Forgetting in SFT8 Sep 2025
- 2025CLoVE: Personalized Federated Learning through Clustering of Loss Vector Embeddings27 Jun 2025
- 2025Distilling A Universal Expert from Clustered Federated Learning25 Jun 2025
24 more
- 2025EBS-CFL: Efficient and Byzantine-robust Secure Clustered Federated Learning16 Jun 2025
- 2025Clustered Federated Learning via Embedding Distributions9 Jun 2025
- 2025Tackling Heterogeneity in Federated Learning via Variance-Reduced Boltzmann Sampling within Homogeneous Social Coalitions3 Jun 2025
- 2025MMiC: Mitigating Modality Incompleteness in Clustered Federated Learning11 May 2025
- 2025A Survey on Foundation Models for Personalized Federated Intelligence11 May 2025
- 2025Stratify: Rethinking Federated Learning for Non-IID Data through Balanced Sampling18 Apr 2025
- 2025A survey on Clustered Federated Learning: Taxonomy, Analysis and Applications29 Jan 2025
- 2024FedRAV: Hierarchically Federated Region-Learning for Traffic Object Classification of Autonomous Vehicles21 Nov 2024
- 2024Equitable Federated Learning with Activation Clustering24 Oct 2024
- 2024FedClust: Tackling Data Heterogeneity in Federated Learning through Weight-Driven Client Clustering9 Jul 2024
- 2024Geometrically Inspired Kernel Machines for Collaborative Learning Beyond Gradient Descent5 Jul 2024
- 2024Personalized Federated Continual Learning via Multi-granularity Prompt27 Jun 2024
- 2024Advances in Robust Federated Learning: A Survey with Heterogeneity Considerations16 May 2024
- 2024Stable Diffusion-based Data Augmentation for Federated Learning with Non-IID Data13 May 2024
- 2024DA-PFL: Dynamic Affinity Aggregation for Personalized Federated Learning14 Mar 2024
- 2024FedClust: Optimizing Federated Learning on Non-IID Data through Weight-Driven Client Clustering7 Mar 2024
- 2024Global and Local Prompts Cooperation via Optimal Transport for Federated Learning29 Feb 2024
- 2023Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologies9 Oct 2023
- 2023A Practical Recipe for Federated Learning Under Statistical Heterogeneity Experimental Design28 Jul 2023
- 2023Optimizing the Collaboration Structure in Cross-Silo Federated Learning10 Jun 2023
- 2023Towards Building the Federated GPT: Federated Instruction Tuning9 May 2023
- 2022When Do Curricula Work in Federated Learning?24 Dec 2022
- 2022Clustered Federated Learning based on Nonconvex Pairwise Fusion8 Nov 2022
- 2022Rethinking Data Heterogeneity in Federated Learning: Introducing a New Notion and Standard Benchmarks30 Sep 2022
- 2025RL Fine-Tuning Heals OOD Forgetting in SFTcited by 31
- 2025RL Fine-Tuning Heals OOD Forgetting in SFT8 Sep 2025
- 2023Towards Building the Federated GPT: Federated Instruction Tuningcited by 185
- 2024Global and Local Prompts Cooperation via Optimal Transport for Federated Learningcited by 46
- 2022Rethinking Data Heterogeneity in Federated Learning: Introducing a New Notion and Standard Benchmarkscited by 24
- 2024Personalized Federated Continual Learning via Multi-granularity Promptcited by 20
- 2023Optimizing the Collaboration Structure in Cross-Silo Federated Learningcited by 19
- 2024Advances in Robust Federated Learning: A Survey with Heterogeneity Considerationscited by 12
- 2022When Do Curricula Work in Federated Learning?cited by 9
- 2024FedClust: Optimizing Federated Learning on Non-IID Data through Weight-Driven Client Clusteringcited by 9
- 2024FedClust: Tackling Data Heterogeneity in Federated Learning through Weight-Driven Client Clusteringcited by 9
- 2023A Practical Recipe for Federated Learning Under Statistical Heterogeneity Experimental Designcited by 5
- 2022Clustered Federated Learning based on Nonconvex Pairwise Fusioncited by 4
- 2023Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologiescited by 3
12 most cited of 33 All 33 (JSON)
- 2026CF-LoRA: Decoupled Factor Aggregation and Adaptation-Aware Client Clustering for Federated LoRA Fine-Tuning29 Sep 2026
- 2026FeDepth: Federated Learning for Depth Estimation under Robot Heterogeneity2 Aug 2026
- 2026Encryption-Compatible Clustered Federated Learning via Distributed Expectation-Maximization over Metadata30 Jul 2026
- 2026FMCL: Class-Aware Client Clustering with Foundation Model Representations for Heterogeneous Federated Learning30 Apr 2026
- 2026Differentially Private Clustered Federated Learning with Privacy-Preserving Initialization and Normality-Driven Aggregation22 Apr 2026
- 2026Few-for-Many Personalized Federated Learning12 Mar 2026
- 2026FedDAG: Clustered Federated Learning via Global Data and Gradient Integration for Heterogeneous Environments26 Feb 2026
- 2025Operator-Theoretic Framework for Gradient-Free Federated Learning30 Nov 2025
- 2025CLoVE: Personalized Federated Learning through Clustering of Loss Vector Embeddings27 Jun 2025
- 2025Distilling A Universal Expert from Clustered Federated Learning25 Jun 2025
- 2025EBS-CFL: Efficient and Byzantine-robust Secure Clustered Federated Learning16 Jun 2025
- 2025Clustered Federated Learning via Embedding Distributions9 Jun 2025
12 newest of 33 All 33 (JSON)