mlsanityData to 14 Oct 2026

Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers

arXiv 2607.19139 ↗First posted 21 Jul 2026cs.CVcited by 4

On the map: Diffusion models → Image generation, text-to-image and editing

Abstract (arXiv)

Modern text-to-image diffusion transformers (DiTs) generate images through joint attention, in which text and image tokens interact directly within a single sequence. In large-scale DiTs, the conditioning input contains not only the user prompt but also chat-template tokens introduced by LLM-based text encoders. Yet how these tokens participate in the denoising computation remains poorly understood. To probe this, we introduce a causal interpretability framework. Using it to separate prompt-content tokens from chat-template tokens, we find that the template tokens carry little prompt-specific information at the encoder output. Yet surprisingly, they emerge as dominant image-to-text attention sinks and causally maintain object identity inside the DiT, acting as implicit semantic registers. We show that they acquire this identity indirectly. Rather than reading the prompt tokens, they draw the identity from the image latents into which the prompt semantics have already been injected at the very first layer. We further reveal a division of labor across heads and depth in DiTs, where distinct heads route semantics or render visual structure, and identity is committed in early blocks, carried by middle blocks, and refined in late ones. As a practical payoff, this analysis yields a training-free pruning rule that removes the causally inert prompt-reading heads and cuts 20% of joint-attention FLOPs at a 1.4-point cost in GenEval accuracy. Overall, our work not only reveals that the tokens encoding semantics at the input need not be those that maintain them during generation, but also provides a causal view of internal mechanisms in diffusion transformers.

Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Image generation, text-to-image and editing, one of 14 directions in Diffusion models. Builds on 56 of its 60 references that are in the map. First posted on arXiv 21 Jul 2026.

Builds on

56 of 60 references.

Closest ancestors · 6

Same direction, newest first

  1. 2026Attention Sinks in Diffusion Transformers: A Causal Analysis10 May
  2. 2025Exploring Multimodal Diffusion Transformers for Enhanced Prompt-based Image Editing11 Aug
  3. 2025Rethinking Cross-Modal Interaction in Multimodal Diffusion Transformers9 Jun
  4. 2025ConceptAttention: Diffusion Transformers Learn Highly Interpretable Features6 Feb
  5. 2024ELLA: Equip Diffusion Models with LLM for Enhanced Semantic Alignment8 Mar
  6. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignment17 Oct

Other references · 50

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  2. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  3. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  4. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  5. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
  6. 2020Denoising Diffusion Implicit Modelscited by 9,781
  7. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  8. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
  9. 2022Scalable Diffusion Models with Transformerscited by 6,054
  10. 2022Classifier-Free Diffusion Guidancecited by 5,847
  11. 2022Flow Matching for Generative Modelingcited by 5,702
  12. 2025Qwen2.5-VL Technical Reportcited by 5,587
38 more
  1. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
  2. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  3. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,761
  4. 2025Qwen3-VL Technical Reportcited by 2,793
  5. 2022Locating and Editing Factual Associations in GPTcited by 2,691
  6. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  7. 2025DINOv3cited by 1,414
  8. 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
  9. 2025Qwen-Image Technical Reportcited by 1,044
  10. 2024SnapKV: LLM Knows What You are Looking for Before Generationcited by 880
  11. 2023Vision Transformers Need Registerscited by 871
  12. 2023H2O: Heavy-Hitter Oracle for Efficient Generative Inference of Large Language Modelscited by 832
  13. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
  14. 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 502
  15. 2024Quest: Query-Aware Sparsity for Efficient Long-Context LLM Inferencecited by 501
  16. 2023Scissorhands: Exploiting the Persistence of Importance Hypothesis for LLM KV Cache Compression at Test Timecited by 439
  17. 2025Step1X-Edit: A Practical Framework for General Image Editingcited by 429
  18. 2024MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attentioncited by 383
  19. 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generationcited by 329
  20. 2024Massive Activations in Large Language Modelscited by 294
  21. 2024DuoAttention: Efficient Long-Context LLM Inference with Retrieval and Streaming Headscited by 288
  22. 2023MDTv2: Masked Diffusion Transformer is a Strong Image Synthesizercited by 262
  23. 2025HPSv3: Towards Wide-Spectrum Human Preference Scorecited by 211
  24. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 208
  25. 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 146
  26. 2025See What You Are Told: Visual Attention Sink in Large Multimodal Modelscited by 132
  27. 2025Why do LLMs attend to the first token?cited by 127
  28. 2025FlexPrefill: A Context-Aware Sparse Attention Mechanism for Efficient Long-Sequence Inferencecited by 123
  29. 2024RazorAttention: Efficient KV Cache Compression Through Retrieval Headscited by 100
  30. 2025Attention Sinks and Compression Valleys in LLMs are Two Sides of the Same Coincited by 52
  31. 2025Vision Transformers Don't Need Trained Registerscited by 41
  32. 2026Qwen-Image-Bench: From Generation to Creation in Text-to-Image Evaluationcited by 22
  33. 2025Attention Sinks in Diffusion Language Modelscited by 20
  34. 2026FASA: Frequency-aware Sparse Attentioncited by 8
  35. 2026Revealing the Attention Floating Mechanism in Masked Diffusion Modelscited by 5
  36. 2026Full Attention Strikes Back: Transferring Full Attention into Sparse within Hundred Training Stepscited by 4
  37. 2026Registers Matter for Pixel-Space Diffusion Transformerscited by 2
  38. 2026Rethinking Cross-Layer Information Routing in Diffusion Transformerscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 4.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Few Channels Draw The Whole Picture: Revealing Massive Activations in Diffusion Transformers
  2. 2025Rare Text Semantics Were Always There in Your Diffusion Transformer
  3. 2025Padding Tone: A Mechanistic Analysis of Padding Tokens in T2I Models
  4. 2023De-Diffusion Makes Text a Strong Cross-Modal Interface
  5. 2026Prompt Reinjection: Alleviating Prompt Forgetting in Multimodal Diffusion Transformers for Text-to-Image Generation
  6. 2024DiffusionPID: Interpreting Diffusion via Partial Information Decomposition
  7. 2026Text-to-Image Models Need Less from Text Encoders Than You Think
  8. 2024Enhancing Semantic Fidelity in Text-to-Image Synthesis: Attention Regulation in Diffusion Models