English

Adam-SHANG: A Convergent Adam-Type Method for Stochastic Smooth Convex Optimization

Optimization and Control 2026-05-14 v1 Machine Learning

Abstract

We propose Adam-SHANG, a Lyapunov-guided Adam-type method that couples momentum, adaptive preconditioning, and a curvature-aware correction through a more stable lagged-preconditioner update. For stochastic smooth convex optimization, we prove convergence in expectation under an admissible stepsize condition that can always be satisfied by a conservative spectral bound, without imposing global monotonicity on the second-moment sequence. To obtain a less conservative practical rule, we introduce a computable trace-ratio stepsize, motivated by a local coordinatewise alignment condition. The same structural update is also tested beyond the convex setting with simplified parameters. Experiments validate the predicted stochastic decay and show competitive training performance against Adam and AdamW on deep learning tasks.

Keywords

Cite

@article{arxiv.2605.12878,
  title  = {Adam-SHANG: A Convergent Adam-Type Method for Stochastic Smooth Convex Optimization},
  author = {Yaxin Yu and Long Chen and Minfu Feng},
  journal= {arXiv preprint arXiv:2605.12878},
  year   = {2026}
}

Comments

25 pages, 13 figures

R2 v1 2026-07-22T07:09:01.057Z