ε-rank and the Staircase Phenomenon: New Insights into Neural Network Training Dynamics
Not on the map. Why?
Abstract (arXiv)
Understanding the training dynamics of deep neural networks (DNNs), particularly how they evolve low-dimensional features from high-dimensional data, remains a central challenge in deep learning theory. In this work, we introduce the concept of ε-rank, a novel metric quantifying the effective feature of neuron functions in the terminal hidden layer. Through extensive experiments across diverse tasks, we observe a universal staircase phenomenon: during training process implemented by the standard stochastic gradient descent methods, the decline of the loss function is accompanied by an increase in the ε-rank and exhibits a staircase pattern. Theoretically, we rigorously prove a negative correlation between the loss lower bound and ε-rank, demonstrating that a high ε-rank is essential for significant loss reduction. Moreover, numerical evidences show that within the same deep neural network, the ε-rank of the subsequent hidden layer is higher than that of the previous hidden layer. Based on these observations, to eliminate the staircase phenomenon, we propose a novel pre-training strategy on the initial hidden layer that elevates the ε-rank of the terminal hidden layer. Numerical experiments validate its effectiveness in reducing training time and improving accuracy across various tasks. Therefore, the newly introduced concept of ε-rank is a computable quantity that serves as an intrinsic effective metric characteristic for deep neural networks, providing a novel perspective for understanding the training dynamics of neural networks and offering a theoretical foundation for designing efficient training strategies in practical applications.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 39 references that are in the map. First posted on arXiv 6 Dec 2024.
Builds on
22 of 39 references.
References · 22
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2013Visualizing and Understanding Convolutional Networkscited by 2,290
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2018On the Spectral Bias of Neural Networkscited by 1,341
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
- 2017DGM: A deep learning algorithm for solving partial differential equationscited by 678
- 2020When and why PINNs fail to train: A neural tangent kernel perspectivecited by 634
- 2017The Deep Ritz method: A deep learning-based numerical algorithm for solving variational problemscited by 583
- 2017Sharp Minima Can Generalize For Deep Netscited by 466
- 2019Frequency Principle: Fourier Analysis Sheds Light on Deep Neural Networkscited by 368
- 2017Deep learning-based numerical methods for high-dimensional parabolic partial differential equations and backward stochastic differential equationscited by 191
- 2019On the Inductive Bias of Neural Tangent Kernelscited by 177
10 more
- 2018Training behavior of deep neural network in frequency domaincited by 174
- 2020Solving Allen-Cahn and Cahn-Hilliard Equations using the Adaptive Physics Informed Neural Networkscited by 123
- 2020The Random Feature Model for Input-Output Maps between Banach Spacescited by 90
- 2020Local Extreme Learning Machines and Domain Decomposition for Solving Linear and Nonlinear Partial Differential Equationscited by 81
- 2020Multi-scale Deep Neural Network (MscaleDNN) for Solving Poisson-Boltzmann Equation in Complex Domainscited by 79
- 2022Overview frequency principle/spectral bias in deep learningcited by 69
- 2021Hybrid FEM-NN models: Combining artificial neural networks with the finite element methodcited by 24
- 2022Finite Element Method-enhanced Neural Network for Forward and Inverse Problemscited by 13
- 2023TransNet: Transferable Neural Networks for Partial Differential Equationscited by 13
- 2021ReLU Deep Neural Networks from the Hierarchical Basis Perspectivecited by 12
Built on by
Citing papers per year: 2025: 1.
1 paper.