Search papers, labs, and topics across Lattice.
This paper introduces CascadeFormer, a Transformer architecture that optimizes layer efficiency by tapering width with depth, addressing the issue of diminishing returns from deeper layers. By leveraging Gradient Fan-in Asymmetry (GFA), the authors demonstrate that their approach achieves comparable perplexity to traditional models while reducing latency by 8.6% and increasing throughput by 9.4%. Additionally, CascadeFlow Pruning effectively removes less valuable layers based on accumulated training gradients, outperforming standard heuristics in perplexity and maintaining competitive downstream accuracy.
Tapering Transformer width with depth can significantly enhance efficiency, reducing latency while preserving performance metrics.
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.
