Fisher-Geometric Sharpness and the Implicit Bias of SGD toward Flat Minima
Not on the map. Why?
Abstract (arXiv)
A widely held intuition in deep learning is that stochastic gradient descent (SGD) implicitly favors flat minima and that flat minima generalize better, but standard Euclidean measures of flatness such as the trace or maximum eigenvalue of the loss Hessian are not invariant under reparametrizations that preserve the network function, which undermines the theoretical foundations of this narrative. In this study we resolve this issue by grounding flatness in the Riemannian geometry of the statistical manifold induced by the Fisher Information Matrix (FIM). We define Riemannian sharpness mathematically and prove that it is invariant under smooth, function-preserving reparametrizations, which directly addresses the critique of Dinh et al. in the paper ``Sharp minima can generalize for deep nets''.We note that this invariance is a property of the true FIM; the diagonal empirical estimator used in practice (and in all experiments below) inherits invariance only approximately, and exact invariance under arbitrary reparametrizations would require structured estimators such as K-FAC. We formalize the gradient noise of mini-batch SGD as having a covariance structure proportional to the FIM, derive the stationary distribution of the resulting stochastic differential equation, and then show that the probability mass is exponentially concentrated at Riemannian-flat minima. A PAC-Bayes generalization bound controlled explicitly by SR formally links this geometric bias to test performance. Our experiments on MNIST and CIFAR-10 confirm that SR reliably tracks generalization in ways that Euclidean sharpness does not, and that its scaling with η/B matches the theoretical predictions. Together these results provide a rigorous, reparametrization-invariant account of why flat minima generalize.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 12 of its 18 references that are in the map. First posted on arXiv 18 Jun 2026.
Builds on
12 of 18 references.
References · 12
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,417
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 796
- 2017Exploring Generalization in Deep Learningcited by 600
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2017Sharp Minima Can Generalize For Deep Netscited by 466
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 359
- 2019Limitations of the Empirical Fisher Approximation for Natural Gradient Descentcited by 217
- 2013Revisiting Natural Gradient for Deep Networkscited by 159
- 2015Stochastic modified equations and adaptive stochastic gradient algorithmscited by 142
- 2021Implicit Bias of SGD for Diagonal Linear Networks: a Provable Benefit of Stochasticitycited by 110
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
Built on by
Citing papers per year: 2026: 1.
1 paper.