cs.LGMay 11, 2026

Refresh-Scaling the Memory of Balanced Adam

Authors: Alberto Fernández-HernándezCristian Pérez-CorralJose I. MestreManuel F. DolzEnrique S. Quintana-Ortí

Organizations: Universitat Polit`ecnica de Val`encia, Val`encia, Spain · Universitat Jaume I, Castell´o de la Plana, Spain

Abstract

Recent evidence suggests that Adam performs robustly when its momentum parameters are tied, β1=β2β_1=β_2, reducing the optimizer to a single remaining parameter. However, how this parameter should be set remains poorly understood. We argue that, in balanced Adam, ββ should not be treated as a dimensionless constant: it defines a statistical memory horizon Hβ=(1β)1H_β=(1-β)^{-1}. In terms of the effective learning horizon TEST_{\mathrm{ES}}, estimated from the validation trajectory, we study the refresh count Rβ=(1β)TESR_β=(1-β)T_{\mathrm{ES}}, which measures how many times Adam renews its internal statistics during the useful phase of training. Across 11 vision and language experiments, we find that choosing ββ so that Rβ1000R_β\approx1000 selects different ββ values depending on the training scale, yet improves robustness over the best fixed-beta baseline. Compared with the strongest fixed choice β=0.944β=0.944, the refresh rule improves worst-case robustness, reducing the maximum relative gap in validation loss by 33.4%, while bringing all 11 runs within 1% of their validation oracle. These results suggest that the remaining hyperparameter of balanced Adam is more naturally viewed as a memory-scale variable than as a fixed constant. This provides a simple budget-aware perspective on optimizer scaling and opens a path toward treating Adam's momentum as part of the learning dynamics rather than as a static default.

Explore similar work

CardsList
  1. Why β1=β2β_1 = β_2 Is Dynamically Special in Adam

    Jan 29, 2026Alberto Fernández-Hernández, Cristian Pérez-Corral, Jose I. Mestre +2AdamResponse Magnitude

  2. Gefen: Optimized Stochastic Optimizer

    Jun 11, 2026Nadav Benedek, Tomer Koren, Ohad FriedAdamScaling Laws