mlsanityData to 14 Oct 2026

CtrlDiff: Boosting Large Diffusion Language Models with Dynamic Block Prediction and Controllable Generation

arXiv 2505.14455 ↗First posted 20 May 2025cs.CL cs.AIcited by 17

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Although autoregressive models have dominated language modeling in recent years, there has been a growing interest in exploring alternative paradigms to the conventional next-token prediction framework. Diffusion-based language models have emerged as a compelling alternative due to their powerful parallel generation capabilities and inherent editability. However, these models are often constrained by fixed-length generation. A promising direction is to combine the strengths of both paradigms, segmenting sequences into blocks, modeling autoregressive dependencies across blocks while leveraging discrete diffusion to estimate the conditional distribution within each block given the preceding context. Nevertheless, their practical application is often hindered by two key limitations: rigid fixed-length outputs and a lack of flexible control mechanisms. In this work, we address the critical limitations of fixed granularity and weak controllability in current large diffusion language models. We propose CtrlDiff, a dynamic and controllable semi-autoregressive framework that adaptively determines the size of each generation block based on local semantics using reinforcement learning. Furthermore, we introduce a classifier-guided control mechanism tailored to discrete diffusion, which significantly reduces computational overhead while facilitating efficient post-hoc conditioning without retraining. Extensive experiments demonstrate that CtrlDiff sets a new standard among hybrid diffusion models, narrows the performance gap to state-of-the-art autoregressive approaches, and enables effective conditional text generation across diverse tasks.

Key facts. Cited by 17 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 51 of its 66 references that are in the map. First posted on arXiv 20 May 2025.

Builds on

51 of 66 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
  2. 2025Continuous Diffusion Model for Language Modeling17 Feb
  3. 2024Energy-Based Diffusion Language Models for Text Generation28 Oct
  4. 2024Masked Diffusion Models are Secretly Time-Agnostic Masked Models and Exploit Inaccurate Categorical Sampling4 Sep
  5. 2024Simple and Effective Masked Diffusion Language Models11 Jun
  6. 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
  7. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
  8. 2022DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models28 Nov
  9. 2022SSD-LM: Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Control31 Oct
  10. 2021Autoregressive Diffusion Models5 Oct

Other references · 41

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020Language Models are Few-Shot Learnerscited by 32,419
  3. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  4. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
  5. 2017Proximal Policy Optimization Algorithmscited by 15,215
  6. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  7. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
  8. 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
  9. 2022Scalable Diffusion Models with Transformerscited by 6,054
  10. 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
  11. 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
  12. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
29 more
  1. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
  2. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
  3. 2016Pointer Sentinel Mixture Modelscited by 3,096
  4. 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
  5. 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
  6. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  7. 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
  8. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  9. 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
  10. 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
  11. 2019CTRL: A Conditional Transformer Language Model for Controllable Generationcited by 762
  12. 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
  13. 2022DiGress: Discrete Denoising diffusion for graph generationcited by 520
  14. 2018A Discourse-Aware Attention Model for Abstractive Summarization of Long Documentscited by 478
  15. 2021FUDGE: Controlled Text Generation With Future Discriminatorscited by 376
  16. 2013One Billion Word Benchmark for Measuring Progress in Statistical Language Modelingcited by 308
  17. 2023Protein Design with Guided Discrete Diffusioncited by 157
  18. 2024Simple Guidance Mechanisms for Discrete Diffusion Modelscited by 155
  19. 2019What makes a good conversation? How controllable attributes affect human judgmentscited by 140
  20. 2024Unlocking Guidance for Discrete State-Space Diffusion and Flow Modelscited by 129
  21. 2024Dirichlet Flow Matching with Applications to DNA Sequence Designcited by 115
  22. 2025OpenCodeReasoning: Advancing Data Distillation for Competitive Codingcited by 113
  23. 2021Oops I Took A Gradient: Scalable Sampling for Discrete Distributionscited by 92
  24. 2023Stay on topic with Classifier-Free Guidancecited by 91
  25. 2019Discrete Flows: Invertible Generative Models of Discrete Datacited by 68
  26. 2019Latent Normalizing Flows for Discrete Sequencescited by 67
  27. 2022Training and Inference on Any-Order Autoregressive Models the Right Waycited by 64
  28. 2021OmniNet: Omnidirectional Representations from Transformerscited by 23
  29. 2024DiscDiff: Latent Diffusion Model for DNA Sequence Generationcited by 12

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 9; 2026: 8.

17 papers.

Sort:

By area:

  1. 2025A Survey on Diffusion Language Modelscited by 78
  2. 2025Diffusion Language Models Know the Answer Before Decodingcited by 70
  3. 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
  4. 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Modelscited by 37
  5. 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Modelscited by 30
  6. 2025Set Block Decoding is a Language Model Inference Acceleratorcited by 23
  7. 2025Unveiling the Potential of Diffusion Large Language Model in Controllable Generationcited by 15
  8. 2026IDLM: Inverse-distilled Diffusion Language Modelscited by 8
  9. 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Modelscited by 5
  10. 2025Guided Transfer Learning for Discrete Diffusion Modelscited by 3
  11. 2026Stopping Computation for Converged Tokens in Masked Diffusion-LM Decodingcited by 3
  12. 2026From Static Inference to Dynamic Interaction: A Survey of Streaming Large Language Modelscited by 3
5 more
  1. 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Modelscited by 2
  2. 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decodingcited by 0
  3. 2026Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixescited by 0
  4. 2026Block-R1: Rethinking the Role of Block Size in Multi-domain Reinforcement Learning for Diffusion Large Language Modelscited by 0
  5. 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplerscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025Sequential Diffusion Language Models
  2. 2026Diffusion Language Models: An Experimental Analysis
  3. 2025Dream 7B: Diffusion Large Language Models
  4. 2023Likelihood-Based Diffusion Language Models
  5. 2026Breaking the Factorization Barrier in Diffusion Language Models
  6. 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models
  7. 2025Breaking AR's Sampling Bottleneck: Provable Acceleration via Diffusion Language Models
  8. 2026Hierarchical Continuous Diffusion Language Models