Explaining Near-Zero Hessian Eigenvalues Through Approximate Symmetries in Neural Networks

· Source: Machine Learning · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Mathematics & Computational Sciences · Depth: Expert, quick

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

Method

Analyze Hessian spectrum by resolving eigenvectors, identifying symmetry subspace, and observing orthogonality of high-curvature directions.

In practice

Topics

Best for: Research Scientist, AI Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Machine Learning.