Large Learning Rates Improve Generalization: But How Large Are We Talking About?
Not on the map. Why?
Abstract (arXiv)
Inspired by recent research that recommends starting neural networks training with large learning rates (LRs) to achieve the best generalization, we explore this hypothesis in detail. Our study clarifies the initial LR ranges that provide optimal results for subsequent training with a small LR or weight averaging. We find that these ranges are in fact significantly narrower than generally assumed. We conduct our main experiments in a simplified setup that allows precise control of the learning rate hyperparameter and validate our key findings in a more practical setting.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 14 of its 18 references that are in the map. First posted on arXiv 19 Nov 2023.
Builds on
14 of 18 references.
References · 14
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 291
- 2020The large learning rate phase of deep learning: the catapult mechanismcited by 216
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 184
- 2019Towards Explaining the Regularization Effect of Initial Large Learning Rate in Training Neural Networkscited by 162
- 2020Implicit Gradient Regularizationcited by 146
- 2022SGD with Large Step Sizes Learns Sparse Featurescited by 68
- 2018Theoretical Analysis of Auto Rate-Tuning by Batch Normalizationcited by 65
- 2020Reconciling Modern Deep Learning with Traditional Optimization Analyses: The Intrinsic Learning Ratecited by 59
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 39
- 2020Wide-minima Density Hypothesis and the Explore-Exploit Learning Rate Schedulecited by 25
Built on by
Citing papers per year: 2024: 1.
1 paper.
- 2024KV Shifting Attention Enhances Language Modelingcited by 8
- 2024KV Shifting Attention Enhances Language Modeling29 Nov 2024