Personalized Federated Learning through Local Memorization
arXiv:2111.09360
Abstract
Federated learning allows clients to collaboratively learn statistical models while keeping their data local. Federated learning was originally used to train a unique global model to be served to all clients, but this approach might be sub-optimal when clients' local data distributions are heterogeneous. In order to tackle this limitation, recent personalized federated learning methods train a separate model for each client while still leveraging the knowledge available at other clients. In this work, we exploit the ability of deep neural networks to extract high quality vectorial representations (embeddings) from non-tabular data, e.g., images and text, to propose a personalization mechanism based on local memorization. Personalization is obtained by interpolating a collectively trained global model with a local -nearest neighbors (kNN) model based on the shared representation provided by the global model. We provide generalization bounds for the proposed approach in the case of binary classification, and we show on a suite of federated datasets that this approach achieves significantly higher accuracy and fairness than state-of-the-art methods.
arXiv comment23 pages, ICML 2022
References in corpus (19)
- Communication-Efficient Learning of Deep Networks from Decentralized Data
- Federated Learning: Strategies for Improving Communication Efficiency
- Ensemble Distillation for Robust Model Fusion in Federated Learning
- FedMD: Heterogenous Federated Learning via Model Distillation
- Improving Federated Learning Personalization via Model Agnostic Meta Learning
- Adaptive Personalized Federated Learning
- Three Approaches for Personalization with Applications to Federated Learning
- LEAF: A Benchmark for Federated Settings
- Agnostic Federated Learning
- Federated Learning of a Mixture of Global and Local Models
- HeteroFL: Computation and Communication Efficient Federated Learning for Heterogeneous Clients
- Think Locally, Act Globally: Federated Learning with Local and Global Representations
- Federated Multi-Task Learning
- Federated Multi-Task Learning under a Mixture of Distributions
- Lower Bounds and Optimal Algorithms for Personalized Federated Learning
- FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropout
- Personalized Cross-Silo Federated Learning on Non-IID Data
- Personalized Federated Learning using Hypernetworks
- Nearest Neighbor Machine Translation