mlsanityData to 7 Oct 2026

CascadeFormer: Depth-Tapered Transformers Motivated by Gradient Fan-in Asymmetry

arXiv 2606.26538 ↗First posted 25 Jun 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Deep Transformers are composed of uniformly stacked residual blocks, yet their deepest layers often add little value. We present two efficiency methods that exploit this asymmetry. CascadeFormer tapers width with depth to match the uneven information flow across layers, achieving comparable perplexity to a uniform baseline at the same training budget while reducing latency by 8.6% and increasing throughput by 9.4%. CascadeFlow Pruning removes layers using accumulated training gradients, with no post hoc analysis. It outperforms standard heuristics on perplexity and rank-stability and stays competitive on downstream accuracy. To motivate these methods, we propose Gradient Fan-in Asymmetry (GFA) as a structural account of why deeper layers contribute less. In Pre-LayerNorm residual stacks, the gradient at a layer is the sum of an identity path and all downstream functional paths, producing a gradient fan-in that decays linearly with depth (and quadratically under deep supervision), yielding richer gradients for early layers and sparser ones for later layers. We provide correlational and interventional evidence for GFA on models trained from scratch up to 1.2B parameters. Across Transformers and ResNets, accumulated training gradients follow the theoretical fan-in and are associated with post hoc layer importance. Two interventions point to structure rather than magnitude as the bottleneck: equalizing per-layer gradient norms does not restore late-layer value, while increasing downstream path counts via parameter-shared repetition restores and elevates it. Whether gradient magnitude proxies fan-in beyond high-rank regimes, and how these dynamics behave at the 100B+ scale, remain open questions.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 31 of its 40 references that are in the map. First posted on arXiv 25 Jun 2026.

Builds on

31 of 40 references.

References · 31

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2020Language Models are Few-Shot Learnerscited by 32,199
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 13,958
  5. 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
  6. 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
  7. 2015Learning both Weights and Connections for Efficient Neural Networkscited by 1,998
  8. 2016Identity Mappings in Deep Residual Networkscited by 1,960
  9. 2017SmoothGrad: removing noise by adding noisecited by 1,097
  10. 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
  11. 2020On Layer Normalization in the Transformer Architecturecited by 906
  12. 2018SNIP: Single-shot Network Pruning based on Connection Sensitivitycited by 802
19 more
  1. 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 434
  2. 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 408
  3. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 369
  4. 2020DeeBERT: Dynamic Early Exiting for Accelerating BERT Inferencecited by 322
  5. 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 302
  6. 2020FastBERT: a Self-distilling BERT with Adaptive Inference Timecited by 224
  7. 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 219
  8. 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 176
  9. 2024LaCo: Large Language Model Pruning via Layer Collapsecited by 124
  10. 2021GSPMD: General and Scalable Parallelization for ML Computation Graphscited by 109
  11. 2024SLEB: Streamlining LLMs through Redundancy Verification and Elimination of Transformer Blockscited by 96
  12. 2021NormFormer: Improved Transformer Pretraining with Extra Normalizationcited by 83
  13. 2024What Matters in Transformers? Not All Attention is Neededcited by 83
  14. 2025The Curse of Depth in Large Language Modelscited by 58
  15. 2023LoRAShear: Efficient Large Language Model Structured Pruning and Knowledge Recoverycited by 37
  16. 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
  17. 2024Tracing Representation Progression: Analyzing and Enhancing Layer-Wise Similaritycited by 36
  18. 2022B2T Connection: Serving Stability and Performance in Deep Transformerscited by 16
  19. 2025SkipGPT: Dynamic Layer Pruning Reinvented with Token Awareness and Module Decouplingcited by 11

Built on by

Not cited yet.