SAM as an Optimal Relaxation of Bayes
Not on the map. Why?
Abstract (arXiv)
Sharpness-aware minimization (SAM) and related adversarial deep-learning methods can drastically improve generalization, but their underlying mechanisms are not yet fully understood. Here, we establish SAM as a relaxation of the Bayes objective where the expected negative-loss is replaced by the optimal convex lower bound, obtained by using the so-called Fenchel biconjugate. The connection enables a new Adam-like extension of SAM to automatically obtain reasonable uncertainty estimates, while sometimes also improving its accuracy. By connecting adversarial and Bayesian methods, our work opens a new path to robustness.
Key facts. Cited by 41 papers in the map (data through 14 Oct 2026). Builds on 33 of its 61 references that are in the map. First posted on arXiv 4 Oct 2022.
Builds on
33 of 61 references.
References · 33
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016CVXPY: A Python-Embedded Modeling Language for Convex Optimizationcited by 641
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 560
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 533
- 2019Fantastic Generalization Measures and Where to Find Themcited by 513
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 359
- 2019Lookahead Optimizer: k steps forward, 1 step backcited by 336
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 308
21 more
- 2017Stochastic Gradient Descent as Approximate Bayesian Inferencecited by 300
- 2021ASAM: Adaptive Sharpness-Aware Minimization for Scale-Invariant Learning of Deep Neural Networkscited by 293
- 2017Noisy Networks for Explorationcited by 266
- 2021When Vision Transformers Outperform ResNets without Pre-training or Strong Data Augmentationscited by 260
- 2021User-friendly introduction to PAC-Bayes boundscited by 239
- 2017Parameter Space Noise for Explorationcited by 161
- 2019Practical Deep Learning with Bayesian Principlescited by 158
- 2017A Bayesian Perspective on Generalization and Stochastic Gradient Descentcited by 135
- 2018Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adamcited by 131
- 2021Bayesian Neural Network Priors Revisitedcited by 121
- 2021Sharpness-Aware Minimization Improves Language Model Generalizationcited by 108
- 2021The Bayesian Learning Rulecited by 99
- 2022Fisher SAM: Information Geometry and Sharpness Aware Minimisationcited by 84
- 2020Regularizing Neural Networks via Adversarial Model Perturbationcited by 78
- 2022Low-Pass Filtering SGD for Recovering Flat Optima in the Deep Learning Optimization Landscapecited by 39
- 2017Deep Relaxation: partial differential equations for optimizing deep neural networkscited by 27
- 2020Training Binary Neural Networks using the Bayesian Learning Rulecited by 23
- 2022Wide Mean-Field Bayesian Neural Networks Ignore the Datacited by 22
- 2017The proximal point method revisitedcited by 9
- 2017Variational Adaptive-Newton Method for Explorative Learningcited by 7
- 2021Lifting the Convex Conjugate in Lagrangian Relaxations: A Tractable Approach for Continuous Markov Random Fieldscited by 1
Built on by
Citing papers per year: 2022: 1; 2023: 12; 2024: 11; 2025: 9; 2026: 8.
41 papers.
Sort:
By area:
- 2024Variational Learning is Effective for Large Deep Networkscited by 65
- 2023Why Does Little Robustness Help? A Further Step Towards Understanding Adversarial Transferabilitycited by 32
- 2023Normalization Layers Are All That Sharpness-Aware Minimization Needscited by 27
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 27
- 2023Practical Sharpness-Aware Minimization Cannot Converge All the Way to Optimacited by 20
- 2022Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspectivecited by 15
- 2023Decentralized SGD and Average-direction SAM are Asymptotically Equivalentcited by 15
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overheadcited by 14
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalizationcited by 13
- 2023The Interpolating Information Criterion for Overparameterized Modelscited by 12
- 2025Uncertainty-Aware Decoding with Minimum Bayes Riskcited by 11
- 2023Flat Seeking Bayesian Neural Networkscited by 10
29 more
- 2023TRAM: Bridging Trust Regions and Sharpness Aware Minimizationcited by 6
- 2023Entropy-MCMC: Sampling from Flat Basins with Easecited by 6
- 2023Optimal Transport Model Distributional Robustnesscited by 5
- 2024Improving Generalization with Flat Hilbert Bayesian Inferencecited by 5
- 2024Improving SAM Requires Rethinking its Optimization Formulationcited by 4
- 2023The Lie-Group Bayesian Learning Rulecited by 3
- 2025Promoting Ensemble Diversity with Interactive Bayesian Distributional Robustness for Fine-tuning Foundation Modelscited by 3
- 2025Branch, or Layer? Zeroth-Order Optimization for Continual Learning of Vision-Language Modelscited by 3
- 2025Federated ADMM from Bayesian Dualitycited by 3
- 2026Variational Routing: A Scalable Bayesian Framework for Calibrated Mixture-of-Experts Transformerscited by 3
- 2026Too Sharp, Too Sure: When Calibration Follows Curvaturecited by 3
- 2024Agnostic Sharpness-Aware Minimizationcited by 2
- 2024Sharpness-Aware Black-Box Optimizationcited 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
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Explorationcited by 2
- 2024Flat Posterior Does Matter For Bayesian Model Averagingcited by 1
- 2024Convergence of Sharpness-Aware Minimization Algorithms using Increasing Batch Size and Decaying Learning Ratecited by 1
- 2026Parameter Exploration for RLVR via Variational Learningcited by 1
- 2023Sharpness-Aware Teleportation on Riemannian Manifoldscited by 0
- 2024Flatness Improves Backbone Generalisation in Few-shot Classificationcited by 0
- 2024Forget Sharpness: Perturbed Forgetting of Model Biases Within SAM Dynamicscited by 0
- 2025Variational Learning Induces Adaptive Label Smoothingcited by 0
- 2025A note on the Artstein-Avidan-Milman's generalized Legendre transformscited by 0
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizerscited by 0
- 2026Sharpness-Aware Hybrid Model Learning for Architecture-Agnostic Parameter Estimationcited by 0
- 2026Incorruptible Neural Networks: Training Models that can Generalize to Large Internal Perturbationscited by 0
- 2026OmniOpt: Taxonomy, Geometry, and Benchmarking of Modern Optimizerscited by 0
- 2026A Generalization of Amari's Bayesian Dualitycited by 0
- 2026A Generalization of Amari's Bayesian Duality8 Sep 2026
- 2026Parameter Exploration for RLVR via Variational Learning10 Aug 2026
- 2026OmniOpt: Taxonomy, Geometry, and Benchmarking of Modern Optimizers4 Jul 2026
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Exploration8 Jun 2026
- 2026Too Sharp, Too Sure: When Calibration Follows Curvature22 Apr 2026
- 2026Variational Routing: A Scalable Bayesian Framework for Calibrated Mixture-of-Experts Transformers10 Mar 2026
- 2026Incorruptible Neural Networks: Training Models that can Generalize to Large Internal Perturbations7 Feb 2026
- 2026Sharpness-Aware Hybrid Model Learning for Architecture-Agnostic Parameter Estimation6 Feb 2026
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizers26 Sep 2025
- 2025Unveiling m-Sharpness Through the Structure of Stochastic Gradient Noise22 Sep 2025
- 2025A note on the Artstein-Avidan-Milman's generalized Legendre transforms28 Jul 2025
- 2025Federated ADMM from Bayesian Duality16 Jun 2025
29 more
- 2025Branch, or Layer? Zeroth-Order Optimization for Continual Learning of Vision-Language Models14 Jun 2025
- 2025Promoting Ensemble Diversity with Interactive Bayesian Distributional Robustness for Fine-tuning Foundation Models8 Jun 2025
- 2025Towards Understanding The Calibration Benefits of Sharpness-Aware Minimization29 May 2025
- 2025Uncertainty-Aware Decoding with Minimum Bayes Risk7 Mar 2025
- 2025Variational Learning Induces Adaptive Label Smoothing11 Feb 2025
- 2024Sharpness-Aware Black-Box Optimization16 Oct 2024
- 2024Improving Generalization with Flat Hilbert Bayesian Inference5 Oct 2024
- 2024Convergence of Sharpness-Aware Minimization Algorithms using Increasing Batch Size and Decaying Learning Rate16 Sep 2024
- 2024Improving SAM Requires Rethinking its Optimization Formulation17 Jul 2024
- 2024Flat Posterior Does Matter For Bayesian Model Averaging21 Jun 2024
- 2024Agnostic Sharpness-Aware Minimization11 Jun 2024
- 2024Forget Sharpness: Perturbed Forgetting of Model Biases Within SAM Dynamics10 Jun 2024
- 2024Flatness Improves Backbone Generalisation in Few-shot Classification11 Apr 2024
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalization30 Mar 2024
- 2024Variational Learning is Effective for Large Deep Networks27 Feb 2024
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overhead22 Jan 2024
- 2023Entropy-MCMC: Sampling from Flat Basins with Ease9 Oct 2023
- 2023TRAM: Bridging Trust Regions and Sharpness Aware Minimization5 Oct 2023
- 2023Sharpness-Aware Teleportation on Riemannian Manifolds29 Sep 2023
- 2023A Primer on Bayesian Neural Networks: Review and Debates28 Sep 2023
- 2023Why Does Little Robustness Help? A Further Step Towards Understanding Adversarial Transferability15 Jul 2023
- 2023The Interpolating Information Criterion for Overparameterized Models15 Jul 2023
- 2023Practical Sharpness-Aware Minimization Cannot Converge All the Way to Optima16 Jun 2023
- 2023Normalization Layers Are All That Sharpness-Aware Minimization Needs7 Jun 2023
- 2023Optimal Transport Model Distributional Robustness7 Jun 2023
- 2023Decentralized SGD and Average-direction SAM are Asymptotically Equivalent5 Jun 2023
- 2023The Lie-Group Bayesian Learning Rule8 Mar 2023
- 2023Flat Seeking Bayesian Neural Networks6 Feb 2023
- 2022Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspective24 Nov 2022
- 2024Sharpness-Aware Black-Box Optimizationcited by 2
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Explorationcited by 2
- 2026SHAPO: Sharpness-Aware Policy Optimization for Safe Exploration8 Jun 2026
- 2024Sharpness-Aware Black-Box Optimization16 Oct 2024
- 2024Variational Learning is Effective for Large Deep Networkscited by 65
- 2023Why Does Little Robustness Help? A Further Step Towards Understanding Adversarial Transferabilitycited by 32
- 2023Normalization Layers Are All That Sharpness-Aware Minimization Needscited by 27
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 27
- 2023Practical Sharpness-Aware Minimization Cannot Converge All the Way to Optimacited by 20
- 2022Beyond Losses Reweighting: Empowering Multi-Task Learning via the Generalization Perspectivecited by 15
- 2023Decentralized SGD and Average-direction SAM are Asymptotically Equivalentcited by 15
- 2024Momentum-SAM: Sharpness Aware Minimization without Computational Overheadcited by 14
- 2024Revisiting Random Weight Perturbation for Efficiently Improving Generalizationcited by 13
- 2023The Interpolating Information Criterion for Overparameterized Modelscited by 12
- 2025Uncertainty-Aware Decoding with Minimum Bayes Riskcited by 11
- 2023Flat Seeking Bayesian Neural Networkscited by 10
12 most cited of 37 All 37 (JSON)
- 2026A Generalization of Amari's Bayesian Duality8 Sep 2026
- 2026Parameter Exploration for RLVR via Variational Learning10 Aug 2026
- 2026OmniOpt: Taxonomy, Geometry, and Benchmarking of Modern Optimizers4 Jul 2026
- 2026Too Sharp, Too Sure: When Calibration Follows Curvature22 Apr 2026
- 2026Incorruptible Neural Networks: Training Models that can Generalize to Large Internal Perturbations7 Feb 2026
- 2026Sharpness-Aware Hybrid Model Learning for Architecture-Agnostic Parameter Estimation6 Feb 2026
- 2025Sharpness-Aware Minimization Can Hallucinate Minimizers26 Sep 2025
- 2025Unveiling m-Sharpness Through the Structure of Stochastic Gradient Noise22 Sep 2025
- 2025A note on the Artstein-Avidan-Milman's generalized Legendre transforms28 Jul 2025
- 2025Federated ADMM from Bayesian Duality16 Jun 2025
- 2025Promoting Ensemble Diversity with Interactive Bayesian Distributional Robustness for Fine-tuning Foundation Models8 Jun 2025
- 2025Towards Understanding The Calibration Benefits of Sharpness-Aware Minimization29 May 2025
12 newest of 37 All 37 (JSON)