Explaining Near-Zero Hessian Eigenvalues Through Approximate Symmetries in Neural Networks
Summary
A new study explains the origin of near-zero Hessian eigenvalues in neural networks, arguing they stem from weakly lifted pseudo-Goldstone modes of continuous symmetries inherent in network parametrization. While deep linear networks exhibit exact symmetries leading to flat directions and exact zero modes, introducing a ReLU nonlinearity weakly breaks these symmetries. The research demonstrates that high-curvature directions are orthogonal to the symmetry subspace, with the bulk of the Hessian spectrum lying almost entirely within it. This mechanism is validated using a two-layer ReLU student-teacher model, a network trained on CIFAR-10, and a convolutional example, extending the diagnostic beyond fully connected layers. These findings clarify the origin of near-zero modes by linking the Hessian bulk to weakly broken symmetries.
Key takeaway
For AI Scientists analyzing neural network optimization landscapes, understanding that near-zero Hessian eigenvalues arise from weakly broken continuous symmetries is crucial. This insight suggests that many flat directions are not random but structured by network parametrization. You should consider how these approximate symmetries influence training dynamics and generalization, potentially guiding the development of new regularization strategies or optimizers that explicitly account for these low-curvature subspaces to improve model stability and performance.
Key insights
Near-zero Hessian eigenvalues in neural networks stem from weakly broken continuous symmetries.
Principles
- Continuous symmetries generate flat directions.
- ReLU nonlinearities weakly break these symmetries.
- High-curvature directions are orthogonal to symmetry.
Method
Analyze Hessian spectrum by resolving eigenvectors, identifying symmetry subspace, and observing orthogonality of high-curvature directions.
In practice
- Diagnose near-zero modes in fully connected layers.
- Extend diagnostic to convolutional networks.
Topics
- Hessian Eigenvalues
- Neural Network Symmetries
- Loss Landscape
- Optimization Theory
- Deep Learning Theory
- ReLU Networks
Best for: Research Scientist, AI Scientist
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.