CtrlDiff: Boosting Large Diffusion Language Models with Dynamic Block Prediction and Controllable Generation
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
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
- 2025Continuous Diffusion Model for Language Modeling17 Feb
- 2024Energy-Based Diffusion Language Models for Text Generation28 Oct
- 2024Masked Diffusion Models are Secretly Time-Agnostic Masked Models and Exploit Inaccurate Categorical Sampling4 Sep
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct
- 2022DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models28 Nov
- 2022SSD-LM: Semi-autoregressive Simplex-based Diffusion Language Model for Text Generation and Modular Control31 Oct
- 2021Autoregressive Diffusion Models5 Oct
Other references · 41
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Scalable Diffusion Models with Transformerscited by 6,054
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,326
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,356
29 more
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,181
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
- 2019Transformer-XL: Attentive Language Models Beyond a Fixed-Length Contextcited by 1,850
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 803
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
- 2019CTRL: A Conditional Transformer Language Model for Controllable Generationcited by 762
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2022DiGress: Discrete Denoising diffusion for graph generationcited by 520
- 2018A Discourse-Aware Attention Model for Abstractive Summarization of Long Documentscited by 478
- 2021FUDGE: Controlled Text Generation With Future Discriminatorscited by 376
- 2013One Billion Word Benchmark for Measuring Progress in Statistical Language Modelingcited by 308
- 2023Protein Design with Guided Discrete Diffusioncited by 157
- 2024Simple Guidance Mechanisms for Discrete Diffusion Modelscited by 155
- 2019What makes a good conversation? How controllable attributes affect human judgmentscited by 140
- 2024Unlocking Guidance for Discrete State-Space Diffusion and Flow Modelscited by 129
- 2024Dirichlet Flow Matching with Applications to DNA Sequence Designcited by 115
- 2025OpenCodeReasoning: Advancing Data Distillation for Competitive Codingcited by 113
- 2021Oops I Took A Gradient: Scalable Sampling for Discrete Distributionscited by 92
- 2023Stay on topic with Classifier-Free Guidancecited by 91
- 2019Discrete Flows: Invertible Generative Models of Discrete Datacited by 68
- 2019Latent Normalizing Flows for Discrete Sequencescited by 67
- 2022Training and Inference on Any-Order Autoregressive Models the Right Waycited by 64
- 2021OmniNet: Omnidirectional Representations from Transformerscited by 23
- 2024DiscDiff: Latent Diffusion Model for DNA Sequence Generationcited by 12
Built on by
Citing papers per year: 2025: 9; 2026: 8.
17 papers.
Sort:
By area:
- 2025A Survey on Diffusion Language Modelscited by 78
- 2025Diffusion Language Models Know the Answer Before Decodingcited by 70
- 2025Discrete Diffusion in Large Language and Multimodal Models: A Surveycited by 61
- 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Modelscited by 37
- 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Modelscited by 30
- 2025Set Block Decoding is a Language Model Inference Acceleratorcited by 23
- 2025Unveiling the Potential of Diffusion Large Language Model in Controllable Generationcited by 15
- 2026IDLM: Inverse-distilled Diffusion Language Modelscited by 8
- 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Modelscited by 5
- 2025Guided Transfer Learning for Discrete Diffusion Modelscited by 3
- 2026Stopping Computation for Converged Tokens in Masked Diffusion-LM Decodingcited by 3
- 2026From Static Inference to Dynamic Interaction: A Survey of Streaming Large Language Modelscited by 3
5 more
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Modelscited by 2
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decodingcited by 0
- 2026Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixescited by 0
- 2026Block-R1: Rethinking the Role of Block Size in Multi-domain Reinforcement Learning for Diffusion Large Language Modelscited by 0
- 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplerscited by 0
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Models28 Jun 2026
- 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplers1 Jun 2026
- 2026Block-R1: Rethinking the Role of Block Size in Multi-domain Reinforcement Learning for Diffusion Large Language Models12 May 2026
- 2026Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixes5 Mar 2026
- 2026From Static Inference to Dynamic Interaction: A Survey of Streaming Large Language Models4 Mar 2026
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decoding4 Mar 2026
- 2026IDLM: Inverse-distilled Diffusion Language Models22 Feb 2026
- 2026Stopping Computation for Converged Tokens in Masked Diffusion-LM Decoding6 Feb 2026
- 2025Guided Transfer Learning for Discrete Diffusion Models11 Dec 2025
- 2025Finish First, Perfect Later: Test-Time Token-Level Cross-Validation for Diffusion Large Language Models6 Oct 2025
- 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Models28 Sep 2025
- 2025Set Block Decoding is a Language Model Inference Accelerator4 Sep 2025
5 more
- 2025Diffusion Language Models Know the Answer Before Decoding27 Aug 2025
- 2025A Survey on Diffusion Language Models14 Aug 2025
- 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Models12 Aug 2025
- 2025Unveiling the Potential of Diffusion Large Language Model in Controllable Generation6 Jul 2025
- 2025Discrete Diffusion in Large Language and Multimodal Models: A Survey16 Jun 2025
- 2025A Survey on Diffusion Language Modelscited by 78
- 2025Diffusion Language Models Know the Answer Before Decodingcited by 70
- 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Modelscited by 37
- 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Modelscited by 30
- 2026IDLM: Inverse-distilled Diffusion Language Modelscited by 8
- 2025Guided Transfer Learning for Discrete Diffusion Modelscited by 3
- 2026Stopping Computation for Converged Tokens in Masked Diffusion-LM Decodingcited by 3
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Modelscited by 2
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decodingcited by 0
- 2026Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixescited by 0
- 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplerscited by 0
- 2026Adaptive Block Diffusion: Resolving Training-Inference Mismatch in Diffusion Language Models28 Jun 2026
- 2026BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplers1 Jun 2026
- 2026Beyond Scattered Acceptance: Fast and Coherent Inference for DLMs via Longest Stable Prefixes5 Mar 2026
- 2026Progressive Refinement Regulation for Accelerating Diffusion Language Model Decoding4 Mar 2026
- 2026IDLM: Inverse-distilled Diffusion Language Models22 Feb 2026
- 2026Stopping Computation for Converged Tokens in Masked Diffusion-LM Decoding6 Feb 2026
- 2025Guided Transfer Learning for Discrete Diffusion Models11 Dec 2025
- 2025Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language Models28 Sep 2025
- 2025Diffusion Language Models Know the Answer Before Decoding27 Aug 2025
- 2025A Survey on Diffusion Language Models14 Aug 2025
- 2025A Survey on Parallel Text Generation: From Parallel Decoding to Diffusion Language Models12 Aug 2025
Similar, not cited
Close in wording, no citation either way
- 2025Sequential Diffusion Language Models
- 2026Diffusion Language Models: An Experimental Analysis
- 2025Dream 7B: Diffusion Large Language Models
- 2023Likelihood-Based Diffusion Language Models
- 2026Breaking the Factorization Barrier in Diffusion Language Models
- 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models
- 2025Breaking AR's Sampling Bottleneck: Provable Acceleration via Diffusion Language Models
- 2026Hierarchical Continuous Diffusion Language Models