Where Should Diffusion Enter a Language Model? Geometry-Guided Hidden-State Replacement
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Continuous diffusion language models require choosing a representation space in which denoising operates, yet it remains unclear which representations are most compatible with diffusion. We study a basic design question: where inside a pretrained language model should continuous diffusion operate? We formulate this as a hidden-state interface-selection problem and hypothesize that diffusion-friendly interfaces can be identified from representation geometry. We operationalize this hypothesis using three training-free geometric proxies: local compactness, global stiffness, and effective rank. Across two 8B-scale backbones, the resulting geometry score predicts fixed-budget diffusion bridgeability beyond the dominant effect of layer depth. We then instantiate DiHAL, a Locate-and-Replace framework that replaces the transformer prefix below a selected interface with conditional diffusion while retaining the pretrained suffix and LM head. Under full training, geometry-selected interfaces remain close to validation-loss oracles and outperform worst-layer controls, while the diffusion bridge outperforms a parameter-matched deterministic replacement under matched diagnostic conditions. These results suggest that the representation space in which diffusion operates should itself be treated as a first-class design variable for continuous diffusion in language models.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 22 of its 28 references that are in the map. First posted on arXiv 14 May 2026.
Builds on
22 of 28 references.
Closest ancestors · 10
Same direction, newest first
- 2026CoDAR: Continuous Diffusion Language Models are More Powerful Than You Think3 Mar
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025Continuous Diffusion Model for Language Modeling17 Feb
- 2025Large Language Diffusion Models14 Feb
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2023DiffuSeq-v2: Bridging Discrete and Continuous Text Spaces for Accelerated Seq2Seq Diffusion Models9 Oct
- 2022Latent Diffusion for Language Generation19 Dec
- 2022Self-conditioned Embedding Diffusion for Text Generation8 Nov
- 2021Autoregressive Diffusion Models5 Oct
Other references · 12
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2019Reducing Transformer Depth on Demand with Structured Dropoutcited by 439
- 2025Diffusion vs. Autoregressive Language Models: A Text Embedding Perspectivecited by 23
- 2022Language modeling via stochastic processescited by 18
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion
- 2026Revise, Don't Freeze: Sampler-Matched Training for Self-Correcting Masked Diffusion Language Models
- 2026Reducing Pretraining-Generation Mismatch in Diffusion Language Models
- 2026Remask, Don't Replace: Token-to-Mask Refinement in Diffusion Language Models
- 2026Scaling Beyond Masked Diffusion Language Models
- 2026Window-Diffusion: Accelerating Diffusion Language Model Inference with Windowed Token Pruning and Caching
- 2026Don't Retrain, Align: Adapting Autoregressive LMs to Diffusion LMs via Representation Alignment
- 2026Nemotron-Labs-TwoTower: Diffusion Language Modeling with Pretrained Autoregressive Context