Skip to Main content Skip to Navigation
Conference papers

Personalized Federated Learning through Local Memorization

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 k-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.
Complete list of metadata

https://hal.archives-ouvertes.fr/hal-03697969
Contributor : Othmane MARFOQ Connect in order to contact the contributor
Submitted on : Friday, June 17, 2022 - 2:28:50 PM
Last modification on : Monday, August 8, 2022 - 5:06:17 PM

File

ICML22__camera_ready_.pdf
Files produced by the author(s)

Identifiers

  • HAL Id : hal-03697969, version 1

Citation

Othmane Marfoq, Laetitia Kameni, Richard Vidal, Giovanni Neglia. Personalized Federated Learning through Local Memorization. ICML 2022 - International Conference on Machine Learning 2022, Jul 2022, Baltimore, United States. ⟨hal-03697969⟩

Share

Metrics

Record views

59

Files downloads

12