On the Implicit Flatness Bias of Sharpness-Aware Minimization: A Linear Stability Analysis with Quantitative Hyperparameter Bounds
Not on the map. Why?
Abstract (arXiv)
Sharpness-Aware Minimization (SAM) improves generalization by seeking parameters whose loss is robust to local adversarial perturbations, but the quantitative mechanism underlying its implicit bias toward flat minima remains unclear. In particular, the perturbation radius ρ is typically treated as an isolated tuning parameter, despite defining the neighborhood in which SAM measures sharpness. We analyze mini-batch SAM near an interpolating minimum through linear stability. Under local linearization and gradient-noise alignment assumptions, we prove that every linearly stable minimum satisfies λmax≤√[3]bGamma/(2ρη2), where λmax is the largest Hessian eigenvalue, b is the batch size, η is the learning rate, and Gamma bounds the gradient norm. The bound quantitatively characterizes SAM's implicit flatness bias: holding the other quantities fixed, a smaller batch size, a larger learning rate, or a larger radius restricts linearly stable SAM to flatter minima. It also exposes a necessary trade-off: ρ should be large enough to promote flatness, yet remain local enough to preserve the approximation and stable training. We validate this prediction in a controlled study of 900 models on CIFAR-100 with ResNet-18 and VGG-19, where increasing ρ is consistently associated with a smaller largest Hessian eigenvalue across batch-size and learning-rate settings. Finally, we instantiate the analysis in Taylor-Locality Controlled SAM (TLC-SAM), which adjusts ρ using the observed Taylor-approximation error and further reduces the top Hessian eigenvalue relative to fixed-radius SAM. Our results provide quantitative hyperparameter bounds and a stability--locality perspective for analyzing and designing SAM variants.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 31 of its 40 references that are in the map. First posted on arXiv 4 Aug 2026.
Builds on
31 of 40 references.
References · 31
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2016Wide Residual Networkscited by 3,057
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
- 2017Exploring Generalization in Deep Learningcited by 595
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2017Sharp Minima Can Generalize For Deep Netscited by 462
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 357
- 2017Implicit Regularization in Matrix Factorizationcited by 318
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
19 more
- 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 291
- 2017Three Factors Influencing Minima in SGDcited by 270
- 2016Deep Pyramidal Residual Networkscited by 163
- 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 82
- 2022Make Sharpness-Aware Minimization Stronger: A Sparsified Perturbation Approachcited by 70
- 2022The alignment property of SGD noise and how it helps select flat minima: A stability analysiscited by 64
- 2021On Linear Stability of SGD and Input-Smoothness of Neural Networkscited by 61
- 2022The Dynamics of Sharpness-Aware Minimization: Bouncing Across Ravines and Drifting Towards Wide Minimacited by 51
- 2023Sharpness Minimization Algorithms Do Not Only Minimize Sharpness To Achieve Better Generalizationcited by 39
- 2023The Crucial Role of Normalization in Sharpness-Aware Minimizationcited by 26
- 2024Sharpness-Aware Minimization Efficiently Selects Flatter Minima Late in Trainingcited by 24
- 2023An SDE for Modeling SAM: Theory and Insightscited by 22
- 2025Explicit Eigenvalue Regularization Improves Sharpness-Aware Minimizationcited by 21
- 2023Practical Sharpness-Aware Minimization Cannot Converge All the Way to Optimacited by 20
- 2023Sharpness-Aware Minimization and the Edge of Stabilitycited by 16
- 2023Stability Analysis of Sharpness-Aware Minimizationcited by 14
- 2023mSAM: Micro-Batch-Averaged Sharpness-Aware Minimizationcited by 10
- 2025Sharpness-Aware Minimization: General Analysis and Improved Ratescited by 10
Built on by
Not cited yet.