Finite Spectral Depth Controls Gaussian Entrance Rank and Avoidance of the First Gradient-Flow Saddle
Yanlin Long
Abstract
Matrix factorization provides a simple model for studying the training dynamics of neural networks. Recent studies of gradient flow under small initialization have analyzed how trajectories escape from a neighborhood of the origin and approach their first nonzero stationary point. In this paper, we study continuous-time dynamics that applies a finite number of Newton-Schulz iterations to the normalized matrix gradient (Muon). We show that the number of Newton-Schulz steps can change the rank selected during the early stages of training and allow the resulting trajectory to bypass the first nonzero stationary point approached by gradient flow. Under an additional residual-inertia condition, we prove that bypass occurs for all sufficiently large finite Newton-Schulz depths and provide an explicit sufficient threshold. We also construct an example in which the bypass threshold is four.
Create a lesson
Related papers
Molecular Prior Networks
Pawel Gomoluch, Anthony Bradley
Dual-VaDE: Efficient and Robust Prototype Learning via Variational Dual-Domain Mixture Priors
Wugeng Zheng, Song Wang, Jian Wang
Effects of Parton Pair Correlations on Direct-Photon Yields and Elliptic Flow in Heavy-Ion Collisions
Xingjian Lu, Shuzhe Shi
Positional Encoding is All You Need for Scalable Equivariance Constraint Relaxation
Hagay Michaeli, Haggai Maron, Daniel Soudry
Federated Transfer Learning for Medical Image Classification under Mixed Skew
Alexandre G. Leclercq, Sébastien Bougleux, Aurélien Corroyer-Dulmont et al.
DRAP: Dynamics-Risk Adaptive Plasticity for Continual Learning in Physical AI
Oswaldo Ludwig