Attention Sinks in Diffusion Language Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Masked Diffusion Language Models (DLMs) have recently emerged as a promising alternative to traditional Autoregressive Models (ARMs). DLMs employ transformer encoders with bidirectional attention, enabling parallel token generation while maintaining competitive performance. Although their efficiency and effectiveness have been extensively studied, the internal mechanisms that govern DLMs remain largely unexplored. In this work, we conduct an empirical analysis of DLM attention patterns, focusing on the attention sinking phenomenon, an effect previously observed in various transformer-based architectures. Our findings reveal that DLMs also exhibit attention sinks, but with distinct characteristics. First, unlike in ARMs, the sink positions in DLMs tend to shift throughout the generation process, displaying a dynamic behaviour. Second, while ARMs are highly sensitive to the removal of attention sinks, DLMs remain robust: masking sinks leads to only a minor degradation in performance. These results provide new insights into the inner workings of diffusion-based language models and highlight fundamental differences in how they allocate and utilize attention compared to autoregressive models.
Key facts. Cited by 18 papers in the map (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 28 of its 48 references that are in the map. First posted on arXiv 17 Oct 2025.
Builds on
28 of 48 references.
Closest ancestors · 8
Same direction, newest first
- 2025Any-Order Flexible Length Masked Diffusion31 Aug
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025Implicit Search via Discrete Diffusion: A Study on Chess27 Feb
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models23 Oct
- 2024Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning18 Oct
- 2022DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models28 Nov
- 2022Self-conditioned Embedding Diffusion for Text Generation8 Nov
- 2022DiffuSeq: Sequence to Sequence Text Generation with Diffusion Models17 Oct
Other references · 20
- 2017Attention Is All You Needcited by 59,932
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 404
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 225
8 more
- 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 200
- 2023On Over-Squashing in Message Passing Neural Networks: The Impact of Width, Depth, and Topologycited by 163
- 2025Fast-dLLM v2: Efficient Block-Diffusion LLMcited by 128
- 2025Why do LLMs attend to the first token?cited by 122
- 2024Transformers need glasses! Information over-squashing in language taskscited by 87
- 2024A Simple and Effective L2 Norm-Based Strategy for KV Cache Compressioncited by 84
- 2024Spectral Filters, Dark Signals, and Attention Sinkscited by 54
- 2025What are you sinking? A geometric approach on attention sinkcited by 26
Built on by
Citing papers per year: 2025: 1; 2026: 17.
18 papers.
Sort:
By area:
- 2026DAWN: Dependency-Aware Fast Inference for Diffusion LLMscited by 12
- 2026Attention Sinks Are Provably Necessary in Softmax Transformers: Evidence from Trigger-Conditional Taskscited by 10
- 2025Masks Can Be Distracting: On Context Comprehension in Diffusion Language Modelscited by 7
- 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusingcited by 6
- 2026Revealing the Attention Floating Mechanism in Masked Diffusion Modelscited by 4
- 2026Surgery: Mitigating Harmful Fine-Tuning for Large Language Models via Attention Sinkcited by 3
- 2026One Token Is Enough: Improving Diffusion Language Models with a Sink Tokencited by 2
- 2026A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMscited by 2
- 2026Attention Sinks in Diffusion Transformers: A Causal Analysiscited by 2
- 2026Registers Matter for Pixel-Space Diffusion Transformerscited by 2
- 2026Sink-Aware Pruning for Diffusion Language Modelscited by 1
- 2026Adaptive Guidance for Retrieval-Augmented Masked Diffusion Modelscited by 1
6 more
- 2026Sink vs. diagonal patterns as mechanisms for attention switch and oversmoothing preventioncited by 1
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformerscited by 1
- 2026Order Matters: LVLMs as Judges for Temporal Reasoning in Image Sequencescited by 1
- 2026PulseCol: Periodically Refreshed Column-Sparse Attention for Accelerating Diffusion Language Modelscited by 0
- 2026Subliminal Clocks: Latent Time Modelling in Diffusion Language Modelscited by 0
- 2026SoloQ: Calibration-Free Quantization for Diffusion Language Modelscited by 0
- 2026SoloQ: Calibration-Free Quantization for Diffusion Language Models5 Oct 2026
- 2026Order Matters: LVLMs as Judges for Temporal Reasoning in Image Sequences11 Aug 2026
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers21 Jul 2026
- 2026Subliminal Clocks: Latent Time Modelling in Diffusion Language Models2 Jul 2026
- 2026PulseCol: Periodically Refreshed Column-Sparse Attention for Accelerating Diffusion Language Models20 May 2026
- 2026Registers Matter for Pixel-Space Diffusion Transformers15 May 2026
- 2026Attention Sinks in Diffusion Transformers: A Causal Analysis10 May 2026
- 2026Sink vs. diagonal patterns as mechanisms for attention switch and oversmoothing prevention8 May 2026
- 2026Adaptive Guidance for Retrieval-Augmented Masked Diffusion Models18 Mar 2026
- 2026Attention Sinks Are Provably Necessary in Softmax Transformers: Evidence from Trigger-Conditional Tasks12 Mar 2026
- 2026A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMs8 Mar 2026
- 2026Sink-Aware Pruning for Diffusion Language Models19 Feb 2026
6 more
- 2026DAWN: Dependency-Aware Fast Inference for Diffusion LLMs6 Feb 2026
- 2026Surgery: Mitigating Harmful Fine-Tuning for Large Language Models via Attention Sink5 Feb 2026
- 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusing2 Feb 2026
- 2026One Token Is Enough: Improving Diffusion Language Models with a Sink Token27 Jan 2026
- 2026Revealing the Attention Floating Mechanism in Masked Diffusion Models12 Jan 2026
- 2025Masks Can Be Distracting: On Context Comprehension in Diffusion Language Models26 Nov 2025
- 2025Masks Can Be Distracting: On Context Comprehension in Diffusion Language Modelscited by 7
- 2026Revealing the Attention Floating Mechanism in Masked Diffusion Modelscited by 4
- 2026One Token Is Enough: Improving Diffusion Language Models with a Sink Tokencited by 2
- 2026A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMscited by 2
- 2026Attention Sinks in Diffusion Transformers: A Causal Analysiscited by 2
- 2026Registers Matter for Pixel-Space Diffusion Transformerscited by 2
- 2026Sink-Aware Pruning for Diffusion Language Modelscited by 1
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformerscited by 1
- 2026PulseCol: Periodically Refreshed Column-Sparse Attention for Accelerating Diffusion Language Modelscited by 0
- 2026Subliminal Clocks: Latent Time Modelling in Diffusion Language Modelscited by 0
- 2026SoloQ: Calibration-Free Quantization for Diffusion Language Modelscited by 0
- 2026SoloQ: Calibration-Free Quantization for Diffusion Language Models5 Oct 2026
- 2026Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers21 Jul 2026
- 2026Subliminal Clocks: Latent Time Modelling in Diffusion Language Models2 Jul 2026
- 2026PulseCol: Periodically Refreshed Column-Sparse Attention for Accelerating Diffusion Language Models20 May 2026
- 2026Registers Matter for Pixel-Space Diffusion Transformers15 May 2026
- 2026Attention Sinks in Diffusion Transformers: A Causal Analysis10 May 2026
- 2026A Comparative analysis of Layer-wise Representational Capacity in AR and Diffusion LLMs8 Mar 2026
- 2026Sink-Aware Pruning for Diffusion Language Models19 Feb 2026
- 2026One Token Is Enough: Improving Diffusion Language Models with a Sink Token27 Jan 2026
- 2026Revealing the Attention Floating Mechanism in Masked Diffusion Models12 Jan 2026
- 2025Masks Can Be Distracting: On Context Comprehension in Diffusion Language Models26 Nov 2025
- 2026DAWN: Dependency-Aware Fast Inference for Diffusion LLMscited by 12
- 2026DAWN: Dependency-Aware Fast Inference for Diffusion LLMs6 Feb 2026
- 2026Attention Sinks Are Provably Necessary in Softmax Transformers: Evidence from Trigger-Conditional Taskscited by 10
- 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusingcited by 6
- 2026Sink vs. diagonal patterns as mechanisms for attention switch and oversmoothing preventioncited by 1
- 2026Sink vs. diagonal patterns as mechanisms for attention switch and oversmoothing prevention8 May 2026
- 2026Attention Sinks Are Provably Necessary in Softmax Transformers: Evidence from Trigger-Conditional Tasks12 Mar 2026
- 2026Focus-dLLM: Accelerating Long-Context Diffusion LLM Inference via Confidence-Guided Context Focusing2 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2025Empirical Analysis of Decoding Biases in Masked Diffusion Models
- 2025A Survey on Diffusion Language Models
- 2026Induction in Both Directions: A Mechanistic Analysis of In-Context Learning in Masked Diffusion Language Models
- 2025Why mask diffusion does not work
- 2026PreDiff-LM: Pretrained Discrete Masked Diffusion Language Modeling with Hybrid Attention
- 2026dQwen3.5: Hybrid-Attention Diffusion Language Models
- 2025Esoteric Language Models: A Family of Any-Order Diffusion LLMs
- 2024Simple and Effective Masked Diffusion Language Models