University of Tübingen and Tübingen AI Center; McGill University and Mila - Quebec AI Institute
Maximilian Mueller, Tiffany Vlaar, David Rolnick, Matthias Hein
sharpness-aware minimization;flatness;generalization;normalization layers
Normalization Layers Are All That Sharpness-Aware Minimization Needs