Lune

NeurIPS2023Top-tier venue

Practical Sharpness-Aware Minimization Cannot Converge All the Way to Optima

Dongkuk Si, Chulhee Yun

2023Year
34Citations
21Top-tier citations

Abstract

Sharpness-Aware Minimization (SAM) is an optimizer that takes a descent step based on the gradient at a perturbation yt=xt+ρ∇f(xt)∥∇f(xt)∥y_t = x_t + \rho \frac{\nabla f(x_t)}{\lVert \nabla f(x_t) \rVert} of the current point xtx_t. Existing studies prove convergence of SAM for smooth functions, but they do so by assuming decaying perturbation size ρ\rho and/or no gradient normalization in yty_t, which is detached from practice. To address this gap, we study deterministic/stochastic versions of SAM with practical configurations (i.e., constant ρ\rho and gradient normalization in yty_t) and explore their convergence properties on smooth functions with (non)convexity assumptions. Perhaps surprisingly, in many scenarios, we find out that SAM has limited capability to converge to global minima or stationary points. For smooth strongly convex functions, we show that while deterministic SAM enjoys tight global convergence rates of Θ~(1T2)\tilde \Theta(\frac{1}{T^2}), the convergence bound of stochastic SAM suffers an inevitable additive term O(ρ2)O(\rho^2), indicating convergence only up to neighborhoods of optima. In fact, such O(ρ2)O(\rho^2) factors arise for stochastic SAM in all the settings we consider, and also for deterministic SAM in nonconvex cases; importantly, we prove by examples that such terms are unavoidable. Our results highlight vastly different characteristics of SAM with vs. without decaying perturbation size or gradient normalization, and suggest that the intuitions gained from one version may not apply to the other.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext ca83106e-829b-42e4-8478-b46512d1677a

Cited by top-tier papers21

Ask how each one uses it

Builds on23

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines