- The paper presents an in-depth taxonomy of diffusion language models, categorizing them into continuous, discrete, and hybrid types.
- It demonstrates efficient training and inference methods, including parallel decoding and iterative denoising, which reduce latency and enhance performance.
- The survey highlights practical applications in NLP, code generation, and biological modeling while addressing scalability and infrastructure challenges.
Comprehensive Survey of Diffusion LLMs
Introduction and Historical Context
Diffusion LLMs (DLMs) have emerged as a non-autoregressive alternative to the dominant autoregressive (AR) paradigm in LLMing. DLMs generate tokens in parallel via iterative denoising, offering inherent advantages in inference latency and bidirectional context modeling. The survey systematically traces the evolution of DLMs, from early continuous-space models inspired by image diffusion to discrete-space and multimodal extensions. The historical trajectory reveals a shift from continuous DLMs to discrete DLMs, with recent years marked by rapid growth in both research activity and practical deployments.
Figure 1: Timeline of Diffusion LLMs, illustrating the transition from continuous to discrete and multimodal DLMs.
Figure 2: Trend of diffusion LLM papers, showing a marked increase in research interest, especially in discrete DLMs.
Modeling Paradigms: Continuous, Discrete, and Hybrid
DLMs are categorized by the space in which the diffusion process operates:
- Continuous DLMs: Tokens are mapped to embeddings, and denoising occurs in continuous space. Early models (Diffusion-LM, SED) leveraged continuous diffusion for controllable text generation and infilling. Recent advances (TESS, TESS-2) diffuse over logit simplex representations and scale up via adaptation from AR models.
- Discrete DLMs: The diffusion process is defined directly on token space. D3PM introduced structured token corruption, while masked DLMs (LLaDA, Dream) employ iterative mask-predict denoising with cross-entropy loss over masked positions. These models achieve competitive performance with AR baselines, especially when scaled to billions of parameters.
- Hybrid AR-Diffusion Models: Block-wise semi-autoregressive models (BD3-LM) combine AR and diffusion, generating blocks autoregressively and tokens within blocks in parallel. This design enables variable-length generation and efficient caching.
Figure 3: Overview of training and inference procedures across AR, continuous, discrete, and block-wise diffusion paradigms.
Training and Post-Training Strategies
DLMs are typically pretrained using strategies analogous to AR models or image diffusion models. Initialization from pretrained AR models (DiffuLLaMA, Dream) accelerates training and enables competitive performance. Supervised fine-tuning (SFT) and reinforcement learning (RL) are adapted for DLMs, with unique challenges due to the non-factorized likelihood and parallel generation.
Post-training for reasoning capabilities is a focal area:
- Chain-of-Thought Parallelization: DoT adapts CoT reasoning to parallel denoising, enabling self-correction and outperforming larger AR models on reasoning benchmarks.
- Policy Gradient Methods: SEPO, diffu-GRPO, coupled-GRPO, and UniGRPO introduce efficient log-probability estimation and structured masking for RL fine-tuning, overcoming the intractability of sequence likelihood in DLMs.
- Preference Optimization: VRPO adapts DPO to DLMs via variance reduction techniques, yielding strong improvements in alignment and reasoning.
Inference Techniques and Efficiency
Inference in DLMs is characterized by parallel decoding, adaptive unmasking/remasking, guidance, and caching:
- Parallel Decoding: Confidence-aware and adaptive strategies (Fast-dLLM, APD, SlowFast Sampling, SpecDiff) achieve up to 34× speed-ups with minimal quality loss.
- Unmasking/Remasking: Selective refinement of low-confidence tokens improves coherence and convergence.
- Guidance: Classifier-free guidance and structural constraints steer generation toward desired attributes.
- Caching and Distillation: KV and feature caches, along with step distillation (DLM-One), reduce per-step and total inference cost, bringing DLM latency close to AR models.
Figure 4: Inference techniques for DLMs, including parallel decoding, unmasking/remasking, guidance, KV/feature cache, and step distillation.
Multimodal and Unified DLMs
DLMs have been extended to multimodal and unified architectures:
- Vision-LLMs: LLaDA-V, LaViDa, and Dimple integrate vision encoders and project image features into token space, supporting visual instruction tuning and multimodal reasoning.
- Unified Multimodal Generation: MMaDA, D-DiT, UniDisc, Fudoki, and Muddit tokenize images and text into a shared vocabulary, enabling joint modeling and cross-modal inpainting. These models demonstrate competitive or superior performance to AR-based VLMs and image generators.
Empirical results across benchmarks (GenEval, MME, CQA, Hellaswag, PIQA, HumanEval, GSM8K, MMMU) indicate that DLMs match or exceed AR models of similar scale in math, science, code, and multimodal tasks. Notably, LLaDA and Dream outperform AR baselines on GSM8K and HumanEval, while MMaDA and LLaDA-V surpass AR-based VLMs in multimodal reasoning.
Figure 5: Performance comparison on eight benchmarks, showing DLMs (orange) competitive with AR models (blue) across tasks and scales.
Applications in NLP, Code, and Biology
DLMs have been applied to a wide range of downstream tasks:
- NLP: Text classification, NER, sentiment analysis, summarization, style transfer, constrained generation, and machine translation.
- Code Generation: DiffuCoder and Mercury Coder demonstrate strong performance and throughput, leveraging parallel planning and coupled sampling.
- Biological Sequence Modeling: TransDLM, TGM-DLM, DRAKES, ForceGen, MeMDLM, DPLM, DPLM2, and CFP-GEN apply DLMs to molecular optimization, protein design, and multimodal sequence-structure co-generation.
Challenges and Future Directions
Key challenges for DLMs include:
Future research directions include improving training efficiency, exploring quantization and pruning, advancing multimodal unified reasoning, and developing DLM-based agents.
Conclusion
This survey provides a comprehensive taxonomy and analysis of diffusion LLMs, covering foundational principles, training and inference strategies, multimodal extensions, performance, and applications. DLMs offer a compelling alternative to AR models, with strong empirical results and unique capabilities in parallel generation and bidirectional context modeling. Addressing current challenges in scalability, infrastructure, and long-sequence handling will be critical for realizing the full potential of DLMs in real-world AI systems.