Three Approaches for Personalization with Applications to Federated Learning
About
The standard objective in machine learning is to train a single model for all users. However, in many learning scenarios, such as cloud computing and federated learning, it is possible to learn a personalized model per user. In this work, we present a systematic learning-theoretic study of personalization. We propose and analyze three approaches: user clustering, data interpolation, and model interpolation. For all three approaches, we provide learning-theoretic guarantees and efficient algorithms for which we also demonstrate the performance empirically. All of our algorithms are model-agnostic and work for any hypothesis class.
Yishay Mansour, Mehryar Mohri, Jae Ro, Ananda Theertha Suresh• 2020
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | CIFAR10 0.1-Dirichlet (test) | Generalized Accuracy (Accg)63.64 | 38 | |
| Next-Character Prediction | Shakespeare (test) | -- | 31 | |
| Image Classification | EMNIST | -- | 30 | |
| Language Modeling | Shakespeare | Accuracy (Mean)51.92 | 25 | |
| Image Classification | CIFAR10 0.6-Dirichlet (test) | Client Accp > Accg Ratio98.18 | 18 | |
| Language Modeling | Stack Overflow | Accuracy23.75 | 15 | |
| Image Classification | CIFAR10 0.1 | Accuracy (Generalized)63.64 | 11 | |
| Image Classification | CIFAR100 0.1 | Accuracy (Global)31.57 | 11 | |
| Image Classification | CIFAR100 0.6 | Acc_g34.76 | 11 | |
| Image Classification | CIFAR10 0.6 | Accuracy (Generalized)65.44 | 11 |
Showing 10 of 16 rows