English

Some Theoretical Results on Layerwise Effective Dimension Oscillations in Finite Width ReLU Networks

Machine Learning 2025-08-01 v2

Abstract

We analyze the layerwise effective dimension (rank of the feature matrix) in fully-connected ReLU networks of finite width. Specifically, for a fixed batch of mm inputs and random Gaussian weights, we derive closed-form expressions for the expected rank of the $m\times n$ hidden activation matrices. Our main result shows that E[EDim()]=m[1(12/π)]+O(ecm)\mathbb{E}[EDim(\ell)]=m[1-(1-2/\pi)^\ell]+O(e^{-c m}) so that the rank deficit decays geometrically with ratio 12/π0.36341-2 / \pi \approx 0.3634. We also prove a sub-Gaussian concentration bound, and identify the "revival" depths at which the expected rank attains local maxima. In particular, these peaks occur at depths k(k+1/2)π/log(1/ρ)\ell_k^*\approx(k+1/2)\pi/\log(1/\rho) with height (1eπ/2)m0.79m\approx (1-e^{-\pi/2}) m \approx 0.79m. We further show that this oscillatory rank behavior is a finite-width phenomenon: under orthogonal weight initialization or strong negative-slope leaky-ReLU, the rank remains (nearly) full. These results provide a precise characterization of how random ReLU layers alternately collapse and partially revive the subspace of input variations, adding nuance to prior work on expressivity of deep networks.

Keywords

Cite

@article{arxiv.2507.07675,
  title  = {Some Theoretical Results on Layerwise Effective Dimension Oscillations in Finite Width ReLU Networks},
  author = {Darshan Makwana},
  journal= {arXiv preprint arXiv:2507.07675},
  year   = {2025}
}

Comments

Incomplete citations