Personalized Federated Learning with Contextualized Generalization
Not on the map. Why?
Abstract (arXiv)
The prevalent personalized federated learning (PFL) usually pursues a trade-off between personalization and generalization by maintaining a shared global model to guide the training process of local models. However, the sole global model may easily transfer deviated context knowledge to some local models when multiple latent contexts exist across the local datasets. In this paper, we propose a novel concept called contextualized generalization (CG) to provide each client with fine-grained context knowledge that can better fit the local data distributions and facilitate faster model convergence, based on which we properly design a framework of PFL, dubbed CGPFL. We conduct detailed theoretical analysis, in which the convergence guarantee is presented and O(√K) speedup over most existing methods is granted. To quantitatively study the generalization-personalization trade-off, we introduce the 'generalization error' measure and prove that the proposed CGPFL can achieve a better trade-off than existing solutions. Moreover, our theoretical analysis further inspires a heuristic algorithm to find a near-optimal trade-off in CGPFL. Experimental results on multiple real-world datasets show that our approach surpasses the state-of-the-art methods on test accuracy by a significant margin.
Key facts. Cited by 23 papers in the map (data through 7 Oct 2026). Builds on 18 of its 32 references that are in the map. First posted on arXiv 24 Jun 2021.
Builds on
18 of 32 references.
References · 18
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2020Adaptive Federated Optimizationcited by 1,036
- 2017Federated Multi-Task Learningcited by 669
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 665
- 2020Personalized Federated Learning with Moreau Envelopescited by 612
- 2019Federated Learning with Personalization Layerscited by 533
- 2020An Efficient Framework for Clustered Federated Learningcited by 505
- 2019Clustered Federated Learning: Model-Agnostic Distributed Multi-Task Optimization under Privacy Constraintscited by 499
- 2019FedMD: Heterogenous Federated Learning via Model Distillationcited by 454
- 2020Adaptive Personalized Federated Learningcited by 353
6 more
- 2020Three Approaches for Personalization with Applications to Federated Learningcited by 321
- 2019The Non-IID Data Quagmire of Decentralized Machine Learningcited by 303
- 2020Federated learning with hierarchical clustering of local updates to improve training on non-IID datacited by 222
- 2020Federated Learning of a Mixture of Global and Local Modelscited by 208
- 2020Lower Bounds and Optimal Algorithms for Personalized Federated Learningcited by 122
- 2018A Tight Convergence Analysis for Stochastic Gradient Descent with Delayed Updatescited by 47
Built on by
Citing papers per year: 2021: 1; 2022: 3; 2023: 7; 2024: 8; 2025: 4.
23 papers.
Sort:
- 2022PromptFL: Let Federated Participants Cooperatively Learn Prompts Instead of Models -- Federated Learning in Age of Foundation Modelcited by 133
- 2024FedAC: An Adaptive Clustered Federated Learning Framework for Heterogeneous Datacited by 25
- 2023Bold but Cautious: Unlocking the Potential of Personalized Federated Learning through Cautiously Aggressive Collaborationcited by 23
- 2025LCFed: An Efficient Clustered Federated Learning Framework for Heterogeneous Datacited by 15
- 2022Fed-FSNet: Mitigating Non-I.I.D. Federated Learning via Fuzzy Synthesizing Networkcited by 14
- 2023Multimodal Federated Learning in Healthcare: a Reviewcited by 13
- 2024Neuron: Learning Context-Aware Evolving Representations for Zero-Shot Skeleton Action Recognitioncited by 13
- 2024Fine-Grained Zero-Shot Learning: Advances, Challenges, and Prospectscited by 12
- 2023Towards Fairer and More Efficient Federated Learning via Multidimensional Personalized Edge Modelscited by 10
- 2021From Deterioration to Acceleration: A Calibration Approach to Rehabilitating Step Asynchronism in Federated Optimizationcited by 8
- 2023GBE-MLZSL: A Group Bi-Enhancement Framework for Multi-Label Zero-Shot Learningcited by 8
- 2024TsCA: On the Semantic Consistency Alignment via Conditional Transport for Compositional Zero-Shot Learningcited by 6
11 more
- 2022Exploiting Personalized Invariance for Better Out-of-distribution Generalization in Federated Learningcited by 4
- 2024Dual Expert Distillation Network for Generalized Zero-Shot Learningcited by 4
- 2024The Diversity Bonus: Learning from Dissimilar Distributed Clients in Personalized Federated Learningcited by 3
- 2024Epsilon: Exploring Comprehensive Visual-Semantic Projection for Multi-Label Zero-Shot Learningcited by 3
- 2023Personalized Federated Learning via Gradient Modulation for Heterogeneous Text Summarizationcited by 1
- 2024Federated Learning Can Find Friends That Are Advantageouscited by 1
- 2025FedBKD: Distilled Federated Learning to Embrace Gerneralization and Personalization on Non-IID Datacited by 1
- 2023IP-FL: Incentivized and Personalized Federated Learningcited by 0
- 2023Entropy Aware Training for Fast and Accurate Distributed GNNcited by 0
- 2025FedAPA: Server-side Gradient-Based Adaptive Personalized Aggregation for Federated Learning on Heterogeneous Datacited by 0
- 2025FLStore: Efficient Federated Learning Storage for non-training workloadscited by 0
- 2025FedBKD: Distilled Federated Learning to Embrace Gerneralization and Personalization on Non-IID Data25 Jun 2025
- 2025FLStore: Efficient Federated Learning Storage for non-training workloads1 Mar 2025
- 2025FedAPA: Server-side Gradient-Based Adaptive Personalized Aggregation for Federated Learning on Heterogeneous Data11 Feb 2025
- 2025LCFed: An Efficient Clustered Federated Learning Framework for Heterogeneous Data3 Jan 2025
- 2024Neuron: Learning Context-Aware Evolving Representations for Zero-Shot Skeleton Action Recognition18 Nov 2024
- 2024Epsilon: Exploring Comprehensive Visual-Semantic Projection for Multi-Label Zero-Shot Learning22 Aug 2024
- 2024TsCA: On the Semantic Consistency Alignment via Conditional Transport for Compositional Zero-Shot Learning16 Aug 2024
- 2024The Diversity Bonus: Learning from Dissimilar Distributed Clients in Personalized Federated Learning22 Jul 2024
- 2024Dual Expert Distillation Network for Generalized Zero-Shot Learning25 Apr 2024
- 2024FedAC: An Adaptive Clustered Federated Learning Framework for Heterogeneous Data25 Mar 2024
- 2024Federated Learning Can Find Friends That Are Advantageous7 Feb 2024
- 2024Fine-Grained Zero-Shot Learning: Advances, Challenges, and Prospects31 Jan 2024
11 more
- 2023Entropy Aware Training for Fast and Accurate Distributed GNN4 Nov 2023
- 2023Multimodal Federated Learning in Healthcare: a Review14 Oct 2023
- 2023Bold but Cautious: Unlocking the Potential of Personalized Federated Learning through Cautiously Aggressive Collaboration20 Sep 2023
- 2023GBE-MLZSL: A Group Bi-Enhancement Framework for Multi-Label Zero-Shot Learning2 Sep 2023
- 2023Personalized Federated Learning via Gradient Modulation for Heterogeneous Text Summarization23 Apr 2023
- 2023IP-FL: Incentivized and Personalized Federated Learning15 Apr 2023
- 2023Towards Fairer and More Efficient Federated Learning via Multidimensional Personalized Edge Models9 Feb 2023
- 2022Exploiting Personalized Invariance for Better Out-of-distribution Generalization in Federated Learning21 Nov 2022
- 2022PromptFL: Let Federated Participants Cooperatively Learn Prompts Instead of Models -- Federated Learning in Age of Foundation Model24 Aug 2022
- 2022Fed-FSNet: Mitigating Non-I.I.D. Federated Learning via Fuzzy Synthesizing Network21 Aug 2022
- 2021From Deterioration to Acceleration: A Calibration Approach to Rehabilitating Step Asynchronism in Federated Optimization17 Dec 2021
- 2022PromptFL: Let Federated Participants Cooperatively Learn Prompts Instead of Models -- Federated Learning in Age of Foundation Modelcited by 133
- 2024FedAC: An Adaptive Clustered Federated Learning Framework for Heterogeneous Datacited by 25
- 2023Bold but Cautious: Unlocking the Potential of Personalized Federated Learning through Cautiously Aggressive Collaborationcited by 23
- 2025LCFed: An Efficient Clustered Federated Learning Framework for Heterogeneous Datacited by 15
- 2022Fed-FSNet: Mitigating Non-I.I.D. Federated Learning via Fuzzy Synthesizing Networkcited by 14
- 2023Multimodal Federated Learning in Healthcare: a Reviewcited by 13
- 2024Neuron: Learning Context-Aware Evolving Representations for Zero-Shot Skeleton Action Recognitioncited by 13
- 2024Fine-Grained Zero-Shot Learning: Advances, Challenges, and Prospectscited by 12
- 2023Towards Fairer and More Efficient Federated Learning via Multidimensional Personalized Edge Modelscited by 10
- 2021From Deterioration to Acceleration: A Calibration Approach to Rehabilitating Step Asynchronism in Federated Optimizationcited by 8
- 2023GBE-MLZSL: A Group Bi-Enhancement Framework for Multi-Label Zero-Shot Learningcited by 8
- 2024TsCA: On the Semantic Consistency Alignment via Conditional Transport for Compositional Zero-Shot Learningcited by 6
12 most cited of 23 All 23 (JSON)
- 2025FedBKD: Distilled Federated Learning to Embrace Gerneralization and Personalization on Non-IID Data25 Jun 2025
- 2025FLStore: Efficient Federated Learning Storage for non-training workloads1 Mar 2025
- 2025FedAPA: Server-side Gradient-Based Adaptive Personalized Aggregation for Federated Learning on Heterogeneous Data11 Feb 2025
- 2025LCFed: An Efficient Clustered Federated Learning Framework for Heterogeneous Data3 Jan 2025
- 2024Neuron: Learning Context-Aware Evolving Representations for Zero-Shot Skeleton Action Recognition18 Nov 2024
- 2024Epsilon: Exploring Comprehensive Visual-Semantic Projection for Multi-Label Zero-Shot Learning22 Aug 2024
- 2024TsCA: On the Semantic Consistency Alignment via Conditional Transport for Compositional Zero-Shot Learning16 Aug 2024
- 2024The Diversity Bonus: Learning from Dissimilar Distributed Clients in Personalized Federated Learning22 Jul 2024
- 2024Dual Expert Distillation Network for Generalized Zero-Shot Learning25 Apr 2024
- 2024FedAC: An Adaptive Clustered Federated Learning Framework for Heterogeneous Data25 Mar 2024
- 2024Federated Learning Can Find Friends That Are Advantageous7 Feb 2024
- 2024Fine-Grained Zero-Shot Learning: Advances, Challenges, and Prospects31 Jan 2024
12 newest of 23 All 23 (JSON)