Text Template Tokens Are Implicit Semantic Registers in Diffusion Transformers
Abstract
Text-to-image diffusion transformers (DiTs) jointly process text and image tokens, yet their internal computation during denoising remains poorly understood. We introduce a causal interpretability framework for modern large-scale DiTs that combines attention decomposition with targeted interventions across token spans, heads, and layers. Using it to separate prompt-content tokens from structural template tokens, we find that the structural 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, with prompt semantics first injected into the image latents and then read back into the template tokens rather than transferred directly from the prompt tokens. Inspired by the above findings, we design a training-free pruning rule for DiTs. Heads that attend most strongly to prompt tokens are dispensable, and pruning them removes 20% of attention FLOPs with only a 1.4-point drop on GenEval. We further reveal how generative computation in DiTs is organized across heads and depth, separating semantic routing from visual synthesis and progressing from identity formation to propagation and refinement. Our work not only reveals that the tokens encoding semantics at input need not be those that maintain it during generation, but also provides a causal view of internal mechanisms in DiTs.
Community
We find that "useless" text template tokens in T2I diffusion transformers are anything but β they act as implicit semantic registers that causally maintain object identity during generation. π§
Using a causal interpretability framework (attention decomposition + targeted interventions across token spans, heads, and layers), we show:
π Template tokens carry little prompt-specific info at the encoder output, yet become dominant image-to-text attention sinks inside the DiT.
π They acquire semantics indirectly: prompt meaning is first injected into image latents, then read back into template tokens β not transferred directly from prompt tokens.
βοΈ This insight yields a training-free pruning rule: heads attending most to prompt tokens are dispensable β pruning them cuts 20% of attention FLOPs with only a 1.4-point drop on GenEval.
πΊοΈ We also map how generative computation is organized across heads and depth: semantic routing vs. visual synthesis, from identity formation to propagation and refinement.
Takeaway: the tokens that encode semantics at input need not be the ones that maintain it during generation.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- FairFlow: Demystifying and Mitigating Stereotype Bias in Text-to-Image Diffusion Transformers (2026)
- Imagine Before You Draw: Visual Prompt Engineering for Image Generation (2026)
- Test-Time Registers as Global Priors for Tokenized Image Generation (2026)
- Decoupled Guidance: Disentangling Subject and Context Pathways in Text-to-Image Personalization (2026)
- DifFRACT: Diffusion Feature Reconstruction and Attribution for Circuit Tracing (2026)
- Unified Safe In-context Image Generation in Multimodal Diffusion Transformers via Restricting Unsafe Information Flows (2026)
- Appearance Pointers -- Multimodal Region Control of Diffusion Transformers (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2607.19139 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper