[2605.15522] Stochastic Non-Smooth Convex Optimization with Unbounded Gradients
Abstract:Much of the existing theory on first-order non-smooth optimization is built on a restrictive assumption that the gradients of the objective function are uniformly bounded. We introduce a much more realistic class of generalized Lipschitz functions, where the gradient norms are bounded by an affine function of the optimality gap. We then ask a natural question: what algorithm achieves the best global convergence rates for solving convex stochastic generalized Lipschitz optimization problems? To address this, we develop a new convergence analysis for several existing algorithms and find that AdamW with clipped updates, theoretically outperforms other popular stochastic optimization methods, such as SGD and AdaGrad. Moreover, our analysis establishes the critical role of AdamW's exponentially weighted gradient accumulation, as opposed to simple averaging. We further show that clipped AdamW is universal and achieves improved rates under the popular generalized smoothness assumption, analyze the convergence of clipped AdamW with diagonal and matrix preconditioners, and extend our results to the quasar-convex setting.
# link_29lh58vpr20.pdf ## Metadata - PDFFormatVersion=1.5 - IsLinearized=false - IsAcroFormPresent=false - IsXFAPresent=false - IsCollectionPresent=false - IsSignaturesPresent=false - Author=Dmitry Kovalev - Creator=arXiv GenPDF (tex2pdf:a6404ea) - Custom.DOI=https://doi.org/10.48550/arXiv.2605.15522 - Custom.License=http://arxiv.org/licenses/nonexclusive-distrib/1.0/ - Custom.PTEX.Fullbanner=This is pdfTeX, Version 3.141592653-2.6-1.40.25 (TeX Live 2023) kpathsea version 6.3.5 - Custom.arXivID=https://arxiv.org/abs/2605.15522v2 - Producer=pikepdf 8.15.1 - Title=Stochastic Non-Smooth Convex Op
Explore this link on the map →saved by
related reading
- [2101.12176] On the Origin of Implicit Regularization in Stochastic Gradient Descentarxiv.org
- [2605.01172] A Theory of Generalization in Deep Learningarxiv.org
- Stochastic gradient descent - Wikipediaen.m.wikipedia.org
- Why Momentum Really Worksdistill.pub
- Effortless optimization through gradient flows – Machine Learning Research Blogfrancisbach.com
- AdaGrad - Cornell University Computational Optimization Open Textbook - Optimization Wikioptimization.cbe.cornell.edu
- A minimizer Far, Far Away – Parameter-free Learning and Optimization Algorithmsparameterfree.com
- Modular Manifolds - Thinking Machines Labthinkingmachines.ai
- Scaling laws of optimization – Machine Learning Research Blogfrancisbach.com
- [1512.04202] Preconditioned Stochastic Gradient Descentarxiv.org
- An overview of gradient descent optimization algorithmsruder.io
- prox_algs.pdfweb.stanford.edu