Diffusion Guided Language Modeling
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Current language models demonstrate remarkable proficiency in text generation. However, for many applications it is desirable to control attributes, such as sentiment, or toxicity, of the generated language -- ideally tailored towards each specific use case and target audience. For auto-regressive language models, existing guidance methods are prone to decoding errors that cascade during generation and degrade performance. In contrast, text diffusion models can easily be guided with, for example, a simple linear sentiment classifier -- however they do suffer from significantly higher perplexity than auto-regressive alternatives. In this paper we use a guided diffusion model to produce a latent proposal that steers an auto-regressive language model to generate text with desired properties. Our model inherits the unmatched fluency of the auto-regressive approach and the plug-and-play flexibility of diffusion. We show that it outperforms previous plug-and-play guidance methods across a wide range of benchmark data sets. Further, controlling a new attribute in our framework is reduced to training a single logistic regression classifier.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 47 of its 58 references that are in the map. First posted on arXiv 8 Aug 2024.
Builds on
47 of 58 references.
Closest ancestors · 3
Same direction, newest first
- 2023PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Model5 Jun
- 2023Likelihood-Based Diffusion Language Models30 May
- 2022DiffuSeq: Sequence to Sequence Text Generation with Diffusion Models17 Oct
Other references · 44
- 2017Attention Is All You Needcited by 59,932
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,243
32 more
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 2,994
- 2023Sparks of Artificial General Intelligence: Early experiments with GPT-4cited by 2,579
- 2020GLU Variants Improve Transformercited by 1,724
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2019Root Mean Square Layer Normalizationcited by 1,425
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,235
- 2021Cascaded Diffusion Models for High Fidelity Image Generationcited by 1,155
- 2021Variational Diffusion Modelscited by 1,145
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 921
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 793
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
- 2021MAUVE: Measuring the Gap Between Neural Text and Human Text using Divergence Frontierscited by 466
- 2021Sentence-T5: Scalable Sentence Encoders from Pre-trained Text-to-Text Modelscited by 462
- 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 434
- 2023Fine-Grained Human Feedback Gives Better Rewards for Language Model Trainingcited by 399
- 2020GeDi: Generative Discriminator Guided Sequence Generationcited by 377
- 2021FUDGE: Controlled Text Generation With Future Discriminatorscited by 374
- 2023Understanding Diffusion Objectives as the ELBO with Simple Data Augmentationcited by 293
- 2023Personalized Soups: Personalized Large Language Model Alignment via Post-hoc Parameter Mergingcited by 271
- 2022A Contrastive Framework for Neural Text Generationcited by 224
- 2023Text Embeddings Reveal (Almost) As Much As Textcited by 189
- 2022Cramming: Training a Language Model on a Single GPU in One Daycited by 81
- 2023Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Modelcited by 79
- 2020Multi-agent Communication meets Natural Language: Synergies between Functional and Structural Language Learningcited by 65
- 2022On the Blind Spots of Model-Based Evaluation Metrics for Text Generationcited by 48
- 2023On the Challenges of Using Black-Box APIs for Toxicity Evaluation in Researchcited by 44
- 2023Language Model Alignment with Elastic Resetcited by 34
Built on by
Citing papers per year: 2024: 2; 2025: 7; 2026: 6.
15 papers.
Sort:
By area:
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2024Large Concept Models: Language Modeling in a Sentence Representation Spacecited by 86
- 2025Theoretical Benefit and Limitation of Diffusion Language Modelcited by 46
- 2026ELF: Embedded Language Flowscited by 27
- 2025LaDiR: Latent Diffusion Enhances LLMs for Text Reasoningcited by 25
- 2025Adaptive Classifier-Free Guidance via Dynamic Low-Confidence Maskingcited by 21
- 2026Learning a Generative Meta-Model of LLM Activationscited by 8
- 2025Language Models Are Implicitly Continuouscited by 4
- 2025From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Trainingcited by 3
- 2025Simple Denoising Diffusion Language Modelscited by 2
- 2026Stop-Think-AutoRegress: Language Modeling with Latent Diffusion Planningcited by 2
- 2026HIVE: Hidden-Evidence Verification for Hallucination Detection in Diffusion Large Language Modelscited by 1
- 2026Repurposing Pre-trained LLMs as High Fidelity Continuous Text Autoencoders23 Sep 2026
- 2026ELF: Embedded Language Flows11 May 2026
- 2026Focus on the Core: Empowering Diffusion Large Language Models by Self-Contrast2 May 2026
- 2026HIVE: Hidden-Evidence Verification for Hallucination Detection in Diffusion Large Language Models28 Apr 2026
- 2026Stop-Think-AutoRegress: Language Modeling with Latent Diffusion Planning24 Feb 2026
- 2026Learning a Generative Meta-Model of LLM Activations6 Feb 2026
- 2025Simple Denoising Diffusion Language Models27 Oct 2025
- 2025LaDiR: Latent Diffusion Enhances LLMs for Text Reasoning6 Oct 2025
- 2025From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Training24 Sep 2025
- 2025Next-Token Prediction Should be Ambiguity-Sensitive: A Meta-Learning Perspective19 Jun 2025
- 2025Adaptive Classifier-Free Guidance via Dynamic Low-Confidence Masking26 May 2025
- 2025Language Models Are Implicitly Continuous4 Apr 2025
3 more
- 2025Theoretical Benefit and Limitation of Diffusion Language Model13 Feb 2025
- 2024Large Concept Models: Language Modeling in a Sentence Representation Space11 Dec 2024
- 2024Scaling up Masked Diffusion Models on Text24 Oct 2024
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2025Theoretical Benefit and Limitation of Diffusion Language Modelcited by 46
- 2026ELF: Embedded Language Flowscited by 27
- 2025Adaptive Classifier-Free Guidance via Dynamic Low-Confidence Maskingcited by 21
- 2025Simple Denoising Diffusion Language Modelscited by 2
- 2026ELF: Embedded Language Flows11 May 2026
- 2025Simple Denoising Diffusion Language Models27 Oct 2025
- 2025Adaptive Classifier-Free Guidance via Dynamic Low-Confidence Masking26 May 2025
- 2025Theoretical Benefit and Limitation of Diffusion Language Model13 Feb 2025
- 2024Scaling up Masked Diffusion Models on Text24 Oct 2024
- 2024Large Concept Models: Language Modeling in a Sentence Representation Spacecited by 86
- 2026Learning a Generative Meta-Model of LLM Activationscited by 8
- 2025Language Models Are Implicitly Continuouscited by 4
- 2025From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Trainingcited by 3
- 2026HIVE: Hidden-Evidence Verification for Hallucination Detection in Diffusion Large Language Modelscited by 1
- 2025Next-Token Prediction Should be Ambiguity-Sensitive: A Meta-Learning Perspectivecited by 0
- 2026Focus on the Core: Empowering Diffusion Large Language Models by Self-Contrastcited by 0
- 2026Repurposing Pre-trained LLMs as High Fidelity Continuous Text Autoencoderscited by 0
- 2026Repurposing Pre-trained LLMs as High Fidelity Continuous Text Autoencoders23 Sep 2026
- 2026Focus on the Core: Empowering Diffusion Large Language Models by Self-Contrast2 May 2026
- 2026HIVE: Hidden-Evidence Verification for Hallucination Detection in Diffusion Large Language Models28 Apr 2026
- 2026Learning a Generative Meta-Model of LLM Activations6 Feb 2026
- 2025From Text to Talk: Audio-Language Model Needs Non-Autoregressive Joint Training24 Sep 2025
- 2025Next-Token Prediction Should be Ambiguity-Sensitive: A Meta-Learning Perspective19 Jun 2025
- 2025Language Models Are Implicitly Continuous4 Apr 2025
- 2024Large Concept Models: Language Modeling in a Sentence Representation Space11 Dec 2024
Similar, not cited
Close in wording, no citation either way
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models
- 2023A Survey of Diffusion Models in Natural Language Processing
- 2023Diffusion Models for Non-autoregressive Text Generation: A Survey
- 2026Steering Without Breaking: Mechanistically Informed Interventions for Discrete Diffusion Language Models
- 2023AR-Diffusion: Auto-Regressive Diffusion Model for Text Generation
- 2024Language Rectified Flow: Advancing Diffusion Language Generation with Probabilistic Flows
- 2026Diffusion Language Models: An Experimental Analysis
- 2022Latent Diffusion for Language Generation