mlsanityData to 14 Oct 2026

ε-rank and the Staircase Phenomenon: New Insights into Neural Network Training Dynamics

arXiv 2412.05144 ↗First posted 6 Dec 2024cs.LG cs.NA math.NAcited by 1

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

  1. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
  2. 2013Visualizing and Understanding Convolutional Networkscited by 2,290
  3. 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
  4. 2018On the Spectral Bias of Neural Networkscited by 1,341
  5. 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
  6. 2017DGM: A deep learning algorithm for solving partial differential equationscited by 678
  7. 2020When and why PINNs fail to train: A neural tangent kernel perspectivecited by 634
  8. 2017The Deep Ritz method: A deep learning-based numerical algorithm for solving variational problemscited by 583
  9. 2017Sharp Minima Can Generalize For Deep Netscited by 466
  10. 2019Frequency Principle: Fourier Analysis Sheds Light on Deep Neural Networkscited by 368
  11. 2017Deep learning-based numerical methods for high-dimensional parabolic partial differential equations and backward stochastic differential equationscited by 191
  12. 2019On the Inductive Bias of Neural Tangent Kernelscited by 177
10 more
  1. 2018Training behavior of deep neural network in frequency domaincited by 174
  2. 2020Solving Allen-Cahn and Cahn-Hilliard Equations using the Adaptive Physics Informed Neural Networkscited by 123
  3. 2020The Random Feature Model for Input-Output Maps between Banach Spacescited by 90
  4. 2020Local Extreme Learning Machines and Domain Decomposition for Solving Linear and Nonlinear Partial Differential Equationscited by 81
  5. 2020Multi-scale Deep Neural Network (MscaleDNN) for Solving Poisson-Boltzmann Equation in Complex Domainscited by 79
  6. 2022Overview frequency principle/spectral bias in deep learningcited by 69
  7. 2021Hybrid FEM-NN models: Combining artificial neural networks with the finite element methodcited by 24
  8. 2022Finite Element Method-enhanced Neural Network for Forward and Inverse Problemscited by 13
  9. 2023TransNet: Transferable Neural Networks for Partial Differential Equationscited by 13
  10. 2021ReLU Deep Neural Networks from the Hierarchical Basis Perspectivecited by 12

Built on by

Cited by year

Citing papers per year: 2025: 1.

1 paper.