mlsanityData to 7 Oct 2026

Why β1 = β2 Is Dynamically Special in Adam

arXiv 2601.21739 ↗First posted 29 Jan 2026cs.LG cs.AI stat.MLcited by 1

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

  1. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,115
  6. 2016Pointer Sentinel Mixture Modelscited by 2,992
  7. 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 561
  8. 2023Symbolic Discovery of Optimization Algorithmscited by 423
  9. 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
  10. 2025Training Deep Learning Models with Norm-Constrained LMOscited by 177
  11. 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 136
  12. 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
4 more
  1. 2025In Search of Adam's Secret Saucecited by 40
  2. 2020A Qualitative Study of the Dynamic Behavior for Adaptive Gradient Algorithmscited by 31
  3. 2018A general system of differential equations to model first order adaptive algorithmscited by 28
  4. 2025Sinusoidal Initialization, Time for a New Startcited by 5

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.