mlsanityData to 14 Oct 2026

Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depth

arXiv 2607.14018 ↗First posted 15 Jul 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

We investigate how each component of the Transformer feedforward block architecture design determines how much rank survives across depth at initialization. We reinterpret skip connections and normalization, long understood as controlling magnitude, as mechanisms for preserving gradient rank across depth, since the very matrix multiplications and nonlinear activations that make the network expressive also reduce the rank. We show that skip connections trade off rank collapse against ensemble-like behavior, controlled by the relative scales of the branch and the skip: skip connections route the gradient around the residual branch, where rank is lost, rather than along the long gradient paths that encourage the layers to compose. The placement of the normalization layer controls this same tradeoff by setting the branch-to-skip ratio across depth, unifying much of the normalization placement and depth scaling literature, in particular why rank collapses for Post-Norm but plateaus for Pre-Norm. Other aspects of the architecture, like the two-matrix structure that expands and contracts the width, use additional parameters to preserve the representation or branch Jacobian rank. The second matrix decorrelates a coherent mean spike that would grow across blocks with a single matrix and uncentered activation, preventing the residual representation from collapsing. The width expansion between the two matrices keeps the branch Jacobian full rank: applying the rank-reducing activation in this expanded space leaves enough directions to span the original, at a width that follows a Marchenko--Pastur law. The initialization rank of the input--output Jacobian predicts which networks train on CIFAR-10. Taken together, we recast architecture design for deep networks as navigating an intrinsic tradeoff among rank collapse, ensemble-like behavior, and parameter count.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 68 of its 79 references that are in the map. First posted on arXiv 15 Jul 2026.

Builds on

68 of 79 references.

References · 68

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2024The Llama 3 Herd of Modelscited by 14,715
  4. 2025Qwen3 Technical Reportcited by 8,820
  5. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  6. 2016Layer Normalizationcited by 5,669
  7. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  8. 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
  9. 2016Gaussian Error Linear Units (GELUs)cited by 3,502
  10. 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
  11. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,998
  12. 2016Identity Mappings in Deep Residual Networkscited by 1,965
38 more
  1. 2025Gemma 3 Technical Reportcited by 1,824
  2. 2020GLU Variants Improve Transformercited by 1,804
  3. 2019Root Mean Square Layer Normalizationcited by 1,480
  4. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,386
  5. 2020On Layer Normalization in the Transformer Architecturecited by 922
  6. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  7. 2021CogView: Mastering Text-to-Image Generation via Transformerscited by 694
  8. 2025Kimi K2: Open Agentic Intelligencecited by 567
  9. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
  10. 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
  11. 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
  12. 2020ReZero is All You Need: Fast Convergence at Large Depthcited by 264
  13. 2018Adaptive Input Representations for Neural Language Modelingcited by 226
  14. 2019Understanding and Improving Layer Normalizationcited by 217
  15. 2018Implicit Self-Regularization in Deep Neural Networks: Evidence from Random Matrix Theory and Implications for Learningcited by 215
  16. 2019Fixup Initialization: Residual Learning Without Normalizationcited by 198
  17. 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
  18. 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
  19. 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
  20. 2022Signal Propagation in Transformers: Theoretical Perspectives and the Role of Rank Collapsecited by 141
  21. 2022Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practicecited by 136
  22. 2017Resurrecting the sigmoid in deep learning through dynamical isometry: theory and practicecited by 132
  23. 2024Physics of Language Models: Part 2.1, Grade-School Math and the Hidden Reasoning Processcited by 108
  24. 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 97
  25. 2021NormFormer: Improved Transformer Pretraining with Extra Normalizationcited by 83
  26. 2025Don't be lazy: CompleteP enables compute-efficient deep transformerscited by 78
  27. 2022Revisiting Over-smoothing in BERT from the Perspective of Graphcited by 72
  28. 2023Depthwise Hyperparameter Transfer in Residual Networks: Dynamics and Scaling Limitcited by 66
  29. 2025The Curse of Depth in Large Language Modelscited by 62
  30. 2020Stable ResNetcited by 49
  31. 2024On the Role of Attention Masks and LayerNorm in Transformerscited by 48
  32. 2022Rank Diminishing in Deep Neural Networkscited by 47
  33. 2023The Shaped Transformer: Attention Models in the Infinite Depth-and-Width Limitcited by 47
  34. 2023Deep Transformers without Shortcuts: Modifying Self-attention for Faithful Signal Propagationcited by 45
  35. 2023Tensor Programs IVb: Adaptive Optimization in the Infinite-Width Limitcited by 38
  36. 2023Simplifying Transformer Blockscited by 37
  37. 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
  38. 2025Peri-LN: Revisiting Normalization Layer in the Transformer Architecturecited by 34

50 most cited of 68 All 68 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.