Layer Collapse in Diffusion Language Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Diffusion language models (DLMs) have recently emerged as competitive alternatives to autoregressive (AR) language models, yet differences in their activation dynamics remain poorly understood. We characterize these dynamics in LLaDA-8B and identify a striking layer-collapse property: a few early layers exhibit highly similar, collapsed activation patterns dominated by a single large super-outlier persisting over a long token range. Despite its apparent redundancy, this outlier is critical: pruning it causes outputs to degrade into repetitive random token loops. Paradoxically, layers in LLaDA contain more redundant representations overall, with redundancy most pronounced in earlier layers -- the reverse of AR models, where deeper layers grow redundant due to undertraining. Our analysis indicates that layer collapse in DLMs is not driven by undertraining but by overtraining: a dominant outlier becomes an indispensable information carrier while remaining representations collapse into redundant structure. These findings have strong practical implications, verified through controlled pre-training experiments. DLMs are surprisingly robust to compression: LLaDA under 3-bit GPTQ quantization drops only -1.8% on GSM8K, whereas Llama-3.1-8B drops -64.7%. Optimal sparsity allocation also reverses between families: at 50% average sparsity, allocating more to early layers in LLaDA yields +8.4% over the reverse strategy, while the same allocation costs Llama -8.4%. Our findings reveal that the DLM training objective fundamentally reshapes layer dynamics relative to AR models, with direct consequences for compression and deployment. Code: github.com/Conzel/super-outlier-dlm.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 41 of its 55 references that are in the map. First posted on arXiv 7 May 2026.
Builds on
41 of 55 references.
Closest ancestors · 5
Same direction, newest first
- 2026A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMs8 Mar
- 2025dInfer: An Efficient Inference Framework for Diffusion Language Models9 Oct
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025Large Language Diffusion Models14 Feb
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
Other references · 36
- 2017Attention Is All You Needcited by 59,932
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2019BoolQ: Exploring the Surprising Difficulty of Natural Yes/No Questionscited by 2,140
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2018Can a Suit of Armor Conduct Electricity? A New Dataset for Open Book Question Answeringcited by 2,114
- 2022GPTQ: Accurate Post-Training Quantization for Generative Pre-trained Transformerscited by 1,838
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,734
- 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,104
24 more
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2024QuaRot: Outlier-Free 4-Bit Inference in Rotated LLMscited by 598
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 399
- 2024Massive Activations in Large Language Modelscited by 277
- 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
- 2023Outlier Weighed Layerwise Sparsity (OWL): A Missing Secret Sauce for Pruning LLMs to High Sparsitycited by 163
- 2023OWQ: Outlier-Aware Weight Quantization for Efficient Fine-Tuning and Inference of Large Language Modelscited by 113
- 2019Traditional and Heavy-Tailed Self Regularization in Neural Network Modelscited by 104
- 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
- 2025dParallel: Learnable Parallel Decoding for dLLMscited by 59
- 2025The Curse of Depth in Large Language Modelscited by 58
- 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
- 2024AlphaPruning: Using Heavy-Tailed Self Regularization Theory for Improved Layer-wise Pruning of Large Language Modelscited by 35
- 2025Straight to Zero: Why Linearly Decaying the Learning Rate to Zero Works Best for LLMscited by 34
- 2025Systematic Outliers in Large Language Modelscited by 32
- 2025Quantization Meets dLLMs: A Systematic Study of Post-training Quantization for Diffusion LLMscited by 21
- 2025AlphaDecay: Module-wise Weight Decay for Heavy-Tailed Balancing in LLMscited by 12
- 2024Model Balancing Helps Low-data Training and Fine-tuningcited by 11
- 2025Quant-dLLM: Post-Training Extreme Low-Bit Quantization for Diffusion Large Language Modelscited by 11
- 2025DLLMQuant: Quantizing Diffusion-based Large Language Modelscited by 10
- 2025Do Depth-Grown Models Overcome the Curse of Depth? An In-Depth Analysiscited by 5
- 2026When Does Sparsity Mitigate the Curse of Depth in LLMscited by 4
- 2026From Words to Amino Acids: Does the Curse of Depth Persist?cited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Don't Retrain, Align: Adapting Autoregressive LMs to Diffusion LMs via Representation Alignment
- 2025A Survey on Diffusion Language Models
- 2026Beyond the Bidirectional Promise: Re-evaluating the Robustness of Diffusion Language Models
- 2025Diffusion Language Models are Super Data Learners
- 2026LESS Is More: Mutual-Stability Sampling for Diffusion Language Models
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
- 2026Enhancing Diffusion Language Models with Autoregressive Post-Training Weights
- 2026UNIFUSION: Adapting Autoregressive Language Models into Discrete Diffusion under a Unified Reverse-Rate Objective