The Homogeneity Trap: Spectral Collapse in Doubly-Stochastic Deep Networks
Yizhi Liu
TL;DR
This work identifies a fundamental Homogeneity Trap in doubly-stochastic neural networks, where projecting mixing operators onto the DSM set biases toward a uniform matrix and drives the subdominant singular value $σ_2$ to near zero. By deriving finite-$n$ bounds and using concentration and perturbation results, the authors show Layer Normalization cannot restore geometry when the spectral SNR $γ$ is below a critical threshold, leading to identity stagnation in residual blocks and shallow effective depth. The study provides both theoretical mechanisms and empirical validations (e.g., $σ_2$ shrinking with entropy, orthogonal collapse at low $γ$, and affine LN ablations) to demonstrate the trade-off between entropic stability and spectral expressivity, with implications for DSM-based layers and potential non-DSM alternatives. Overall, the paper highlights a key constraint on architectural design: stabilizing projections can irreversibly filter out high-frequency structure, motivating exploration of parameterizations beyond strict DSM constraints for deep, expressive models.
Abstract
Doubly-stochastic matrices (DSM) are increasingly utilized in structure-preserving deep architectures -- such as Optimal Transport layers and Sinkhorn-based attention -- to enforce numerical stability and probabilistic interpretability. In this work, we identify a critical spectral degradation phenomenon inherent to these constraints, termed the Homogeneity Trap. We demonstrate that the maximum-entropy bias, typical of Sinkhorn-based projections, drives the mixing operator towards the uniform barycenter, thereby suppressing the subdominant singular value σ_2 and filtering out high-frequency feature components. We derive a spectral bound linking σ_2 to the network's effective depth, showing that high-entropy constraints restrict feature transformation to a shallow effective receptive field. Furthermore, we formally demonstrate that Layer Normalization fails to mitigate this collapse in noise-dominated regimes; specifically, when spectral filtering degrades the Signal-to-Noise Ratio (SNR) below a critical threshold, geometric structure is irreversibly lost to noise-induced orthogonal collapse. Our findings highlight a fundamental trade-off between entropic stability and spectral expressivity in DSM-constrained networks.
