mlsanityData to 14 Oct 2026

Mean Mode Screaming: Mean--Variance Split Residuals for 1000-Layer Diffusion Transformers

arXiv 2605.06169 ↗First posted 7 May 2026cs.LG cs.CVcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Scaling Diffusion Transformers (DiTs) to hundreds of layers introduces a structural vulnerability: networks can enter a silent, mean-dominated collapse state that homogenizes token representations and suppresses centered variation. Through mechanistic auditing, we isolate the trigger event of this collapse as Mean Mode Screaming (MMS). MMS can occur even when training appears stable, with a mean-coherent backward shock on residual writers that opens deep residual branches and drives the network into a mean-dominated state. We show this behavior is driven by an exact decomposition of these gradients into mean-coherent and centered components, compounded by the structural suppression of attention-logit gradients through the null space of the Softmax Jacobian once values homogenize. To address this, we propose Mean-Variance Split (MV-Split) Residuals, which combine a separately gained centered residual update with a leaky trunk-mean replacement. On a 400-layer single-stream DiT, MV-Split prevents the divergent collapse that crashes the un-stabilized baseline; it tracks close to the baseline's pre-crash trajectory while remaining substantially better than token-isotropic gating methods such as LayerScale across the full schedule. Finally, we present a 1000-layer DiT as a scale-validation run at boundary scales, establishing that the architecture remains stably trainable at extreme depth.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 30 of its 42 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

30 of 42 references.

Closest ancestors · 8

Same direction, newest first

  1. 2026The Geometry of Noise: Why Diffusion Models Don't Need Noise Conditioning20 Feb
  2. 2025Is Noise Conditioning Necessary for Denoising Generative Models?18 Feb
  3. 2022Scalable Diffusion Models with Transformers19 Dec
  4. 2022Classifier-Free Diffusion Guidance26 Jul
  5. 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
  6. 2021Diffusion Models Beat GANs on Image Synthesis11 May
  7. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  8. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 22

  1. 2017Attention Is All You Needcited by 60,361
  2. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  3. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,503
  4. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  5. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  6. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  7. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
  8. 2020GLU Variants Improve Transformercited by 1,804
  9. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  10. 2019Root Mean Square Layer Normalizationcited by 1,480
  11. 2020On Layer Normalization in the Transformer Architecturecited by 922
  12. 2016Training Deep Nets with Sublinear Memory Costcited by 848
10 more
  1. 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
  2. 2021Going deeper with Image Transformerscited by 764
  3. 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
  4. 2020Query-Key Normalization for Transformerscited by 280
  5. 2020ReZero is All You Need: Fast Convergence at Large Depthcited by 264
  6. 2020Understanding the Difficulty of Training Transformerscited by 210
  7. 2019Fixup Initialization: Residual Learning Without Normalizationcited by 198
  8. 2023Small-scale proxies for large-scale Transformer training instabilitiescited by 183
  9. 2023Stabilizing Transformer Training by Preventing Attention Entropy Collapsecited by 161
  10. 2023Spike No More: Stabilizing the Pre-training of Large Language Modelscited by 52

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Residual-Stream Burden Shapes Representation Learning in Diffusion Transformers
  2. 2025Sprint: Sparse-Dense Residual Fusion for Efficient Diffusion Transformers
  3. 2026Structured Residual Connectivity Matters for Diffusion Transformers
  4. 2026Rethinking Cross-Layer Information Routing in Diffusion Transformers
  5. 2026RT-Lynx: Putting GEMM Sparsity in the Right Place for Diffusion Models
  6. 2025Exploring Magnitude Preservation and Rotation Modulation in Diffusion Transformers
  7. 2025DiffusionBlocks: Block-wise Neural Network Training via Diffusion Interpretation
  8. 2026Steering Video Diffusion Transformers with Massive Activations