English

Why Adam Works Better with $\beta_1 = \beta_2$: The Missing Gradient Scale Invariance Principle

Machine Learning 2026-05-12 v2 Artificial Intelligence Machine Learning

Abstract

Adam has been at the core of large-scale training for almost a decade, yet a simple empirical fact remains unaccounted for: both validation scores and the qualitative behaviour of the training runs improve when the momentum parameters satisfy β1=β2\beta_{1}=\beta_{2}. Some recent studies have reported this pattern, but there is still no explanation for why this choice helps. We show that this choice is closely tied to a structural property that we refer to as \textit{gradient scale invariance}. We formalize this notion and prove that Adam becomes gradient scale invariant of first order if and only if β1=β2\beta_{1}=\beta_{2}. This perspective places the balanced regime of Adam in direct alignment with the design principles underlying several recent optimizers that explicitly enforce scale-robust updates. The theory is supported by experiments across vision and language tasks, and across different architectural families, in which rescaling the gradient has a markedly smoother effect on the update when β1=β2\beta_{1}=\beta_{2}. Overall, our results offer a coherent explanation for an open question in the behavior of Adam and provide a simple principle that helps guide the design of future optimizers.

Keywords

Cite

@article{arxiv.2601.21739,
  title  = {Why Adam Works Better with $\beta_1 = \beta_2$: The Missing Gradient Scale Invariance Principle},
  author = {Alberto Fernández-Hernández and Cristian Pérez-Corral and Jose I. Mestre and Manuel F. Dolz and Enrique S. Quintana-Ortí},
  journal= {arXiv preprint arXiv:2601.21739},
  year   = {2026}
}

Comments

23 pages, 8 figures. Preprint

R2 v1 2026-07-01T09:25:44.571Z