On the Outsized Importance of Learning Rates in Local Update Methods
Not on the map. Why?
Abstract (arXiv)
We study a family of algorithms, which we refer to as local update methods, that generalize many federated learning and meta-learning algorithms. We prove that for quadratic objectives, local update methods perform stochastic gradient descent on a surrogate loss function which we exactly characterize. We show that the choice of client learning rate controls the condition number of that surrogate loss, as well as the distance between the minimizers of the surrogate and true loss functions. We use this theory to derive novel convergence rates for federated averaging that showcase this trade-off between the condition number of the surrogate loss and its alignment with the true loss function. We validate our results empirically, showing that in communication-limited settings, proper learning rate tuning is often sufficient to reach near-optimal behavior. We also present a practical method for automatic learning rate decay in local update methods that helps reduce the need for learning rate tuning, and highlight its empirical performance on a variety of tasks and datasets.
Key facts. Cited by 36 papers in the map (data through 14 Oct 2026). First posted on arXiv 2 Jul 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 5; 2023: 7; 2024: 5; 2025: 3; 2026: 2.
36 papers.
Sort:
- 2021A Field Guide to Federated Optimizationcited by 315
- 2021Linear Convergence in Federated Learning: Tackling Client Heterogeneity and Sparse Gradientscited by 123
- 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2022FedAvg with Fine Tuning: Local Updates Lead to Representation Learningcited by 56
- 2021Sharp Bounds for Federated Averaging (Local SGD) and Continuous Perspectivecited by 38
- 2023Federated Temporal Difference Learning with Linear Function Approximation under Environmental Heterogeneitycited by 27
- 2021Exploiting Heterogeneity in Robust Federated Best-Arm Identificationcited by 20
- 2021CodedPaddedFL and CodedSecAgg: Straggler Mitigation and Secure Aggregation in Federated Learningcited by 20
- 2023Federated Learning as Variational Inference: A Scalable Expectation Propagation Approachcited by 17
- 2021On Data Efficiency of Meta-learningcited by 16
- 2023Federated TD Learning over Finite-Rate Erasure Channels: Linear Speedup under Markovian Samplingcited by 15
24 more
- 2021On Convergence of Federated Averaging Langevin Dynamicscited by 14
- 2021A Non-parametric View of FedAvg and FedProx: Beyond Stationary Pointscited by 13
- 2024Randomized Asymmetric Chain of LoRA: The First Meaningful Theoretical Framework for Low-Rank Adaptationcited by 13
- 2024The Limits and Potentials of Local SGD for Distributed Heterogeneous Learning with Intermittent Communicationcited by 12
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learningcited by 12
- 2021FedChain: Chained Algorithms for Near-Optimal Communication Cost in Federated Learningcited by 11
- 2023Over-the-Air Federated Learning and Optimizationcited by 10
- 2025Understanding Outer Optimizers in Local SGD: Learning Rates, Momentum, and Accelerationcited by 9
- 2021A Marketplace for Trading AI Models based on Blockchain and Incentives for IoT Datacited by 8
- 2023Improving Accelerated Federated Learning with Compression and Importance Samplingcited by 8
- 2021Adaptive Client Sampling in Federated Learning via Online Learning with Bandit Feedbackcited by 7
- 2022FedNST: Federated Noisy Student Training for Automatic Speech Recognitioncited by 5
- 2021Coding for Straggler Mitigation in Federated Learningcited by 4
- 2022Straggler-Resilient Differentially-Private Decentralized Learningcited by 3
- 2022Online Meta-Learning for Model Update Aggregation in Federated Learning for Click-Through Rate Predictioncited by 2
- 2022Federated Hypergradient Descentcited by 2
- 2024FedECADO: A Dynamical System Model of Federated Learningcited by 2
- 2023SalientGrads: Sparse Models for Communication Efficient and Data Aware Distributed Federated Trainingcited by 1
- 2025First Provable Guarantees for Practical Private FL: Beyond Restrictive Assumptionscited by 1
- 2021Boosting Resource-Constrained Federated Learning Systems with Guessed Updatescited by 0
- 2024Approximate Gradient Coding for Privacy-Flexible Federated Learning with Non-IID Datacited by 0
- 2025Adaptive Federated Learning via Dynamical System Modelcited by 0
- 2026What's in a Smoothness Constant? Tighter Rates for Local SGD with Bounded Second-order Heterogeneitycited by 0
- 2026Theoretical Foundations of Communication-Efficient, Robust, and Practical Distributed and Federated Optimizationcited by 0
- 2026Theoretical Foundations of Communication-Efficient, Robust, and Practical Distributed and Federated Optimization6 Aug 2026
- 2026What's in a Smoothness Constant? Tighter Rates for Local SGD with Bounded Second-order Heterogeneity16 Jul 2026
- 2025First Provable Guarantees for Practical Private FL: Beyond Restrictive Assumptions25 Dec 2025
- 2025Adaptive Federated Learning via Dynamical System Model5 Oct 2025
- 2025Understanding Outer Optimizers in Local SGD: Learning Rates, Momentum, and Acceleration12 Sep 2025
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learning27 Nov 2024
- 2024FedECADO: A Dynamical System Model of Federated Learning13 Oct 2024
- 2024Randomized Asymmetric Chain of LoRA: The First Meaningful Theoretical Framework for Low-Rank Adaptation10 Oct 2024
- 2024The Limits and Potentials of Local SGD for Distributed Heterogeneous Learning with Intermittent Communication19 May 2024
- 2024Approximate Gradient Coding for Privacy-Flexible Federated Learning with Non-IID Data4 Apr 2024
- 2023Over-the-Air Federated Learning and Optimization16 Oct 2023
- 2023Improving Accelerated Federated Learning with Compression and Importance Sampling5 Jun 2023
24 more
- 2023Federated TD Learning over Finite-Rate Erasure Channels: Linear Speedup under Markovian Sampling14 May 2023
- 2023SalientGrads: Sparse Models for Communication Efficient and Data Aware Distributed Federated Training15 Apr 2023
- 2023Federated Learning as Variational Inference: A Scalable Expectation Propagation Approach8 Feb 2023
- 2023Federated Temporal Difference Learning with Linear Function Approximation under Environmental Heterogeneity4 Feb 2023
- 2023FedExP: Speeding Up Federated Averaging via Extrapolation23 Jan 2023
- 2022Straggler-Resilient Differentially-Private Decentralized Learning6 Dec 2022
- 2022Federated Hypergradient Descent3 Nov 2022
- 2022Online Meta-Learning for Model Update Aggregation in Federated Learning for Click-Through Rate Prediction30 Aug 2022
- 2022FedNST: Federated Noisy Student Training for Automatic Speech Recognition6 Jun 2022
- 2022FedAvg with Fine Tuning: Local Updates Lead to Representation Learning27 May 2022
- 2021Adaptive Client Sampling in Federated Learning via Online Learning with Bandit Feedback28 Dec 2021
- 2021CodedPaddedFL and CodedSecAgg: Straggler Mitigation and Secure Aggregation in Federated Learning16 Dec 2021
- 2021On Convergence of Federated Averaging Langevin Dynamics9 Dec 2021
- 2021A Marketplace for Trading AI Models based on Blockchain and Incentives for IoT Data6 Dec 2021
- 2021Sharp Bounds for Federated Averaging (Local SGD) and Continuous Perspective5 Nov 2021
- 2021What Do We Mean by Generalization in Federated Learning?27 Oct 2021
- 2021Boosting Resource-Constrained Federated Learning Systems with Guessed Updates21 Oct 2021
- 2021Coding for Straggler Mitigation in Federated Learning30 Sep 2021
- 2021Exploiting Heterogeneity in Robust Federated Best-Arm Identification13 Sep 2021
- 2021FedChain: Chained Algorithms for Near-Optimal Communication Cost in Federated Learning16 Aug 2021
- 2021A Field Guide to Federated Optimization14 Jul 2021
- 2021A Non-parametric View of FedAvg and FedProx: Beyond Stationary Points29 Jun 2021
- 2021Linear Convergence in Federated Learning: Tackling Client Heterogeneity and Sparse Gradients14 Feb 2021
- 2021On Data Efficiency of Meta-learning30 Jan 2021
- 2021A Field Guide to Federated Optimizationcited by 315
- 2021Linear Convergence in Federated Learning: Tackling Client Heterogeneity and Sparse Gradientscited by 123
- 2023FedExP: Speeding Up Federated Averaging via Extrapolationcited by 69
- 2021What Do We Mean by Generalization in Federated Learning?cited by 62
- 2022FedAvg with Fine Tuning: Local Updates Lead to Representation Learningcited by 56
- 2021Sharp Bounds for Federated Averaging (Local SGD) and Continuous Perspectivecited by 38
- 2023Federated Temporal Difference Learning with Linear Function Approximation under Environmental Heterogeneitycited by 27
- 2021Exploiting Heterogeneity in Robust Federated Best-Arm Identificationcited by 20
- 2021CodedPaddedFL and CodedSecAgg: Straggler Mitigation and Secure Aggregation in Federated Learningcited by 20
- 2023Federated Learning as Variational Inference: A Scalable Expectation Propagation Approachcited by 17
- 2021On Data Efficiency of Meta-learningcited by 16
- 2023Federated TD Learning over Finite-Rate Erasure Channels: Linear Speedup under Markovian Samplingcited by 15
12 most cited of 36 All 36 (JSON)
- 2026Theoretical Foundations of Communication-Efficient, Robust, and Practical Distributed and Federated Optimization6 Aug 2026
- 2026What's in a Smoothness Constant? Tighter Rates for Local SGD with Bounded Second-order Heterogeneity16 Jul 2026
- 2025First Provable Guarantees for Practical Private FL: Beyond Restrictive Assumptions25 Dec 2025
- 2025Adaptive Federated Learning via Dynamical System Model5 Oct 2025
- 2025Understanding Outer Optimizers in Local SGD: Learning Rates, Momentum, and Acceleration12 Sep 2025
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learning27 Nov 2024
- 2024FedECADO: A Dynamical System Model of Federated Learning13 Oct 2024
- 2024Randomized Asymmetric Chain of LoRA: The First Meaningful Theoretical Framework for Low-Rank Adaptation10 Oct 2024
- 2024The Limits and Potentials of Local SGD for Distributed Heterogeneous Learning with Intermittent Communication19 May 2024
- 2024Approximate Gradient Coding for Privacy-Flexible Federated Learning with Non-IID Data4 Apr 2024
- 2023Over-the-Air Federated Learning and Optimization16 Oct 2023
- 2023Improving Accelerated Federated Learning with Compression and Importance Sampling5 Jun 2023
12 newest of 36 All 36 (JSON)