Linear Speedup in Personalized Collaborative Learning
Not on the map. Why?
Abstract (arXiv)
Collaborative training can improve the accuracy of a model for a user by trading off the model's bias (introduced by using data from other users who are potentially different) against its variance (due to the limited amount of data on any single user). In this work, we formalize the personalized collaborative learning problem as a stochastic optimization of a task 0 while giving access to N related but different tasks 1,..., N. We provide convergence guarantees for two algorithms in this setting -- a popular collaboration method known as weighted gradient averaging, and a novel bias correction method -- and explore conditions under which we can achieve linear speedup w.r.t. the number of auxiliary tasks N. Further, we also empirically study their performance confirming our theoretical insights.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 42 of its 53 references that are in the map. First posted on arXiv 10 Nov 2021.
Builds on
42 of 53 references.
References · 42
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2019Federated Learning: Challenges, Methods, and Future Directionscited by 1,531
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2020Ditto: Fair and Robust Federated Learning Through Personalizationcited by 665
- 2016Federated Optimization: Distributed Machine Learning for On-Device Intelligencecited by 558
- 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 526
- 2019Agnostic Federated Learningcited by 451
- 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 401
- 2019Fair Resource Allocation in Federated Learningcited by 389
- 2020Adaptive Personalized Federated Learningcited by 353
- 2020Three Approaches for Personalization with Applications to Federated Learningcited by 321
30 more
- 2021A Field Guide to Federated Optimizationcited by 312
- 2020Personalized Federated Learning: A Meta-Learning Approachcited by 309
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 302
- 2019A Little Is Enough: Circumventing Defenses For Distributed Learningcited by 285
- 2020Federated Learning of a Mixture of Global and Local Modelscited by 208
- 2020Personalized Federated Learning with First Order Model Optimizationcited by 201
- 2015The Benefit of Multitask Representation Learningcited by 194
- 2020On the Theory of Transfer Learning: The Importance of Task Diversitycited by 186
- 2019Adaptive Gradient-Based Meta-Learning Methodscited by 183
- 2013Minimizing Finite Sums with the Stochastic Average Gradientcited by 164
- 2020Mime: Mimicking Centralized Stochastic Algorithms in Federated Learningcited by 152
- 2019Federated Evaluation of On-device Personalizationcited by 147
- 2020Survey of Personalization Techniques for Federated Learningcited by 144
- 2020Salvaging Federated Learning by Local Adaptationcited by 142
- 2020Learning from History for Byzantine Robust Optimizationcited by 137
- 2013Communication Efficient Distributed Optimization using an Approximate Newton-type Methodcited by 131
- 2019Which Algorithmic Choices Matter at Which Batch Sizes? Insights From a Noisy Quadratic Modelcited by 116
- 2020On the Convergence of SGD with Biased Gradientscited by 79
- 2018Understanding Short-Horizon Bias in Stochastic Meta-Optimizationcited by 68
- 2020Model-sharing Games: Analyzing Federated Learning Under Voluntary Participationcited by 60
- 2020Auxiliary Learning by Implicit Differentiationcited by 45
- 2016AIDE: Fast and Communication Efficient Distributed Optimizationcited by 42
- 2021Personalized Federated Learning: A Unified Framework and Universal Optimization Techniquescited by 34
- 2021Sample Efficient Linear Meta-Learning by Alternating Minimizationcited by 25
- 2020Auxiliary Task Reweighting for Minimum-data Learningcited by 23
- 2017Byzantine-Tolerant Machine Learningcited by 18
- 2021WAFFLE: Weighted Averaging for Personalized Federated Learningcited by 9
- 2021Optimal Model Averaging: Towards Personalized Collaborative Learningcited by 7
- 2021Provable Adaptation across Multiway Domains via Representation Learningcited by 3
- 2021User-Centric Federated Learningcited by 3
Built on by
Citing papers per year: 2022: 1; 2023: 3; 2024: 2; 2025: 2.
8 papers.
- 2022Optimization with Access to Auxiliary Informationcited by 12
- 2023Incentivizing Honesty among Competitors in Collaborative Learning and Optimizationcited by 6
- 2023DePRL: Achieving Linear Convergence Speedup in Personalized Decentralized Learning with Shared Representationscited by 5
- 2024Incentivizing Truthful Collaboration in Heterogeneous Federated Learningcited by 3
- 2024Scalable Decentralized Algorithms for Online Personalized Mean Estimationcited by 2
- 2025Adaptive collaboration for online personalized distributed learning with heterogeneous clientscited by 2
- 2025Personalized Collaborative Learning with Affinity-Based Variance Reductioncited by 1
- 2023Utilizing Free Clients in Federated Learning for Focused Model Enhancementcited by 0
- 2025Personalized Collaborative Learning with Affinity-Based Variance Reduction17 Oct 2025
- 2025Adaptive collaboration for online personalized distributed learning with heterogeneous clients9 Jul 2025
- 2024Incentivizing Truthful Collaboration in Heterogeneous Federated Learning1 Dec 2024
- 2024Scalable Decentralized Algorithms for Online Personalized Mean Estimation20 Feb 2024
- 2023DePRL: Achieving Linear Convergence Speedup in Personalized Decentralized Learning with Shared Representations17 Dec 2023
- 2023Utilizing Free Clients in Federated Learning for Focused Model Enhancement6 Oct 2023
- 2023Incentivizing Honesty among Competitors in Collaborative Learning and Optimization25 May 2023
- 2022Optimization with Access to Auxiliary Information1 Jun 2022