Rethinking Sharpness-Aware Minimization as Variational Inference

Szilvia Ujváry, Zsigmond Telek, Anna Kerekes +2

Sharpness-aware minimization (SAM) aims to improve the generalisation of gradient-based learning by seeking out flat minima. In this work, we establish connections between SAM and…