Marvin F. da Silva

Hide & Seek: Transformer Symmetries Obscure Sharpness & Riemannian Geometry Finds It

The concept of sharpness has been successfully applied to traditional architectures like MLPs and CNNs to predict their generalization. For transformers, however, recent work reported weak correlation between flatness and generalization. We argue that existing sharpness measures fail for transformer…