Why β1 = β2 Is Dynamically Special in Adam
Not on the map. Why?
Abstract (arXiv)
Adam has been at the core of large-scale training for almost a decade, yet the role of its two momentum parameters remains poorly understood. Recent work shows that tying β1=β2 can preserve Adam's strong performance despite collapsing two memory scales into one, raising a basic question: what becomes dynamically special when the memories are tied? We identify a concrete mechanism. In the continuous-time limit, each normalized-update coordinate decomposes into a sign component, an explicit magnitude-lag term proportional to the difference between the two memory times, and additional transition, curvature, and nonlinear ratio terms. This lag channel vanishes exactly when β1=β2, making the diagonal the unique regime in which this mismatch-induced response is structurally absent. A full-history discrete decomposition on real training gradients recovers this change in composition: tied updates are sign-dominated, whereas the lag term becomes substantial off the diagonal and leaves a comparatively small residual. Across six vision and language tasks, tied configurations also typically exhibit smoother update-norm trajectories. Overall, our results identify memory-scale mismatch as a concrete source of magnitude sensitivity in Adam and provide a mechanistic account of why tied momentum is dynamically distinctive.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 16 of its 25 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
16 of 25 references.
References · 16
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
- 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 561
- 2023Symbolic Discovery of Optimization Algorithmscited by 423
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
- 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 136
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
4 more
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Refresh-Scaling the Memory of Balanced Adamcited by 0
- 2026Refresh-Scaling the Memory of Balanced Adam11 May 2026