Exploring Attention Sinks in Text-to-Image Transformers from Diffusion and Autoregressive Models
cs.CV, cs.CL
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: 10 pages
Code: https://github.com/black-forest-labs/flux
License: http://creativecommons.org/licenses/by/4.0/
The gist: Attention sinks are tokens or positions that consistently receive disproportionate attention mass, often beyond what their semantic content would suggest.
Terminology
Abstract
Attention sinks are tokens or positions that consistently receive disproportionate attention mass, often beyond what their semantic content would suggest. Despite extensive study in large language models, their behavior and functional relevance in text-to-image (T2I) Transformers remain largely unexplored. To fill this gap and better understand how text-to-image Transformers allocate and route attention, we explore this phenomenon of text-to-image transformers across bidirectional diffusion models and causal autoregressive models. Specifically, we explore whether attention sinks exist, where they emerge, and how they functionally influence generation. Through systematic attention analysis and targeted interventions, we identify that attention sinks emerge in both bidirectional diffusion and autoregressive models but exhibit distinct patterns. The autoregressive model concentrates attention on the first and global-scale tokens, whereas the bidirectional models develop sinks near modality boundaries, within text regions, and at spatial corners. Moreover, these patterns vary across generation stages and are localized to a sparse subset of layers and heads. To assess their functional relevance, we perturb sink-related heads and token positions. It turns out that perturbing sink-ranked heads produces larger average changes than non-sink comparisons, while some sink-position perturbations have limited effects on bidirectional model scores. We further introduce register tokens to redistribute attention sinks and find that several configurations preserve or modestly improve generation performance. Overall, attention sinks are architecture-, inference-stage-, and component-dependent phenomena, providing a basis for mechanistic analysis and principled attention design in text-to-image transformers.
Sources
- Imagen 3
- Z-Image: An Efficient Image Generation Foundation Model with Single-Stream Diffusion Transformer
- HunyuanImage 3.0 Technical Report
- Mitigating Attention Sinks and Massive Activations in Audio-Visual Speech Recognition with LLMs
- Seedance 1.0: Exploring the Boundaries of Video Generation Models
- Seedream 2.0: A Native Chinese-English Bilingual Image Generation Foundation Model
- ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignment
- Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers
- To Sink or Not to Sink: Visual Information Pathways in Large Vision-Language Models
- Attention Sink in Transformers: A Survey on Utilization, Interpretation, and Mitigation
- Massive Activations in Large Language Models
- Autoregressive Model Beats Diffusion: Llama for Scalable Image Generation
- Qwen-Image Technical Report
- Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesis
- Taming Outlier Tokens in Diffusion Transformers
- Scaling Autoregressive Models for Content-Rich Text-to-Image Generation
- One Token Is Enough: Improving Diffusion Language Models with a Sink Token
- Qwen-Image-2.0 Technical Report
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models