arXiv · 2210.01620
SAM as an Optimal Relaxation of Bayes
Abstract
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.
Explore related subjects
Keep this discovery
Thomas Möllenhoff, Mohammad Emtiyaz Khan. 2022-10-04. SAM as an Optimal Relaxation of Bayes. https://arxiv.org/abs/2210.01620
Cite the original work for its findings. Save a collection to share your selection of sources.