Sharpness-aware Minimization for Efficiently Improving Generalization

Pierre Foret, Ariel Kleiner, Hossein Mobahi, Behnam Neyshabur

Keywords: Sharpness Minimization, Generalization, Regularization, Training Method, Deep Learning

Abstract: In today's heavily overparameterized models, the value of the training loss provides few guarantees on model generalization ability. Indeed, optimizing only the training loss value, as is commonly done, can easily lead to suboptimal model quality. Motivated by the connection between geometry of the loss landscape and generalization---including a generalization bound that we prove here---we introduce a novel, effective procedure for instead simultaneously minimizing loss value and loss sharpness. In particular, our procedure, Sharpness-Aware Minimization (SAM), seeks parameters that lie in neighborhoods having uniformly low loss; this formulation results in a min-max optimization problem on which gradient descent can be performed efficiently. We present empirical results showing that SAM improves model generalization across a variety of benchmark datasets (e.g., CIFAR-{10, 100}, ImageNet, finetuning tasks) and models, yielding novel state-of-the-art performance for several. Additionally, we find that SAM natively provides robustness to label noise on par with that provided by state-of-the-art procedures that specifically target learning with noisy labels.

One-sentence Summary: Motivated by the connection between geometry of the loss landscape and generalization, we introduce a procedure for simultaneously minimizing loss value and loss sharpness.

Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics

Code: [ google-research/sam](https://github.com/google-research/sam) + [ 12 community implementations](https://paperswithcode.com/paper/?openreview=6Tm1mposlrM)

Data: [Birdsnap](https://paperswithcode.com/dataset/birdsnap), [CIFAR-100](https://paperswithcode.com/dataset/cifar-100), [FGVC-Aircraft](https://paperswithcode.com/dataset/fgvc-aircraft-1), [Fashion-MNIST](https://paperswithcode.com/dataset/fashion-mnist), [Food-101](https://paperswithcode.com/dataset/food-101), [ImageNet](https://paperswithcode.com/dataset/imagenet), [Oxford 102 Flower](https://paperswithcode.com/dataset/oxford-102-flower), [Oxford-IIIT Pets](https://paperswithcode.com/dataset/oxford-iiit-pets-1), [SVHN](https://paperswithcode.com/dataset/svhn), [Stanford Cars](https://paperswithcode.com/dataset/stanford-cars)

Community Implementations: [ 4 code implementations](https://www.catalyzex.com/paper/arxiv:2010.01412/code)