Enhancing Sharpness-Aware Optimization Through Variance Suppression
Not on the map. Why?
Abstract (arXiv)
Sharpness-aware minimization (SAM) has well documented merits in enhancing generalization of deep neural networks, even without sizable data augmentation. Embracing the geometry of the loss function, where neighborhoods of 'flat minima' heighten generalization ability, SAM seeks 'flat valleys' by minimizing the maximum loss caused by an adversary perturbing parameters within the neighborhood. Although critical to account for sharpness of the loss function, such an 'over-friendly adversary' can curtail the outmost level of generalization. The novel approach of this contribution fosters stabilization of adversaries through variance suppression (VaSSO) to avoid such friendliness. VaSSO's provable stability safeguards its numerical improvement over SAM in model-agnostic tasks, including image classification and machine translation. In addition, experiments confirm that VaSSO endows SAM with robustness against high levels of label noise.
Key facts. Cited by 37 papers in the map (data through 7 Oct 2026). Builds on 44 of its 57 references that are in the map. First posted on arXiv 27 Sep 2023.
Builds on
44 of 57 references.
References · 44
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2017Improved Regularization of Convolutional Neural Networks with Cutoutcited by 1,623
- 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
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
- 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
32 more
- 2019Fantastic Generalization Measures and Where to Find Themcited by 507
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 357
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 289
- 2017Three Factors Influencing Minima in SGDcited by 270
- 2021When Vision Transformers Outperform ResNets without Pre-training or Strong Data Augmentationscited by 259
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 248
- 2017SARAH: A Novel Method for Machine Learning Problems Using Stochastic Recursive Gradientcited by 218
- 2022Surrogate Gap Minimization Improves Sharpness-Aware Trainingcited by 171
- 2020The Break-Even Point on Optimization Trajectories of Deep Neural Networkscited by 149
- 2020Implicit Gradient Regularizationcited by 145
- 2022Towards Understanding Sharpness-Aware Minimizationcited by 145
- 2021Efficient Sharpness-aware Minimization for Improved Training of Neural Networkscited by 137
- 2022Towards Efficient and Scalable Sharpness-Aware Minimizationcited by 129
- 2022Penalizing Gradient Norm for Efficiently Improving Generalization in Deep Learningcited by 122
- 2023Sharpness-Aware Gradient Matching for Domain Generalizationcited by 102
- 2022Sharpness-Aware Training for Freecited by 95
- 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
- 2022Make Sharpness-Aware Minimization Stronger: A Sparsified Perturbation Approachcited by 70
- 2022How Does Sharpness-Aware Minimization Minimize Sharpness?cited by 64
- 2022The Dynamics of Sharpness-Aware Minimization: Bouncing Across Ravines and Drifting Towards Wide Minimacited by 51
- 2018Stochastic Conditional Gradient Methods: From Convex Minimization to Submodular Maximizationcited by 47
- 2023An Adaptive Policy to Employ Sharpness-Aware Minimizationcited by 43
- 2016Stochastic Frank-Wolfe Methods for Nonconvex Optimizationcited by 40
- 2021On the Generalization of Models Trained with SGD: Information-Theoretic Bounds and Implicationscited by 31
- 2020Understanding and Increasing Efficiency of Frank-Wolfe Adversarial Trainingcited by 14
- 2022GA-SAM: Gradient-Strength based Adaptive Sharpness-Aware Minimization for Improved Generalizationcited by 13
- 2020Accelerated Stochastic Gradient-free and Projection-free Methodscited by 12
- 2019On the Convergence of SARAH and Beyondcited by 10
- 2023Exploring the Effect of Multi-step Ascent in Sharpness-Aware Minimizationcited by 9
- 2019Almost Tune-Free Variance Reductioncited by 7
- 2021Heavy Ball Momentum for Conditional Gradientcited by 7
Built on by
Citing papers per year: 2024: 16; 2025: 14; 2026: 7.
37 papers.
Sort:
By area:
- 2024Friendly Sharpness-Aware Minimizationcited by 52
- 2024Locally Estimated Global Perturbations are Better than Local Perturbations for Federated Sharpness-aware Minimizationcited by 28
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overheadcited by 13
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalizationcited by 13
- 2024Domain-Inspired Sharpness-Aware Minimization Under Domain Shiftscited by 12
- 2025Sharpness-Aware Minimization: General Analysis and Improved Ratescited by 10
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problemscited by 9
- 2025RefLoRA: Refactored Low-Rank Adaptation for Efficient Fine-Tuning of Large Modelscited by 9
- 2024Stabilizing Sharpness-aware Minimization Through A Simple Renormalization Strategycited by 8
- 2024SAMPa: Sharpness-aware Minimization Parallelizedcited by 8
- 2024Efficient Sharpness-Aware Minimization for Molecular Graph Transformer Modelscited by 7
25 more
- 2024Tilted Sharpness-Aware Minimizationcited by 6
- 2025Zeroth-Order Optimization Finds Flat Minimacited by 6
- 2025Towards Robust Influence Functions with Flat Validation Minimacited by 4
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adamcited by 4
- 2025Bi-LoRA: Efficient Sharpness-Aware Minimization for Fine-Tuning Large-Scale Modelscited by 3
- 2025Flat Minima and Generalization: Insights from Stochastic Convex Optimizationcited by 3
- 2024GAQAT: gradient-adaptive quantization-aware training for domain generalizationcited by 2
- 2025Towards Understanding The Calibration Benefits of Sharpness-Aware Minimizationcited by 2
- 2025Unveiling m-Sharpness Through the Structure of Stochastic Gradient Noisecited by 2
- 2026Low-Rank Adaptation Redux for Large Modelscited by 2
- 2024Convergence of Sharpness-Aware Minimization Algorithms using Increasing Batch Size and Decaying Learning Ratecited by 1
- 2024Sharpness-Aware Minimization with Adaptive Regularization for Training Deep Neural Networkscited by 1
- 2025Preconditioned Sharpness-Aware Minimization: Unifying Analysis and a Novel Learning Algorithmcited by 1
- 2025VASSO: Variance Suppression for Sharpness-Aware Minimizationcited by 1
- 2025R'enyi Sharpness: A Novel Sharpness that Strongly Correlates with Generalizationcited by 1
- 2026Sharpness-Aware Minimization in Logit Space Efficiently Enhances Direct Preference Optimizationcited by 1
- 2024Neighborhood and Global Perturbations Supported SAM in Federated Learning: From Local Tweaks To Global Awarenesscited by 0
- 2024Understanding SAM's Robustness to Noisy Labels through Gradient Down-weightingcited by 0
- 2025Communication-Efficient Distributed Training for Collaborative Flat Optima Recovery in Deep Learningcited by 0
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizerscited by 0
- 2025Sharpness-aware Federated Graph Learningcited by 0
- 2026Revisiting Sharpness-Aware Minimization: A More Faithful and Effective Implementationcited by 0
- 2026Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learningcited by 0
- 2026FedVSSAM: Mitigating Flatness Incompatibility in Sharpness-Aware Federated Learningcited by 0
- 2026Navigating Potholes with Geometry-Aware Sharpness Minimizationcited by 0
- 2026Navigating Potholes with Geometry-Aware Sharpness Minimization15 May 2026
- 2026FedVSSAM: Mitigating Flatness Incompatibility in Sharpness-Aware Federated Learning9 May 2026
- 2026Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learning8 May 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2026Sharpness-Aware Minimization in Logit Space Efficiently Enhances Direct Preference Optimization18 Mar 2026
- 2026Revisiting Sharpness-Aware Minimization: A More Faithful and Effective Implementation9 Mar 2026
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adam2 Feb 2026
- 2025Sharpness-aware Federated Graph Learning18 Dec 2025
- 2025Flat Minima and Generalization: Insights from Stochastic Convex Optimization5 Nov 2025
- 2025R'enyi Sharpness: A Novel Sharpness that Strongly Correlates with Generalization9 Oct 2025
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizers26 Sep 2025
- 2025Unveiling m-Sharpness Through the Structure of Stochastic Gradient Noise22 Sep 2025
25 more
- 2025VASSO: Variance Suppression for Sharpness-Aware Minimization2 Sep 2025
- 2025Bi-LoRA: Efficient Sharpness-Aware Minimization for Fine-Tuning Large-Scale Models27 Aug 2025
- 2025Communication-Efficient Distributed Training for Collaborative Flat Optima Recovery in Deep Learning27 Jul 2025
- 2025Zeroth-Order Optimization Finds Flat Minima5 Jun 2025
- 2025Towards Understanding The Calibration Benefits of Sharpness-Aware Minimization29 May 2025
- 2025Towards Robust Influence Functions with Flat Validation Minima25 May 2025
- 2025RefLoRA: Refactored Low-Rank Adaptation for Efficient Fine-Tuning of Large Models24 May 2025
- 2025Sharpness-Aware Minimization: General Analysis and Improved Rates4 Mar 2025
- 2025Preconditioned Sharpness-Aware Minimization: Unifying Analysis and a Novel Learning Algorithm11 Jan 2025
- 2024Sharpness-Aware Minimization with Adaptive Regularization for Training Deep Neural Networks22 Dec 2024
- 2024GAQAT: gradient-adaptive quantization-aware training for domain generalization7 Dec 2024
- 2024Understanding SAM's Robustness to Noisy Labels through Gradient Down-weighting26 Nov 2024
- 2024Tilted Sharpness-Aware Minimization30 Oct 2024
- 2024On the Crucial Role of Initialization for Matrix Factorization24 Oct 2024
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problems18 Oct 2024
- 2024SAMPa: Sharpness-aware Minimization Parallelized14 Oct 2024
- 2024Convergence of Sharpness-Aware Minimization Algorithms using Increasing Batch Size and Decaying Learning Rate16 Sep 2024
- 2024Neighborhood and Global Perturbations Supported SAM in Federated Learning: From Local Tweaks To Global Awareness26 Aug 2024
- 2024Efficient Sharpness-Aware Minimization for Molecular Graph Transformer Models19 Jun 2024
- 2024Locally Estimated Global Perturbations are Better than Local Perturbations for Federated Sharpness-aware Minimization29 May 2024
- 2024Domain-Inspired Sharpness-Aware Minimization Under Domain Shifts29 May 2024
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalization30 Mar 2024
- 2024Friendly Sharpness-Aware Minimization19 Mar 2024
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overhead22 Jan 2024
- 2024Stabilizing Sharpness-aware Minimization Through A Simple Renormalization Strategy14 Jan 2024
- 2024Friendly Sharpness-Aware Minimizationcited by 52
- 2024Locally Estimated Global Perturbations are Better than Local Perturbations for Federated Sharpness-aware Minimizationcited by 28
- 2024On the Crucial Role of Initialization for Matrix Factorizationcited by 15
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overheadcited by 13
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalizationcited by 13
- 2024Domain-Inspired Sharpness-Aware Minimization Under Domain Shiftscited by 12
- 2025Sharpness-Aware Minimization: General Analysis and Improved Ratescited by 10
- 2024Implicit Regularization of Sharpness-Aware Minimization for Scale-Invariant Problemscited by 9
- 2025RefLoRA: Refactored Low-Rank Adaptation for Efficient Fine-Tuning of Large Modelscited by 9
- 2024Stabilizing Sharpness-aware Minimization Through A Simple Renormalization Strategycited by 8
- 2024SAMPa: Sharpness-aware Minimization Parallelizedcited by 8
- 2024Efficient Sharpness-Aware Minimization for Molecular Graph Transformer Modelscited by 7
12 most cited of 36 All 36 (JSON)
- 2026Navigating Potholes with Geometry-Aware Sharpness Minimization15 May 2026
- 2026FedVSSAM: Mitigating Flatness Incompatibility in Sharpness-Aware Federated Learning9 May 2026
- 2026Flatness and Gradient Alignment Are Both Necessary: Spectral-Aware Gradient-Aligned Exploration for Multi-Distribution Learning8 May 2026
- 2026Low-Rank Adaptation Redux for Large Models23 Apr 2026
- 2026Revisiting Sharpness-Aware Minimization: A More Faithful and Effective Implementation9 Mar 2026
- 2026The Effect of Mini-Batch Noise on the Implicit Bias of Adam2 Feb 2026
- 2025Sharpness-aware Federated Graph Learning18 Dec 2025
- 2025Flat Minima and Generalization: Insights from Stochastic Convex Optimization5 Nov 2025
- 2025R'enyi Sharpness: A Novel Sharpness that Strongly Correlates with Generalization9 Oct 2025
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizers26 Sep 2025
- 2025Unveiling m-Sharpness Through the Structure of Stochastic Gradient Noise22 Sep 2025
- 2025VASSO: Variance Suppression for Sharpness-Aware Minimization2 Sep 2025
12 newest of 36 All 36 (JSON)