How Memory in Optimization Algorithms Implicitly Modifies the Loss
Open accessAbstract
In modern optimization methods used in deep learning, each update depends on the history of previous iterations, often referred to as memory, and this dependence decays fast as the iterates go further into the past. For example, gradient descent with momentum has exponentially decaying memory through exponentially averaged past gradients. We introduce a general technique for identifying a memoryless algorithm that approximates an optimization algorithm with memory. It is obtained by replacing all past iterates in the update by the current one, and then adding a correction term arising from memory (also a function of the current iterate). This correction term can be interpreted as a perturbation of the loss, and the nature of this perturbation can inform how memory implicitly (anti-)regularizes the optimization dynamics. As an application of our theory, we find that Lion does not have the kind of implicit anti-regularization induced by memory that AdamW does, providing a theory-based explanation for Lion's better generalization performance recently documented.
AI snapshot
Written from the paper’s full text.
TL;DR
The paper introduces a general technique for replacing an optimization algorithm with exponentially decaying memory by a memoryless iteration plus a correction term that acts as a perturbation of the loss. Applying it to AdamW and LionK, it argues AdamW’s memory anti-regularizes while Lion’s does not, providing a theory-based explanation for Lion’s better generalization, with preliminary experiments consistent with the theory.
Key findings
- Under Assumption 3.1, a memoryful iteration can be approximated by a memoryless iteration with correction term \(M^{(n)}(\theta)\); Theorem 3.2 gives a one-step error bound, and Corollary 3.3 gives a global bound \(\max_{n\in[0:\lfloor T/h\rfloor]}\|\theta^{(n)}-\tilde\theta^{(n)}\|\le C_2 h^2\) on a finite horizon.
- For heavy-ball momentum GD, memory implicitly adds a term proportional to \(h\beta/(2(1-\beta)^2)\|\nabla L(\theta)\|^2\) to the loss, penalizing the squared gradient norm; since \(\beta\) is usually close to one, this strongly penalizes \(\|\nabla L\|^2\) and is interpreted as implicit regularization toward flatter minima.
- For AdamW, neglecting exponentially decaying terms, memory anti-penalizes the one-norm of the gradient when \(\beta_2>\beta_1\); if weight decay is sufficiently small, memory anti-regularizes large-batch AdamW. For LionK, the correction is small as \(\varepsilon\) is small, so Lion does not suffer this anti-regularization; the correction terms are the same for Adam with \(\beta_1=\beta_2\) and Lion with \(\rho_1=\rho_2\) (Signum).
- Empirically, Figure 1 shows the memoryless iteration is closer than the first-order approximation (signGD) for Adam and Lion on an MLP with GELU on MNIST-10K, with learning rates \(h\in\{10^{-4},3\times10^{-4}\}\), weight decay \(10^{-3}/h\), and \(\varepsilon=10^{-6}\). ResNet-50 on CIFAR-10 with full-batch Adam shows test accuracy at training-loss threshold 0.05 drops as \(\beta_2\) approaches one; lower \(\beta_2\) can sometimes outperform default Lion and close the generalization gap. Transformer-XL on WikiText-2 shows higher \(\beta_2\) increases best validation perplexity. Compute was about \(12\times10\times3=360\) A100-GPU-hours per sweep.
Methods
- Theoretical framework: study iterations \(\theta^{(n+1)}=\theta^{(n)}-hF^{(n)}(\theta^{(n)},\dots,\theta^{(0)})\) with exponentially decaying memory (Assumption 3.1); derive a memoryless iteration with correction term \(M^{(n)}(\theta)\), prove Theorem 3.2 and Corollary 3.3, specialize to heavy-ball, Nesterov, AdamW, NAdam, and LionK, and derive modified equations and mini-batch noise corrections.
- Empirical design: preliminary sanity check on MLP with GELU on MNIST-10K; ResNet-50 on CIFAR-10 with full-batch Adam and Lion, \(\beta_1=0.99\), \(\varepsilon=10^{-6}\), weight decay \(0.005\), learning rate \(h=10^{-3.5}\), test accuracy at training-loss threshold 0.05; Transformer-XL on WikiText-2 with full-batch Adam, learning rate \(10^{-4}\), \(\beta_1=0.9\), \(\varepsilon=10^{-6}\); results averaged over three iterations.
- Code is available at the linked GitHub repository.
Limitations
- Stated in the paper: the approximation bounds are in terms of learning rate \(h\), so \(h\) must be sufficiently small; mini-batch noise has a non-negligible effect, and Lion does not necessarily outperform Adam if batches are small; large-scale empirical testing is out of scope due to compute constraints; interpreting higher-order corrections for other optimizers such as Shampoo is left as future work.
- Evident from the text: the main theoretical guarantees rely on smoothness, bounded derivatives, and exponential memory decay; the AdamW/Lion conclusions are developed for full-batch or large-batch regimes; the empirical evidence is preliminary and limited to MNIST-10K, CIFAR-10, and WikiText-2, not large-scale LLM training.
Model-written summary. Check it against the paper before relying on it.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
3 of 49 cited sources checked · 2 supported · 0 flagged · accuracy not yet measured
- Interpretationreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 8
Better generalization of Lion on a number of tasks may be partially attributed to the fact that memory does not anti-regularize Lion.
“Therefore, better generalization of Lion on a number of tasks [13] may be partially attributed to the fact that memory does not anti-regularize Lion.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [13] [13] X. Chen, C. Liang, D. Huang, E. Real, K. Wang, H. Pham, X. Dong, T. Luong, C.J. Hsieh, Y. Lu, and Q. V. Le. Symbolic discovery of optimization algorithms. In Advances in Neural Information Processing Systems , volume 36, pages 49205-49233, 2023. URL https://proceedings.neurips.cc/paper\_files/paper/2023/hash/ 9a39b4925e35cf447ccba8757137d84f-Abstract.html . - source
- Findingnone given§ Abstract · p. 1
Empirical evaluations confirm the paper's theoretical findings.
“Empirical evaluations confirm our theoretical findings.”
Quote in the paperNot a substantial claim - Findingcitation§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18
Lemma H.2 implies that in the full-batch case the iteration becomes close to a simplified form.
“Lemma H.2 then implies that the iteration becomes close to”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Lemma H.2
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
- The authors state this as an assumption: In the full-batch case, b(n+1)ℓ → bℓ as n→∞, where bℓ is constant in n.
- Findingown data§ J Empirical Illustrations · p. 26
In ResNet-50 on CIFAR-10 with full-batch Adam, test accuracy at a fixed small training loss threshold drops as β2 approaches one.
“As predicted in Section 4 and confirming the observation from [11] for pure Adam without weight decay, the test accuracy drops as β 2 approaches one.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites [11] [11] M. D. Cattaneo, J. M. Klusowski, and B. Shigida. On the implicit bias of Adam. In Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 5862-5906. PMLR, 2024. URL https://proceedings. mlr.press/v235/cattaneo24a.html . -
Not checked in the free run.
- Findingown data§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 28
If β1 = 0.99, preconditioned sharpness does not reach the stability threshold.
“If we take β 1 = 0 . 99 which is the 'smooth' regime of training, preconditioned sharpness does not reach the stability threshold (Figure 6).”
Quote in the paperNot a substantial claim
- Findingreasoning§ 5.1 Modified Equations · p. 8
For full-batch heavy-ball momentum GD, the additional term arising from discreteness is h^2(1-β)^-2 ∇||∇L(θ)||^2/4, providing additional implicit regularization.
“For the example of full-batch heavy-ball momentum GD as in Section 2, where G 1 ( θ ) = -(1 -β ) -1 ∇L ( θ ) (ignoring coefficients going to zero exponentially fast in n ), this additional term is equal to h 2 (1 -β ) -2 ∇∥∇L ( θ ) ∥ 2 / 4 , providing additional implicit regularization.”
Quote in the paperSubstantial claim - Findingreasoning§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18
In the special case of Assumption C.1, the correction term in the memoryless iteration (4) is given by a formula for r = 1, ..., d.
“Therefore, in this special case, the correction term in the memoryless iteration (4) is given by, for r = 1 , . . . , d ,”
Quote in the paperNot a substantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
- Findingreasoning§ 5.2 Mini-Batch Training · p. 9
Taking the average over all permutations π, for large n the result is given by the paper's expression.
“We can take the average E over all permutations π (re-orderings of mini-batches). Deferring some details to Section E, the result is that for large n”
Quote in the paperSubstantial claim- The authors state this as an assumption: In mini-batch training, the form of F(n)(θ(n),...,θ(0)) is given by the paper's expression, where the g(k) functions are uniformly bounded along with two derivatives, and π is a random permutation of (0,...,n) chosen uniformly among all such permutations.
- Findingreasoning§ 1 Introduction · p. 4
The paper proves an explicit bound on the approximation error between the memoryfull iteration and the memoryless iteration as a function of the learning rate h.
“We then prove an explicit bound on the approximation error ∥ θ ( n ) -˜ θ ( n ) ∥ , as a function of the learning rate h .”
Quote in the paperSubstantial claim - Findingreasoning§ 3.2 Approximation Bound · p. 6
Under Assumption 3.1, there exists a discrete memoryless iteration satisfying (4) with initial condition equal to the original initial condition, correction function defined in Equation (9), and a constant C1 not depending on h, such that a one-step error bound holds.
“Theorem 3.2 (Memoryless approximation: 1-step error bound) . Under Assumption 3.1, there exists a discrete memoryless iteration { ˜ θ ( n ) } ∞ n =0 satisfying (4) with initial condition ˜ θ (0) = θ (0) , correction function defined in Equation (9) , and a constant C 1 not depending on h , such that”
Quote in the paperSubstantial claim- The authors state this as an assumption: Memory decays exponentially fast uniformly in n, as formalized by Assumption 3.1.
- Findingreasoning§ 5.2 Mini-Batch Training · p. 9
For mini-batch GD with momentum, taking g(k)(θ)=∇L(k)(θ), on average the iteration is like (6) except the modified loss has an additional regularization term.
“For example, take g ( k ) ( θ ) = ∇L ( k ) ( θ ) the k th minibatch loss. Then we obtained that 'on average' mini-batch GD with momentum is given by the iteration like (6), except the modified loss has an additional regularization term:”
Quote in the paperSubstantial claim- The authors state this as an assumption: In mini-batch training, the form of F(n)(θ(n),...,θ(0)) is given by the paper's expression, where the g(k) functions are uniformly bounded along with two derivatives, and π is a random permutation of (0,...,n) chosen uniformly among all such permutations.
- Findingreasoning§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5
Heavy-ball momentum gradient descent is close to ordinary gradient descent without momentum, with a different step size and a different loss.
“Now using ∇ 2 L ( θ ) ∇L ( θ ) = (1 / 2) ∇∥∇L ( θ ) ∥ 2 , we obtain that heavy-ball momentum GD is close to ordinary GD (no momentum) with a different step size and different loss, given by”
Quote in the paperSubstantial claim - Findingown data§ J Empirical Illustrations · p. 26
The memoryless iteration (4) is closer than the first-order approximation (signGD) to the original algorithm, as measured by the l∞ norm of the weight difference, with the correction term.
“As a sanity check, we verify that the memoryless iteration (4) that we find is closer than the first-order approximation (the one with no correction from memory: for example, the first-order approximation of Adam is sign gradient descent [47]). We see in Figure 1 that the ℓ ∞ (maximal) norm of the weight difference is smaller with the correction term.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [47] [47] C. Ma, L. Wu, and W. E. A qualitative study of the dynamic behavior for adaptive gradient algorithms. In Proceedings of the 2nd Mathematical and Scientific Machine Learning Conference , volume 145 of Proceedings of Machine Learning Research , pages 671-692. PMLR, 2022. URL https://proceedings.mlr.press/v145/ma22a.html . -
Not checked in the free run.
- Findingnone given§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 8
The correction terms are exactly the same for Adam with β1=β2=:β and Lion with ρ1=ρ2=β.
“In addition, notice that the correction terms are exactly the same for Adam with β 1 = β 2 =: β and Lion with ρ 1 = ρ 2 = β .”
Quote in the paperSubstantial claim - Findingreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7
For LionK, neglecting terms going to zero exponentially fast as n→∞, the memoryless iteration is given by the paper's expression.
“Consider now LionK (Example 1.5). Neglecting terms going to zero exponentially fast as n →∞ , the memoryless iteration is”
Quote in the paperSubstantial claim- The authors state this as an assumption: Memory decays exponentially fast uniformly in n, as formalized by Assumption 3.1.
- Findingreasoning§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18
In the full-batch case g(n)l(θ) ≡ gl(θ), if b(n+1)l → bl, then ̄g(n+1)l(θ) settle to ̄gl(θ) := bl(1−βl)−1 gl(θ).
“In the full-batch case g ( n ) ℓ ( θ ) ≡ g ℓ ( θ ) , this can be simplified further. Let us assume b ( n +1) ℓ -→ n →∞ b ℓ , where b ℓ is constant in n . Then ¯ g ( n +1) ℓ ( θ ) also become constant in n : specifically, they settle to ¯ g ℓ ( θ ) := b ℓ (1 -β ℓ ) -1 g ℓ ( θ ) .”
Quote in the paperNot a substantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
- The authors state this as an assumption: In the full-batch case, b(n+1)ℓ → bℓ as n→∞, where bℓ is constant in n.
- Findingnone given§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 28
Ordinary sharpness λ1(Ht) is much lower for small β2 (especially noticeable for β1 ≤ 0.9).
“Note also that ordinary sharpness λ 1 ( H t ) (top hessian eigenvalue) is much lower for small β 2 (especially noticeable for β 1 ⩽ 0 . 9 ).”
Quote in the paperSubstantial claim - Findingreasoning§ E Mini-Batch Training: Details · p. 20
For large n, the expected correction term can be written in a simplified form.
“So, for large n we can write”
Quote in the paperNot a substantial claim - Findingreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 8
For Lion, the correction term is small as long as ε is small.
“This term is small as long as ε is small.”
Quote in the paperSubstantial claim - Findingreasoning§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18
Specializing to the setup of Assumption C.1, for any s, n ∈ Z≥0, F(s)(θ) = Φ( ̄g(s+1)1(θ), ..., ̄g(s+1)L(θ)), where ̄g(s+1)l(θ) := b(n+1)l ∑s k=0 βkl g(n−k)l(θ).
“Specializing to the setup of Assumption C.1, and for any s, n ∈ Z ⩾ 0 , F ( s ) ( θ ) = Φ ( ¯ g ( s +1) 1 ( θ ) , . . . , ¯ g ( s +1) L ( θ ) ) , where ¯ g ( s +1) ℓ ( θ ) := b ( n +1) ℓ ∑ s k =0 β k ℓ g ( n -k ) ℓ ( θ )”
Quote in the paperSubstantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
- Premisecitation§ 1 Introduction · p. 3
NAdam with decoupled weight decay can be written in form (1).
“Example 1.4 (NAdam with decoupled weight decay; Dozat [22]) . The optimizer can be written in the form (1) with”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [22] [22] T. Dozat. Incorporating Nesterov momentum into Adam. ICLR Workshops , 2016. URL https://openreview.net/forum?id=OM0jvwB8jIp57ZJjtNEZ . -
Not checked in the free run.
- Interpretationnone given§ Abstract · p. 1
The paper introduces a general technique for identifying a memoryless algorithm that approximates an optimization algorithm with memory by replacing all past iterates in the update by the current one and adding a correction term arising from memory.
“We introduce a general technique for identifying a memoryless algorithm that approximates an optimization algorithm with memory. It is obtained by replacing all past iterates in the update by the current one, and then adding a correction term arising from memory (also a function of the current iterate).”
- Also stated in: § 1 Introduction · p. 3 · plain
“This technique starts with an iteration having memory, and replaces it by a memoryless iteration that approximates the original one, provided a correction term is added.”
Quote in the paperSubstantial claim - Also stated in: § 1 Introduction · p. 3 · plain
- Premisecitation§ 1 Introduction · p. 1
Many optimization methods used in deep learning are first-order methods with exponentially decaying memory.
“Many optimization methods used in deep learning are first-order methods with exponentially decaying memory.”
Quote in the paperSubstantial claim0 of 3 sources checked- cites [39] [39] A. Krizhevsky, I. Sutskever, and G. E. Hinton. Imagenet classification with deep convolutional neural networks. In Advances in Neural Information Processing Systems , volume 25, 2012. URL https://proceedings.neurips.cc/paper\_files/paper/2012/ hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html . -
Not checked in the free run.
- cites [36] [36] D. P. Kingma and J. Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980 , 2017. URL https://arxiv.org/abs/1412.6980 . -
Not checked in the free run.
- cites [46] [46] I. Loshchilov and F. Hutter. Decoupled weight decay regularization. In International Conference on Learning Representations , 2019. URL https://openreview.net/forum?id= Bkg6RiCqY7 . -
Not checked in the free run.
- Premisecitation§ 1 Introduction · p. 2
Nesterov's accelerated gradient descent can be written in form (1).
“Example 1.2 (Nesterov's accelerated gradient descent; Nesterov [53]) . This optimizer can be written in the form (1) with”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [53] [53] Y. Nesterov. A method for solving the convex programming problem with convergence rate o ( 1 k 2 ) . In Dokl. Akad. Nauk SSSR , volume 269, pages 543-547, 1983. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- Premisecitation§ 1 Introduction · p. 1
Adaptive methods such as Adam, RMSProp, AdamW, and AdaFactor all have exponentially decaying memory.
“Adaptive methods such as Adam [36], RMSProp [61], AdamW [46], and AdaFactor [57], which are commonly used to train large language models [27, 21, 15], all have exponentially decaying memory.”
Quote in the paperSubstantial claim0 of 3 sources checked- cites [36] [36] D. P. Kingma and J. Ba. Adam: A method for stochastic optimization. arXiv preprint arXiv:1412.6980 , 2017. URL https://arxiv.org/abs/1412.6980 . -
Not checked in the free run.
- cites [61] [61] T. Tieleman, G. Hinton, et al. Lecture 6.5-rmsprop: Divide the gradient by a running average of its recent magnitude. COURSERA: Neural networks for machine learning , 4(2): 26-31, 2012. URL https://www.cs.toronto.edu/ ~ tijmen/csc321/slides/lecture\_ slides\_lec6.pdf . -
Not checked in the free run.
- cites [46] [46] I. Loshchilov and F. Hutter. Decoupled weight decay regularization. In International Conference on Learning Representations , 2019. URL https://openreview.net/forum?id= Bkg6RiCqY7 . -
Not checked in the free run.
- Premisenone given§ 1 Introduction · p. 2
For many algorithms used in practice, dependence on the history comes in the form of momentum variables, that is, exponential averages of some functions of the iterate, usually functions of the loss gradient.
“For many algorithms used in practice, dependence on the history comes in one specific form: by using what we call ' momentum variables ', that is, exponential averages of some functions of the iterate θ ( n ) (usually, more specifically, functions of the loss gradient).”
Quote in the paperSubstantial claim - Interpretationreasoning§ 1 Introduction · p. 4
Interpreting the correction term can sometimes generate predictions on whether memory helps or hurts generalization of first-order methods with momentum.
“Interpreting the correction term can sometimes generate predictions on whether memory helps or hurts generalization of first-order methods with momentum.”
Quote in the paperSubstantial claim - Premisecitation§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7
By Lemma 3.8 in [66], θ is a KKT point of the constrained optimization problem if and only if the constraint is satisfied and (∗)=0.
“Incidentally, by Lemma 3.8 in that work, θ is a KKT point of this optimization problem if and only if the constraint is satisfied and ( ∗ ) = 0 .”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [66] [66] S. Xie and Z. Li. Implicit bias of AdamW: ℓ ∞ -norm constrained optimization. In R. Salakhutdinov, Z. Kolter, K. Heller, A. Weller, N. Oliver, J. Scarlett, and F. Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 54488-54510. PMLR, 2024. URL https://proceedings.mlr.press/v235/xie24e.html . -
Not checked in the free run.
- Premisecitation§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7
By Theorem 1.1 in [66], full-batch AdamW converges to a KKT point of the constrained optimization min_{||θ||∞ ≤ 1/λ} L(θ).
“By Theorem 1.1 in [66], full-batch AdamW converges to a KKT point of the constrained optimization min ∥ θ ∥ ∞ ⩽ 1 /λ L ( θ ) .”
Quote in the paperSubstantial claim0 of 1 sources checked- cites [66] [66] S. Xie and Z. Li. Implicit bias of AdamW: ℓ ∞ -norm constrained optimization. In R. Salakhutdinov, Z. Kolter, K. Heller, A. Weller, N. Oliver, J. Scarlett, and F. Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 54488-54510. PMLR, 2024. URL https://proceedings.mlr.press/v235/xie24e.html . -
Not checked in the free run.
- Interpretationreasoning§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5
The term added implicitly to the loss by momentum can be interpreted as implicit regularization.
“The term hβ 2(1 -β ) 2 ∥∇L ( θ ) ∥ 2 that is added implicitly to the loss by the momentum can be interpreted as implicit regularization.”
Quote in the paperSubstantial claim - Interpretationreasoning§ 5.2 Mini-Batch Training · p. 9
The second term in the averaged correction can be interpreted as implicit regularization by noise.
“The second term can be interpreted as implicit regularization by noise.”
Quote in the paperSubstantial claim - Interpretationreasoning§ 5.1 Modified Equations · p. 8
Since the memoryless iteration has no memory, standard methods can be used to derive an ODE whose continuous solution approximates this iteration and hence the initial algorithm with the same approximation guarantee.
“Since this iteration has no memory, standard methods can be used to derive an ordinary differential equation (ODE) in the form ˙ θ = G h ( θ ) =: G 1 ( θ ) + h G 2 ( h ) whose continuous solution approximates this iteration and hence the initial algorithm (with the same approximation guarantee).”
Quote in the paperSubstantial claim - Interpretationnone given§ A Related Literature · p. 17
Our results cover a much broader class of optimizers than Adam (with weight decay).
“Although we also discuss Adam (with weight decay) extensively, it is only because of its importance in practice, and our results cover a much broader class of optimizers.”
Quote in the paperSubstantial claim - Interpretationreasoning§ 5.2 Mini-Batch Training · p. 9
In specific cases, it is possible to identify the additional implicit regularization introduced to the algorithm by noise if mini-batch training is used as opposed to full-batch.
“In specific cases, it is possible to identify the additional implicit regularization that is introduced to the algorithm by noise, if mini-batch training is used as opposed to full-batch.”
Quote in the paperSubstantial claim - Premisecitation§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5
There is empirical evidence that penalization of the squared norm of the gradient improves generalization.
“There is empirical evidence that such penalization improves generalization [6, 59, 26].”
Quote in the paperNot a substantial claim0 of 3 sources checked- cites [6] [6] D. Barrett and B. Dherin. Implicit gradient regularization. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=3q5IqUrkcF . -
Not checked in the free run.
- cites [59] [59] S. L. Smith, B. Dherin, D. Barrett, and S. De. On the origin of implicit regularization in stochastic gradient descent. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=rq\_Qr0c1Hyo . -
Not checked in the free run.
- cites [26] [26] A. Ghosh, H. Lyu, X. Zhang, and R. Wang. Implicit regularization in heavy-ball momentum accelerated stochastic gradient descent. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview.net/forum?id=ZzdBhtEH9yB . -
Not checked in the free run.
- Interpretationreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7
If weight decay is sufficiently small, memory anti-regularizes large-batch AdamW.
“Thus, if weight decay is sufficiently small, memory anti-regularizes (large-batch) AdamW .”
Quote in the paperSubstantial claim - Interpretationreasoning§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5
Since β is usually taken to be close to one, the implicit regularization term strongly penalizes the squared norm of the gradient.
“Since β is usually taken to be close to one, the term strongly penalizes the squared norm of the gradient.”
Quote in the paperSubstantial claim - Premisecitation§ A Related Literature · p. 17
Cattaneo et al. [11] do not have approximation bounds between two discrete iterations like we do.
“Their focus is not on memory but on continuous approximations; in particular, they do not have approximation bounds between two discrete iterations like we do.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [11] [11] M. D. Cattaneo, J. M. Klusowski, and B. Shigida. On the implicit bias of Adam. In Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 5862-5906. PMLR, 2024. URL https://proceedings. mlr.press/v235/cattaneo24a.html . -
Not checked in the free run.
- Premisereasoning§ 5.1 Modified Equations · p. 8
The paper recovers the ODE derived by Kovachki and Stuart [37] and Ghosh et al. [26].
“We recover the ODE derived by Kovachki and Stuart [37], Ghosh et al. [26]:”
Quote in the paperNot a substantial claim0 of 2 sources checked- cites [37] [37] N. B. Kovachki and A. M. Stuart. Continuous time analysis of momentum methods. Journal of Machine Learning Research , 22(17):1-40, 2021. URL http://jmlr.org/papers/v22/ 19-466.html . -
Not checked in the free run.
- cites [26] [26] A. Ghosh, H. Lyu, X. Zhang, and R. Wang. Implicit regularization in heavy-ball momentum accelerated stochastic gradient descent. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview.net/forum?id=ZzdBhtEH9yB . -
Not checked in the free run.
- May combine two claims. 22
- Classified as probably not a substantial claim. 31
- The quote only partly states this claim. 4
May combine two claims.
Interpretationreasoning§ Abstract · p. 1Lion does not have the kind of implicit anti-regularization induced by memory that AdamW does, providing a theory-based explanation for Lion's better generalization performance recently documented.
“As an application of our theory, we find that Lion does not have the kind of implicit anti-regularization induced by memory that AdamW does, providing a theory-based explanation for Lion's better generalization performance recently documented [13].”
- Also stated in: § 1 Introduction · p. 4 · boosted
“As a substantive application, Section 4 studies AdamW (Example 1.3) and LionK (Example 1.5), and demonstrates that Lion does not suffer from the anti-regularization effect that AdamW's memory has, which predicts better generalization of (full-batch) Lion.”
Quote in the paperSubstantial claim1 of 1 sources checked- cites [13] [13] X. Chen, C. Liang, D. Huang, E. Real, K. Wang, H. Pham, X. Dong, T. Luong, C.J. Hsieh, Y. Lu, and Q. V. Le. Symbolic discovery of optimization algorithms. In Advances in Neural Information Processing Systems , volume 36, pages 49205-49233, 2023. URL https://proceedings.neurips.cc/paper\_files/paper/2023/hash/ 9a39b4925e35cf447ccba8757137d84f-Abstract.html . - source
- Also stated in: § 1 Introduction · p. 4 · boosted
May combine two claims.
Interpretationcitation§ 6 Limitations and Future Directions · p. 9There is a non-negligible effect of mini-batch noise on the picture described in Section J; in particular, Lion does not necessarily outperform Adam if batches are small.
“Additionally, there is a non-negligible effect of mini-batch noise on the picture we are describing in Section J; in particular, Lion does not necessarily outperform Adam if batches are small [13].”
Quote in the paperNot a substantial claim1 of 1 sources checked- cites [13] [13] X. Chen, C. Liang, D. Huang, E. Real, K. Wang, H. Pham, X. Dong, T. Luong, C.J. Hsieh, Y. Lu, and Q. V. Le. Symbolic discovery of optimization algorithms. In Advances in Neural Information Processing Systems , volume 36, pages 49205-49233, 2023. URL https://proceedings.neurips.cc/paper\_files/paper/2023/hash/ 9a39b4925e35cf447ccba8757137d84f-Abstract.html . - source
Section 6 (p. 14) says Lion is likely to perform no better than AdamW at batch sizes below 64, and Section 4.6 (p. 13) shows that Lion prefers larger batches. This supports the small-batch qualification, but the paper compares Lion with AdamW rather than Adam and discusses added noise from Lion’s sign operation rather than explicitly attributing the effect to mini-batch noise. (LLM, conf=0.85)
Experimental — accuracy not yet measured
May combine two claims.
Findingreasoning§ I Corollaries for Special Cases · p. 25Lemma I.3: For LionK (Example 1.5), under the stated conditions, Assumption 3.1 holds; in particular, by Corollary 3.3 the inequality max n∈[0:⌊T/h⌋] ||θ(n)− ̃θ(n)||∞ ≤ C2 h2 holds.
“Lemma I.3 (Application of Corollary 3.3 to Example 1.5) . Let { θ ( n ) } n ∈ Z ⩾ 0 be the sequence of vectors generated by the iteration in Equation (1) with initial condition θ (0) , where F ( n ) ( · ) is as defined in Example 1.5, the function K ( · ) : M→ R defined in a open bounded region M⊂ R d is three times continuously differentiable with bounded derivatives, and the loss function L ( ·”
Quote in the paperNot a substantial claim- The authors state this as an assumption: The loss function L is three times continuously differentiable with bounded derivatives on an open convex bounded domain D.
- The authors state this as an assumption: The function K is three times continuously differentiable with bounded derivatives on an open bounded region M.
May combine two claims.
Findingown data§ J Empirical Illustrations · p. 26Adam with lower values of β2 can sometimes outperform Lion with default hyperparameters and thus close the generalization gap between these two algorithms.
“We see that Adam with lower values of β 2 can sometimes outperform Lion with default hyperparameters and thus close the generalization gap between these two algorithms, consistent with the theory.”
Quote in the paperNot a substantial claimMay combine two claims.
Findingreasoning§ H Auxiliary Results · p. 22Lemma H.1: If F(n) is a function of 'momentum variables' as described in Assumption C.1, then for any n and k ≤ n, the memory decays exponentially fast, and similarly for any n and k1, k2 ≤ n.
“Lemma H.1 (Memory decays exponentially fast) . If F ( n ) is a function of 'momentum variables' as described in Assumption C.1, then for any n and k ⩽ n”
Quote in the paperSubstantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
Probably not a substantial claim: may not pass S3 (load-bearing).
Findingown data§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 27We reproduce the result that preconditioned sharpness oscillates around the stability threshold in a CNN trained on CIFAR-10.
“We train a CNN on CIFAR-10 as well, and reproduce this result in Figure 5.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Findingnone given§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18In the case of LionK (Example 1.5), Assumption C.1 applies with L = 3.
“In the case of LionK (Example 1.5), the assumption applies with L = 3”
Quote in the paperNot a substantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
The quote only partly states this claim.
Findingreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7For AdamW, taking ε to zero, the memoryless iteration can be written in the form of preconditioned gradient descent with decoupled weight decay and a modified loss.
“Taking ε to zero, we can write this in the form of preconditioned gradient descent (with decoupled weight decay):”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S4 (based).
Findingnone given§ E Mini-Batch Training: Details · p. 19E ∇g(i)r(θ)T g(j)(θ) depends only on whether i = j or i ≠ j.
“Note that E ∇ g ( i ) r ( θ ) T g ( j ) ( θ ) depends only on whether i = j or i = j .”
Quote in the paperNot a substantial claimMay combine two claims.
Findingown data§ J Empirical Illustrations · p. 26In Transformer-XL on WikiText-2 with full-batch Adam, higher β2 increases the best validation perplexity (hurts generalization), and sometimes taking lower β2 can close the gap between Adam and Lion.
“We observe in Figure 2(b) the same trends as above (in large-batch training, higher β 2 increases the best validation perplexity, that is, hurts generalization; sometimes, taking lower β 2 can close the gap between Adam and Lion).”
Quote in the paperNot a substantial claimClassified as probably not a substantial claim. The quote only partly states this claim.
Findingreasoning§ 5.2 Mini-Batch Training · p. 9The correction term introduced by memory (9) is given by the paper's expression.
“The correction term introduced by memory (9) is”
Quote in the paperSubstantial claim- The authors state this as an assumption: Memory decays exponentially fast uniformly in n, as formalized by Assumption 3.1.
Probably not a substantial claim: may not pass S3 (load-bearing).
Findingown data§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 27Ordinary sharpness λ1(Ht) first increases and then decreases during training.
“We also plot ordinary sharpness λ 1 ( H t ) (top hessian eigenvalue), which first increases and then decreases.”
Quote in the paperNot a substantial claimThe quote only partly states this claim.
Findingreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7For AdamW, neglecting coefficients decaying to zero exponentially fast, the memoryless iteration has a correction term M(n)(θ) given by the paper's expression.
“We first study AdamW with memory by an application of Theorem 3.2 and Corollary 3.3. Neglecting coefficients decaying to zero exponentially fast, we have”
- Also stated in: § 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7 · plain
“where M ( n ) ( θ ) is given by”
Quote in the paperSubstantial claim- The authors state this as an assumption: Memory decays exponentially fast uniformly in n, as formalized by Assumption 3.1.
- Also stated in: § 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7 · plain
Probably not a substantial claim: may not pass S3 (load-bearing). The quote only partly states this claim.
Findingreasoning§ H Auxiliary Results · p. 23Lemma H.2: Let {ak}∞k=1 and {bk}∞k=1 be sequences of reals such that ∑∞k=1(|ak|+|bk|) < ∞. Then a certain limit identity holds.
“Lemma H.2. Let { a k } ∞ k =1 and { b k } ∞ k =1 be sequences of reals such that ∑ ∞ k =1 ( | a k | + | b k | ) < ∞ . Then”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Findingnone given§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 18In the case of AdamW (Example 1.3), Assumption C.1 applies with L = 3.
“For instance, in the case of AdamW (Example 1.3), Assumption C.1 applies with L = 3”
Quote in the paperNot a substantial claim- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
May combine two claims.
Premisecitation§ 1 Introduction · p. 2Heavy-ball momentum gradient descent can be written in form (1), and its momentum variable exponentially averages past gradients.
“Example 1.1 (Heavy-ball momentum gradient descent; Polyak [54]) . This optimizer can be written in the form (1) with”
- Also stated in: § 1 Introduction · p. 2 · plain
“The optimizer in Example 1.1 is often just referred to as GD with momentum, where the exponential sum m ( n +1) 1 in Equation (2) is the 'momentum variable': it exponentially averages past gradients.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites [54] [54] B. Polyak. Some methods of speeding up the convergence of iteration methods. USSR Computational Mathematics and Mathematical Physics , 4(5):1-17, 1964. ISSN 0041-5553. doi: https://doi.org/10.1016/0041-5553(64)90137-5. URL https://www.sciencedirect. com/science/article/pii/0041555364901375 . -
Not checked in the free run.
- Also stated in: § 1 Introduction · p. 2 · plain
Classified as probably not a substantial claim.
Premisecitation§ Abstract · p. 1Gradient descent with momentum has exponentially decaying memory through exponentially averaged past gradients.
“For example, gradient descent with momentum has exponentially decaying memory through exponentially averaged past gradients.”
Quote in the paperSubstantial claim0 of 1 sources checked- cites [54] [54] B. Polyak. Some methods of speeding up the convergence of iteration methods. USSR Computational Mathematics and Mathematical Physics , 4(5):1-17, 1964. ISSN 0041-5553. doi: https://doi.org/10.1016/0041-5553(64)90137-5. URL https://www.sciencedirect. com/science/article/pii/0041555364901375 . -
Not checked in the free run.
May combine two claims.
Premisecitation§ 1 Introduction · p. 2AdamW can be written in form (1), with momentum variables m1 and m2 being exponentially averaged gradients and exponentially averaged squared gradient components, and m3 a degenerate momentum variable.
“Example 1.3 (AdamW; Loshchilov and Hutter [46]) . The optimizer can be written in the form (1) with”
- Also stated in: § 1 Introduction · p. 2 · plain
“In Example 1.3, m ( n +1) 1 and m ( n +1) 2 are also 'momentum variables': exponentially averaged gradients and exponentially averaged squared gradient components respectively, with coefficients in front of the sum, such as (1 -β 1 )(1 -β n +1 1 ) -1 , providing 'bias correction' [36]. The variable m ( n +1) 3 here is a degenerate 'momentum variable', with memory decaying infinitely fast.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [46] [46] I. Loshchilov and F. Hutter. Decoupled weight decay regularization. In International Conference on Learning Representations , 2019. URL https://openreview.net/forum?id= Bkg6RiCqY7 . -
Not checked in the free run.
- Also stated in: § 1 Introduction · p. 2 · plain
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 1 Introduction · p. 3LionK can be written in form (1).
“Example 1.5 (LionK ; Chen et al. [12]) . The optimizer can be written in the form of (1) with”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
May combine two claims.
Premisenone given§ Abstract · p. 1Modern optimization methods used in deep learning have memory: each update depends on the history of previous iterations, and this dependence decays fast as the iterates go further into the past.
“In modern optimization methods used in deep learning, each update depends on the history of previous iterations, often referred to as memory , and this dependence decays fast as the iterates go further into the past.”
Quote in the paperSubstantial claimMay combine two claims.
Interpretationreasoning§ Abstract · p. 1The correction term can be interpreted as a perturbation of the loss, and its nature can inform how memory implicitly (anti-)regularizes the optimization dynamics.
“This correction term can be interpreted as a perturbation of the loss, and the nature of this perturbation can inform how memory implicitly (anti-)regularizes the optimization dynamics.”
Quote in the paperSubstantial claimMay combine two claims.
Interpretationown data§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 28In large-batch training on vision tasks, taking β2 < β1 = 0.99 strongly regularizes training, moving the model parameters to flatter regions of the loss space.
“This suggests that in large-batch training on vision tasks, taking β 2 < β 1 = 0 . 99 strongly regularizes training, moving the model parameters to flatter regions of the loss space.”
Quote in the paperSubstantial claimMay combine two claims.
Interpretationreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 8Since Lion with ρ1=ρ2 is Signum, the paper provides a novel perspective on the similarity between Adam with β1≈β2 and Signum, a point discussed and verified empirically in [69].
“Since Lion with ρ 1 = ρ 2 is Signum [8], we provide a novel perspective on the similarity between Adam with β 1 ≈ β 2 and Signum, a point discussed and verified empirically in [69].”
Quote in the paperSubstantial claim0 of 2 sources checked- cites [8] [8] J. Bernstein, Y.-X. Wang, K. Azizzadenesheli, and A. Anandkumar. signSGD: Compressed optimisation for non-convex problems. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 560-569. PMLR, 10-15 Jul 2018. URL https://proceedings.mlr.press/v80/bernstein18a. html . -
Not checked in the free run.
- cites [69] [69] R. Zhao, D. Morwani, D. Brandfonbrener, N. Vyas, and S. M. Kakade. Deconstructing what makes a good optimizer for autoregressive language models. In The Thirteenth International Conference on Learning Representations , 2025. URL https://openreview.net/forum? id=zfeso8ceqr .
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing).
Interpretationreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7The observation that the correction term is zero if and only if the point is stationary is true of simpler full-batch algorithms, including Adam with λ=0 and full-batch GD with momentum.
“This is a generalization of the observation that the correction term is zero if and only if the point is stationary, true of simpler full-batch algorithms (for Adam with λ = 0 it follows from the above; for full-batch GD with momentum it is clear from (6)).”
Quote in the paperNot a substantial claimMay combine two claims.
Interpretationreasoning§ 1 Introduction · p. 4The theory only relies on memory decaying sufficiently fast, not necessarily in the form of momentum variables, and thus covers all listed examples and many others, while allowing for both full-batch and mini-batch training.
“Our theory only relies on memory decaying sufficiently fast, not necessarily in the form of momentum variables, and thus covers all the examples listed above and many others, while also allowing for both full-batch and mini-batch training.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S4 (based).
Interpretationnone given§ 6 Limitations and Future Directions · p. 9Practically relevant learning rates are often small enough, especially mid-training if a learning rate decay schedule is used.
“Fortunately, practically relevant learning rates are often small enough (especially mid-training, if a learning rate decay schedule is used).”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S2 (checkable), S3 (load-bearing). May combine two claims.
Premisecitation§ 6 Limitations and Future Directions · p. 9Between 2023 and 2025, the mainstream paradigm in deep learning is best described as scaling-centric rather than generalization-centric, which decreased interest in implicit regularization.
“Between 2023 and 2025, the mainstream paradigm in deep learning (especially large language models) is best described as scaling-centric rather than generalizationcentric [65] which decreased interest in implicit regularization.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [65] [65] L. Xiao. Rethinking conventional wisdom in machine learning: From generalization to scaling. arXiv preprint arXiv:2409.15156 , 2024. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ A Related Literature · p. 17Bernstein and Newhouse [7] view Adam and Lion as smoothed-out versions of steepest descent.
“Bernstein and Newhouse [7] also view Adam and Lion as smoothed-out versions of steepest descent.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [7] [7] J. Bernstein and L. Newhouse. Old optimizer, new norm: An anthology. In OPT 2024: Optimization for Machine Learning . -
Not checked in the free run.
May combine two claims.
Interpretationcitation§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5The term added implicitly to the loss by momentum can be interpreted as a first-order approximation of l2 sharpness, which suggests that it moves the trajectory towards flatter minima, often thought to improve generalization.
“In fact, this term (up to coefficients) can be interpreted as a first-order approximation of ℓ 2 sharpness [26], which suggests that it moves the trajectory towards 'flatter' minima; this is often thought to improve generalization [25].”
Quote in the paperSubstantial claim0 of 2 sources checked- cites [26] [26] A. Ghosh, H. Lyu, X. Zhang, and R. Wang. Implicit regularization in heavy-ball momentum accelerated stochastic gradient descent. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview.net/forum?id=ZzdBhtEH9yB . -
Not checked in the free run.
- cites [25] [25] P. Foret, A. Kleiner, H. Mobahi, and B. Neyshabur. Sharpness-aware minimization for efficiently improving generalization. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=6Tm1mposlrM . -
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing), S4 (based).
Interpretationnone given§ 6 Limitations and Future Directions · p. 10The approximation results in Section 3.2 hold for other algorithms like Shampoo or its versions, assuming a typical choice of momentum schemes.
“Finally, we discuss some of the most popular optimizers in recent years, but other important algorithms like Shampoo [30, 58] or its versions are also amenable to this analysis, and the approximation results in Section 3.2 hold for them (assuming a typical choice of momentum schemes).”
Quote in the paperNot a substantial claim0 of 2 sources checked- cites [30] [30] V. Gupta, T. Koren, and Y. Singer. Shampoo: Preconditioned stochastic tensor optimization. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 1842-1850. PMLR, 10-15 Jul 2018. URL https://proceedings.mlr.press/v80/gupta18a.html . -
Not checked in the free run.
- cites [58] [58] H.-J. M. Shi, T.-H. Lee, S. Iwasaki, J. Gallego-Posada, Z. Li, K. Rangadurai, D. Mudigere, and M. Rabbat. A distributed data-parallel pytorch implementation of the distributed Shampoo optimizer for training neural networks at-scale. arXiv preprint arXiv:2309.06497 , 2023. URL https://arxiv.org/abs/2309.06497 . -
Not checked in the free run.
Probably not a substantial claim: may not pass S4 (based). May combine two claims.
Interpretationnone given§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 28The parameter controlling the exponential forgetting of gradients β1 corresponds to the ρ2 parameter of Lion, so the default ρ2 = 0.99 in Lion would match β1 = 0.99 rather than β1 = 0.9 in Adam.
“Note that the parameter controlling the exponential forgetting of gradients β 1 corresponds to the ρ 2 parameter of Lion, so the default ρ 2 = 0 . 99 in Lion would match β 1 = 0 . 99 rather than β 1 = 0 . 9 in Adam.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Premisecitation§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 27Cohen et al. [16] define 'preconditioned sharpness' as the top eigenvalue of the preconditioned Hessian λ1(P−1t Ht), and observe that this quantity often oscillates around the stability threshold 2+2β1(1−β1)η.
“They define 'preconditioned sharpness' to be the top eigenvalue of the preconditioned Hessian λ 1 ( P -1 t H t ) , where H t is the Hessian of the loss, and observe that this quantity often oscillates around the stability threshold 2+2 β 1 (1 -β 1 ) η , where η is the learning rate.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [16] [16] J. M. Cohen, B. Ghorbani, S. Krishnan, N. Agarwal, S. Medapati, M. Badura, D. Suo, D. Cardoze, Z. Nado, G. E. Dahl, et al. Adaptive gradient methods at the edge of stability. arXiv preprint arXiv:2207.14484 , 2022. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7Ordinary Lion is recovered by setting K(x)=||x||_1.
“As mentioned above, ordinary Lion is recovered by setting K ( x ) = ∥ x ∥ 1 .”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
Classified as probably not a substantial claim. May combine two claims.
Study design and conductnone given§ K Experiment Details and Licenses · p. 28The implementations of the optimizers used for comparing the trajectories in Figure 1 are custom and match exactly the analytical formulas.
“The implementations of the optimizers used for comparing the trajectories in Figure 1 of the paper are custom and match exactly our analytical formulas (in particular, Lion has bias correction and the soft-sign function x ↦→ x/ √ x 2 + ε instead of the sign function), given below.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing), S4 (based). May combine two claims.
Interpretationnone given§ A Related Literature · p. 17Our approach works for a wide class of algorithms, and we recover their main results in Section 5.
“Our approach works for a wide class of algorithms, and we recover their main results in Section 5.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 1 Introduction · p. 3Ordinary Lion corresponds to K(x)=||x||_1 and ∇K(x)=sign(x) in Example 1.5, with the sign function understood component-wise.
“Ordinary Lion corresponds to K ( x ) = ∥ x ∥ 1 and ∇K ( x ) = sign( x ) in Example 1.5, where the sign function is understood component-wise.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing).
Interpretationreasoning§ C Special Case: F ( n ) as a Function of 'Momentum Variables' · p. 17Assumption C.1 is a special case of Assumption 3.1 by Lemma H.1.
“The following assumption, which is a special case of Assumption 3.1 by Lemma H.1, may look a bit technical but allows for a simpler calculation of correction terms.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites Lemma H.1
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
- The authors state this as an assumption: F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ A Related Literature · p. 17Weight decay causes AdamW and Lion to solve an l∞-norm constrained optimization problem.
“Xie and Li [66] and Chen et al. [12] prove that weight decay causes AdamW and Lion to solve an ℓ ∞ -norm constrained optimization problem.”
Quote in the paperNot a substantial claim0 of 2 sources checked- cites [66] [66] S. Xie and Z. Li. Implicit bias of AdamW: ℓ ∞ -norm constrained optimization. In R. Salakhutdinov, Z. Kolter, K. Heller, A. Weller, N. Oliver, J. Scarlett, and F. Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 54488-54510. PMLR, 2024. URL https://proceedings.mlr.press/v235/xie24e.html . -
Not checked in the free run.
- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ A Related Literature · p. 17AdamW and Lion behave asymptotically like (normalized) steepest descent with respect to l∞-norm.
“In that, they behave asymptotically like (normalized) steepest descent with respect to ℓ ∞ -norm.”
Quote in the paperNot a substantial claim0 of 2 sources checked- cites [66] [66] S. Xie and Z. Li. Implicit bias of AdamW: ℓ ∞ -norm constrained optimization. In R. Salakhutdinov, Z. Kolter, K. Heller, A. Weller, N. Oliver, J. Scarlett, and F. Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 54488-54510. PMLR, 2024. URL https://proceedings.mlr.press/v235/xie24e.html . -
Not checked in the free run.
- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
Classified as probably not a substantial claim.
Study design and conductreasoning§ 6 Limitations and Future Directions · p. 9The approximation bounds in Section 3.2 are in terms of the learning rate h, which means h has to be sufficiently small for the approximations and optimization trajectories to be close.
“The approximation bounds in Section 3.2 are in terms of the learning rate h , which means that h has to be sufficiently small for (our approximations and hence) the optimization trajectories to be close.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 27Cohen et al. [16] view Adam as momentum gradient descent with evolving preconditioner.
“They view Adam as momentum gradient descent with evolving preconditioner”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [16] [16] J. M. Cohen, B. Ghorbani, S. Krishnan, N. Agarwal, S. Medapati, M. Badura, D. Suo, D. Cardoze, Z. Nado, G. E. Dahl, et al. Adaptive gradient methods at the edge of stability. arXiv preprint arXiv:2207.14484 , 2022. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S4 (based).
Interpretationnone given§ A Related Literature · p. 17The general technique introduced in this paper can be used to derive a memoryless discrete iteration which can then be approximated by a continuous trajectory.
“The general technique introduced in this paper can be used to derive a memoryless discrete iteration which can then be approximated by a continuous trajectory.”
Quote in the paperNot a substantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ A Related Literature · p. 17The Moreau-Yosida approximation of the loss function provides a concrete way to write down popular optimizers (including SGD with momentum, RMSProp and Adam) as a sequence of optimization problems.
“the latter work provides a concrete way to write down popular optimizers (including SGD with momentum, RMSProp and Adam) as a sequence of optimization problems.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [9] [9] A. Betti, G. Ciravegna, M. Gori, S. Melacci, K. Mottin, and F. Precioso. A new perspective on optimizers: leveraging Moreau-Yosida approximation in gradient-based learning. Intelligenza Artificiale , 18(2):301-311, 2024. - source
May combine two claims.
Interpretationreasoning§ 4 Application: the Effect of Memory on AdamW, Lion and Signum · p. 7If ||θ||∞ ≤ 1/λ, then ||∇L(θ)||_1 dominates the term λ∇L(θ)^T θ in absolute value, so the main effect of memory is anti-penalizing the one-norm of the gradient.
“If ∥ θ ∥ ∞ ⩽ 1 /λ , then the norm ∥∇L ( θ ) ∥ 1 dominates the term λ ∇L ( θ ) T θ in absolute value, so the main effect of memory is anti-penalizing the one-norm of the gradient.”
Quote in the paperSubstantial claimProbably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ 2 Building Intuition: Memory Regularizes GD with Momentum · p. 5A much more fine-grained analysis of heavy-ball momentum gradient descent is provided in [10], where the modified loss is described to arbitrary precision rather than just to first order in h.
“A much more fine-grained analysis of this algorithm is provided in [10] where, in particular, the modified loss is described to arbitrary precision rather than just to first order in h .”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [10] [10] M. D. Cattaneo and B. Shigida. Modified loss of momentum gradient descent: Fine-grained analysis. arXiv preprint arXiv:2509.08483 , 2025. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ A Related Literature · p. 17Li et al. [43] prove in a certain setting that a larger learning rate leads to better generalization.
“Li et al. [43] prove in a certain setting that a larger learning rate leads to better generalization.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [43] [43] Y. Li, C. Wei, and T. Ma. Towards explaining the regularization effect of initial large learning rate in training neural networks. In Advances in Neural Information Processing Systems , volume 32, 2019. URL https://proceedings.neurips.cc/paper\_files/paper/2019/ hash/bce9abf229ffd7e570818476ee5d7dde-Abstract.html . -
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing).
Premisecitation§ J.1 A Note on the Edge of Stability and Comparisons with Lion on Vision Tasks · p. 27Cohen et al. [16] notice that in a sense Adam trains at the edge of stability.
“Cohen et al. [16] notice that in a sense Adam trains at the edge of stability.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [16] [16] J. M. Cohen, B. Ghorbani, S. Krishnan, N. Agarwal, S. Medapati, M. Badura, D. Suo, D. Cardoze, Z. Nado, G. E. Dahl, et al. Adaptive gradient methods at the edge of stability. arXiv preprint arXiv:2207.14484 , 2022. -
Couldn’t be matched to a paper in Crossref, so it can’t be checked.
May combine two claims.
Premisecitation§ A Related Literature · p. 17Cattaneo et al. [11]'s arguments are highly specialized to Adam, and they do not incorporate weight decay.
“In addition, their arguments are highly specialized to Adam, and they do not incorporate weight decay.”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [11] [11] M. D. Cattaneo, J. M. Klusowski, and B. Shigida. On the implicit bias of Adam. In Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 5862-5906. PMLR, 2024. URL https://proceedings. mlr.press/v235/cattaneo24a.html . -
Not checked in the free run.
Probably not a substantial claim: may not pass S3 (load-bearing). May combine two claims.
Premisecitation§ 1 Introduction · p. 3LionK covers a few known algorithms as special cases, and it includes heavy-ball momentum gradient descent as a special case by taking K(x)=||x||^2/2, ρ1=ρ2, and λ=0.
“We consider the generalized LionK algorithm because it covers a few known algorithms as special cases: see Table 1 and Section 3.1 in Chen et al. [12]. In fact, it also includes Example 1.1 as a special case by taking K ( x ) = ∥ x ∥ 2 / 2 , ρ 1 = ρ 2 , and λ = 0”
Quote in the paperNot a substantial claim0 of 1 sources checked- cites [12] [12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
Not checked in the free run.
The paper relies on this without support.
There is little theoretical knowledge about the implicit bias memory introduces to optimization methods.
“Despite the popularity of such optimization methods, there is little theoretical knowledge about the implicit bias memory introduces to them”
The authors state this as an assumption.
Memory decays exponentially fast uniformly in n, as formalized by Assumption 3.1.
“The only property of memory we use is that it (uniformly in n ) decays exponentially fast, as made precise by Assumption 3.1 below.”
The authors state this as an assumption.
The domain of optimization D is open and convex.
“Openness and convexity of the domain of optimization D , that is, where all { θ ( n ) } will be assumed to lie, are innocuous assumptions”
The authors state this as an assumption.
In mini-batch training, the form of F(n)(θ(n),...,θ(0)) is given by the paper's expression, where the g(k) functions are uniformly bounded along with two derivatives, and π is a random permutation of (0,...,n) chosen uniformly among all such permutations.
“Assume that the form of F ( n ) ( θ ( n ) , . . . , θ (0) ) is given by”
The authors state this as an assumption.
F(n) is a function of 'momentum variables' as described in Assumption C.1, with L two times continuously differentiable functions g(n)ℓ, fixed βℓ in [0,1), bounded nonnegative sequences b(n+1)ℓ, and a fixed two times continuously differentiable Φ.
“Assumption C.1 (Special case of Assumption 3.1: F ( n ) is a function of 'momentum variables') . Let { g ( n ) ℓ } L ℓ =1 be L two times continuously differentiable functions D → R d , uniformly bounded along with two derivatives. Let { β ℓ } L ℓ =1 be fixed reals in [0 , 1) , and { b ( n +1) ℓ } L ℓ =1 be L bounded nonnegative sequences of reals (for n ∈ Z ⩾ 0 ). Assume the function F ( n ) has”
The authors state this as an assumption.
In the full-batch case, b(n+1)ℓ → bℓ as n→∞, where bℓ is constant in n.
“Let us assume b ( n +1) ℓ -→ n →∞ b ℓ , where b ℓ is constant in n .”
The authors state this as an assumption.
The loss function L is three times continuously differentiable with bounded derivatives on an open convex bounded domain D.
“the loss function L ( · ) defined in an open convex bounded domain D ⊂ R d is three times continuously differentiable with bounded derivatives”
The authors state this as an assumption.
The function K is three times continuously differentiable with bounded derivatives on an open bounded region M.
“the function K ( · ) : M→ R defined in a open bounded region M⊂ R d is three times continuously differentiable with bounded derivatives”
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
Each contradiction between two passages listed here was confirmed by an independent check.
Mini-batch average includes an extra index
MinorThe mini-batches are indexed from 0 through n, but the stated average sums through n + 1 while still dividing by n + 1. It therefore includes an index outside the defined collection and is not the stated average.
“g ( θ ) = E g (1) ( θ ) = ( n + 1) -1 ∑ n +1 k =0 g ( k ) ( θ ) is the average of { g ( k ) ( θ ) }”
“the random permutation of minibatches, that is, take the average over all re-orderings π of ( g (0) , . . . , g ( n ) )”
Cites
The publisher hasn’t deposited this paper’s reference list.
Part of this paper’s reference list could not be split into single entries.
Research works printed but not linked: 43
- [3][3] S. Arora, N. Cohen, W. Hu, and Y. Luo. Implicit regularization in deep matrix factorization. In Advances in Neural Information Processing Systems , volume 32, 2019. URL https://proceedings.neurips.cc/paper\_files/paper/2019/hash/ c0c783b5fc0d7d808f1d14a6e9c8280d-Abstract.html . -
- [4][4] S. Arora, Z. Li, and A. Panigrahi. Understanding gradient descent on the edge of stability in deep learning. In Proceedings of the 39th International Conference on Machine Learning , volume 162 of Proceedings of Machine Learning Research , pages 948-1024. PMLR, 2022. URL https://proceedings.mlr.press/v162/arora22a.html . -
- [6][6] D. Barrett and B. Dherin. Implicit gradient regularization. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=3q5IqUrkcF . -
- [8][8] J. Bernstein, Y.-X. Wang, K. Azizzadenesheli, and A. Anandkumar. signSGD: Compressed optimisation for non-convex problems. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 560-569. PMLR, 10-15 Jul 2018. URL https://proceedings.mlr.press/v80/bernstein18a. html . -
- [11][11] M. D. Cattaneo, J. M. Klusowski, and B. Shigida. On the implicit bias of Adam. In Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 5862-5906. PMLR, 2024. URL https://proceedings. mlr.press/v235/cattaneo24a.html . -
- [12][12] L. Chen, B. Liu, K. Liang, and Q. Liu. Lion secretly solves a constrained optimization: As lyapunov predicts. In The Twelfth International Conference on Learning Representations , 2024. URL https://openreview.net/forum?id=e4xS9ZarDr . -
- [15][15] A. Chowdhery, S. Narang, J. Devlin, M. Bosma, G. Mishra, A. Roberts, P. Barham, H. W. Chung, C. Sutton, S. Gehrmann, P. Schuh, K. Shi, S. Tsvyashchenko, J. Maynez, A. Rao, P. Barnes, Y. Tay, N. Shazeer, V. Prabhakaran, E. Reif, N. Du, B. Hutchinson, R. Pope, J. Bradbury, J. Austin, M. Isard, G. Gur-Ari, P. Yin, T. Duke, A. Levskaya, S. Ghemawat, S. Dev, H. Michalewski, X. Garcia, V. Misra, K. Robinson, L. Fedus, D. Zhou, D. Ippolito, D. Luan, H. Lim, B. Zoph, A. Spiridonov, R. Sepassi, D. Dohan, S. Agrawal, M. Omernick, A. M. Dai, T. S. Pillai, M. Pellat, A. Lewkowycz, E. Moreira, R. Child, O. Polozov, K. Lee, Z. Zhou, X. Wang, B. Saeta, M. Diaz, O. Firat, M. Catasta, J. Wei, K. Meier-Hellstern, D. Eck, J. Dean, S. Petrov, and N. Fiedel. PaLM: Scaling language modeling with pathways. Journal of Machine Learning Research , 24(240):1-113, 2023. URL http://jmlr.org/papers/v24/22-1144. html . -
- [17][17] E. M. Compagnoni, T. Liu, R. Islamov, F. N. Proske, A. Orvieto, and A. Lucchi. Adaptive methods through the lens of SDEs: Theoretical insights on the role of noise. In The Thirteenth International Conference on Learning Representations , 2025. URL https://openreview. net/forum?id=ww3CLRhF1v . -
- [19][19] A. Damian, T. Ma, and J. D. Lee. Label noise SGD provably prefers flat global minimizers. In Advances in Neural Information Processing Systems , volume 34, pages 2744927461, 2021. URL https://proceedings.neurips.cc/paper\_files/paper/2021/ hash/e6af401c28c1790eaef7d55c92ab6ab6-Abstract.html . -
- [20][20] A. Damian, E. Nichani, and J. D. Lee. Self-stabilization: The implicit bias of gradient descent at the edge of stability. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview.net/forum?id=nhKHA59gXz . -
- [22][22] T. Dozat. Incorporating Nesterov momentum into Adam. ICLR Workshops , 2016. URL https://openreview.net/forum?id=OM0jvwB8jIp57ZJjtNEZ . -
- [23][23] C. L. Ernst Hairer and G. Wanner. Geometric numerical integration . Springer-Verlag, Berlin, 2 edition, 2006. ISBN 3-540-30663-3. -
- [25][25] P. Foret, A. Kleiner, H. Mobahi, and B. Neyshabur. Sharpness-aware minimization for efficiently improving generalization. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=6Tm1mposlrM . -
- [26][26] A. Ghosh, H. Lyu, X. Zhang, and R. Wang. Implicit regularization in heavy-ball momentum accelerated stochastic gradient descent. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview.net/forum?id=ZzdBhtEH9yB . -
- [29][29] S. Gunasekar, J. Lee, D. Soudry, and N. Srebro. Characterizing implicit bias in terms of optimization geometry. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 1832-1841. PMLR, 2018. URL https://proceedings.mlr.press/v80/gunasekar18a.html . -
- [30][30] V. Gupta, T. Koren, and Y. Singer. Shampoo: Preconditioned stochastic tensor optimization. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 1842-1850. PMLR, 10-15 Jul 2018. URL https://proceedings.mlr.press/v80/gupta18a.html . -
- [33][33] Z. Ji and M. Telgarsky. The implicit bias of gradient descent on nonseparable data. In Proceedings of the Thirty-Second Conference on Learning Theory , volume 99 of Proceedings of Machine Learning Research , pages 1772-1798. PMLR, 2019. URL https://proceedings. mlr.press/v99/ji19a.html . arXiv -
- [34][34] Y. Jiang*, B. Neyshabur*, H. Mobahi, D. Krishnan, and S. Bengio. Fantastic generalization measures and where to find them. In International Conference on Learning Representations , 2020. URL https://openreview.net/forum?id=SJgIPJBFvH . -
- [37][37] N. B. Kovachki and A. M. Stuart. Continuous time analysis of momentum methods. Journal of Machine Learning Research , 22(17):1-40, 2021. URL http://jmlr.org/papers/v22/ 19-466.html . -
- [39][39] A. Krizhevsky, I. Sutskever, and G. E. Hinton. Imagenet classification with deep convolutional neural networks. In Advances in Neural Information Processing Systems , volume 25, 2012. URL https://proceedings.neurips.cc/paper\_files/paper/2012/ hash/c399862d3b9d6b76c8436e924a68c45b-Abstract.html . -
- [41][41] F. Kunstner, J. Chen, J. W. Lavington, and M. Schmidt. Noise is not the main factor behind the gap between SGD and Adam on Transformers, but sign descent might be. In The Eleventh International Conference on Learning Representations , 2023. URL https://openreview. net/forum?id=a65YK0cqH8g . -
- [42][42] Q. Li, C. Tai, and W. E. Stochastic modified equations and adaptive stochastic gradient algorithms. In Proceedings of the 34th International Conference on Machine Learning , volume 70 of Proceedings of Machine Learning Research , pages 2101-2110. PMLR, 2017. URL https://proceedings.mlr.press/v70/li17f.html . -
- [43][43] Y. Li, C. Wei, and T. Ma. Towards explaining the regularization effect of initial large learning rate in training neural networks. In Advances in Neural Information Processing Systems , volume 32, 2019. URL https://proceedings.neurips.cc/paper\_files/paper/2019/ hash/bce9abf229ffd7e570818476ee5d7dde-Abstract.html . -
- [44][44] Z. Li, T. Wang, and S. Arora. What happens after SGD reaches zero loss? -a mathematical framework. In International Conference on Learning Representations , 2022. URL https: //openreview.net/forum?id=siCt4xZn5Ve . -
- [46][46] I. Loshchilov and F. Hutter. Decoupled weight decay regularization. In International Conference on Learning Representations , 2019. URL https://openreview.net/forum?id= Bkg6RiCqY7 . -
- [47][47] C. Ma, L. Wu, and W. E. A qualitative study of the dynamic behavior for adaptive gradient algorithms. In Proceedings of the 2nd Mathematical and Scientific Machine Learning Conference , volume 145 of Proceedings of Machine Learning Research , pages 671-692. PMLR, 2022. URL https://proceedings.mlr.press/v145/ma22a.html . -
- [49][49] S. Merity, C. Xiong, J. Bradbury, and R. Socher. Pointer sentinel mixture models. In International Conference on Learning Representations , 2017. URL https://openreview.net/ forum?id=Byj72udxe . -
- [50][50] T. Miyagawa. Toward equation of motion for deep neural networks: Continuoustime gradient descent and discretization error analysis. In Advances in Neural Information Processing Systems , volume 35, pages 37778-37791, 2022. URL https://proceedings.neurips.cc/paper\_files/paper/2022/hash/ f6499ab2a923fa691accdc0077af9677-Abstract-Conference.html . -
- [51][51] M. S. Nacson, J. Lee, S. Gunasekar, P. H. P. Savarese, N. Srebro, and D. Soudry. Convergence of gradient descent on separable data. In Proceedings of the Twenty-Second International Conference on Artificial Intelligence and Statistics , volume 89 of Proceedings of Machine Learning Research , pages 3420-3428. PMLR, 2019. URL https://proceedings.mlr. press/v89/nacson19b.html . -
- [52][52] M. S. Nacson, N. Srebro, and D. Soudry. Stochastic gradient descent on separable data: Exact convergence with a fixed learning rate. In Proceedings of the Twenty-Second International Conference on Artificial Intelligence and Statistics , volume 89 of Proceedings of Machine Learning Research , pages 3051-3059. PMLR, 2019. URL https://proceedings.mlr. press/v89/nacson19a.html . -
- [53][53] Y. Nesterov. A method for solving the convex programming problem with convergence rate o ( 1 k 2 ) . In Dokl. Akad. Nauk SSSR , volume 269, pages 543-547, 1983. -
- [55][55] Q. Qian and X. Qian. The implicit bias of AdaGrad on separable data. In Advances in Neural Information Processing Systems , volume 32, 2019. URL https://proceedings.neurips.cc/ paper\_files/paper/2019/hash/3335881e06d4d23091389226225e17c7-Abstract. html . -
- [56][56] M. Rosca, Y. Wu, C. Qin, and B. Dherin. On a continuous time model of gradient descent dynamics and instability in deep learning. Transactions on Machine Learning Research , 2023. ISSN 2835-8856. URL https://openreview.net/forum?id=EYrRzKPinA . -
- [57][57] N. Shazeer and M. Stern. Adafactor: Adaptive learning rates with sublinear memory cost. In Proceedings of the 35th International Conference on Machine Learning , volume 80 of Proceedings of Machine Learning Research , pages 4596-4604. PMLR, 2018. URL https: //proceedings.mlr.press/v80/shazeer18a.html . -
- [59][59] S. L. Smith, B. Dherin, D. Barrett, and S. De. On the origin of implicit regularization in stochastic gradient descent. In International Conference on Learning Representations , 2021. URL https://openreview.net/forum?id=rq\_Qr0c1Hyo . -
- [60][60] D. Soudry, E. Hoffer, and N. Srebro. The implicit bias of gradient descent on separable data. In International Conference on Learning Representations , 2018. URL https://openreview. net/forum?id=r1q7n9gAb . -
- [61][61] T. Tieleman, G. Hinton, et al. Lecture 6.5-rmsprop: Divide the gradient by a running average of its recent magnitude. COURSERA: Neural networks for machine learning , 4(2): 26-31, 2012. URL https://www.cs.toronto.edu/ ~ tijmen/csc321/slides/lecture\_ slides\_lec6.pdf . -
- [62][62] P. Villalobos, A. Ho, J. Sevilla, T. Besiroglu, L. Heim, and M. Hobbhahn. Position: Will we run out of data? limits of LLM scaling based on human-generated data. In Forty-first International Conference on Machine Learning , 2024. URL https://openreview.net/ forum?id=ViZcgDQjyG . -
- [63][63] B. Wang, Q. Meng, H. Zhang, R. Sun, W. Chen, Z.-M. Ma, and T.-Y. Liu. Does momentum change the implicit regularization on separable data? In Advances in Neural Information Processing Systems , volume 35, pages 26764-26776, 2022. URL https://proceedings.neurips.cc/paper\_files/paper/2022/hash/ ab3f6bbe121a8f7a0263a9b393000741-Abstract-Conference.html . -
- [64][64] K. Wen, T. Ma, and Z. Li. How does sharpness-aware minimization minimizes sharpness? In OPT 2022: Optimization for Machine Learning (NeurIPS 2022 Workshop) , 2022. URL https://openreview.net/forum?id=7H4YznLAwAj . -
- [66][66] S. Xie and Z. Li. Implicit bias of AdamW: ℓ ∞ -norm constrained optimization. In R. Salakhutdinov, Z. Kolter, K. Heller, A. Weller, N. Oliver, J. Scarlett, and F. Berkenkamp, editors, Proceedings of the 41st International Conference on Machine Learning , volume 235 of Proceedings of Machine Learning Research , pages 54488-54510. PMLR, 2024. URL https://proceedings.mlr.press/v235/xie24e.html . -
- [68][68] J. Zhang, S. P. Karimireddy, A. Veit, S. Kim, S. Reddi, S. Kumar, and S. Sra. Why are adaptive methods good for attention models? In Advances in Neural Information Processing Systems , volume 33, pages 15383-15393, 2020. URL https://proceedings.neurips.cc/paper/ 2020/hash/b05b57f6add810d3b7490866d74c0053-Abstract.html . -
- [69][69] R. Zhao, D. Morwani, D. Brandfonbrener, N. Vyas, and S. M. Kakade. Deconstructing what makes a good optimizer for autoregressive language models. In The Thirteenth International Conference on Learning Representations , 2025. URL https://openreview.net/forum? id=zfeso8ceqr .
Cited by
Still being checked.
- Authors
- Cattaneo, M., & Shigida, B.
- Affiliations
- Princeton University (US)
- ORCID
- Cattaneo, Matias
- Venue
- Advances in Neural Information Processing Systems 38
- Published
- 2025
- Pages
- 173262-173299
- ISSN
- 1049-5258 (ISSN-L)
- OpenAlex
- W4407185734
- Cited by
- 0
- References
- 0
- FWCI
- 0.00
- Type
- conference paper
- Language
- en
- Open access
- green
- arxiv.org
- Licence
- cc-by
- Locations
- neural information processing systems (publishedVersion)
- arXiv (Cornell University) (submittedVersion, cc-by)
- arXiv (Cornell University) (cc-by)
- Funders
- National Science Foundation (2210561, 2241575)
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Stochastic Gradient Optimization Techniques
- Physical Sciences › Mathematics › Numerical Analysis › Advanced Optimization Algorithms Research
- Physical Sciences › Computer Science › Artificial Intelligence › Machine Learning and Data Classification
- Keywords
- implicit memory, optimization algorithms, gradient descent with momentum, Lion optimizer, AdamW, implicit regularization, generalization, deep learning optimization
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-11-05.
Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-11-05.
Cited sources are checked by gpt-6-sol.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review