mlsanityData to 14 Oct 2026

Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspective

arXiv 2512.03759 ↗First posted 3 Dec 2025cs.CL cs.AI cs.LGcited by 34

Not on the map. Why?

Cited by 27 in Diffusion models, but placed by its own abstract, which didn’t fit.

Abstract (arXiv)

Reinforcement Learning (RL) has proven highly effective for autoregressive language models, but adapting these methods to diffusion large language models (dLLMs) presents fundamental challenges. The core difficulty lies in likelihood approximation: while autoregressive models naturally provide token-level conditional probabilities essential for token-level RL objectives (e.g., GRPO), dLLMs generate sequences through iterative non-autoregressive denoising steps that lack this factorization. To address this fundamental mismatch, we propose ELBO-based Sequence-level Policy Optimization (ESPO), a principled RL framework that treats entire sequence generation as a single action and uses the ELBO as a tractable sequence-level likelihood proxy. Our method incorporates per-token normalization of importance ratios and robust KL-divergence estimation to ensure stable large-scale training. Extensive experiments on mathematical reasoning, coding, and planning tasks demonstrate that ESPO significantly outperforms token-level baselines, achieving dramatic improvements of 20-40 points on the Countdown task, while maintaining consistent gains on math and coding benchmarks. Our approach establishes sequence-level optimization as a principled and empirically effective paradigm for RL in dLLMs. Our code is available at https://github.com/ML-GSAI/ESPO.

Key facts. Cited by 34 papers in the map (data through 14 Oct 2026). Builds on 44 of its 55 references that are in the map. First posted on arXiv 3 Dec 2025.

Builds on

44 of 55 references.

References · 44

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2022Training language models to follow instructions with human feedbackcited by 15,971
  3. 2017Proximal Policy Optimization Algorithmscited by 15,215
  4. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  6. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  7. 2020Scaling Laws for Neural Language Modelscited by 6,586
  8. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  9. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  10. 2021Program Synthesis with Large Language Modelscited by 3,637
  11. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
  12. 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
32 more
  1. 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
  2. 2022MaskGIT: Masked Generative Image Transformercited by 1,099
  3. 2025Large Language Diffusion Modelscited by 912
  4. 2024Simple and Effective Masked Diffusion Language Modelscited by 848
  5. 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 626
  6. 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
  7. 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
  8. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
  9. 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 381
  10. 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
  11. 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
  12. 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusionscited by 214
  13. 2024Scaling up Masked Diffusion Models on Textcited by 199
  14. 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 187
  15. 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 186
  16. 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 178
  17. 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 169
  18. 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 148
  19. 2025dKV-Cache: The Cache for Diffusion Language Modelscited by 143
  20. 2025Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decodingcited by 101
  21. 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 93
  22. 2025LaViDa: A Large Diffusion Language Model for Multimodal Understandingcited by 84
  23. 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusioncited by 77
  24. 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 72
  25. 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Modelscited by 71
  26. 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Modelscited by 54
  27. 2025Diffusion Language Models are Super Data Learnerscited by 48
  28. 2025LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMscited by 40
  29. 2025Fine-Tuning Discrete Diffusion Models with Policy Gradient Methodscited by 38
  30. 2025On a few pitfalls in KL divergence gradient estimation for RLcited by 29
  31. 2025Target Concrete Score Matching: A Holistic Framework for Discrete Diffusioncited by 25
  32. 2025Effective and Efficient Masked Image Generation Modelscited by 13

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 34.

34 papers.

Sort:

By area:

  1. 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editingcited by 54
  2. 2026Improved Large Language Diffusion Modelscited by 14
  3. 2026LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Modelcited by 10
  4. 2026Stabilizing Reinforcement Learning for Diffusion Language Modelscited by 10
  5. 2026Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Modelscited by 9
  6. 2026DLLM-Searcher: Adapting Diffusion Large Language Model for Search Agentscited by 8
  7. 2026LaDi-RL: Latent Diffusion Reasoning Prevents Entropy Collapse in Reinforcement Learningcited by 7
  8. 2026LaViDa-R1: Advancing Reasoning for Unified Multimodal Diffusion Language Modelscited by 7
  9. 2026Trace-Based On-Policy Distillation for Masked Diffusion Language Modelscited by 7
  10. 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learningcited by 6
  11. 2026GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Modelscited by 6
  12. 2026Learning from the Self-future: On-policy Self-distillation for dLLMscited by 5
22 more
  1. 2026DreamReasoner-8B: Block-Size Curriculum Learning for Diffusion Reasoning Modelscited by 5
  2. 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantagescited by 4
  3. 2026Beyond Mode-Seeking RL: Trajectory-Balance Post-Training for Diffusion Language Modelscited by 3
  4. 2026Mask-Aware Policy Gradients for Diffusion Language Modelscited by 3
  5. 2026dFlowGRPO: Rate-Aware Policy Optimization for Discrete Flow Modelscited by 2
  6. 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLMcited by 2
  7. 2026A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Modelscited by 2
  8. 2026Relative Score Policy Optimization for Diffusion Language Modelscited by 1
  9. 2026Self-Distilled Trajectory-Aware Boltzmann Modeling: Bridging the Training-Inference Discrepancy in Diffusion Language Modelscited by 1
  10. 2026ForkLeft: Entropy-First Rollouts for Prefix-Aligned Autoregressive-to-Diffusion Distillationcited by 1
  11. 2026Hesitation-Aware On-Policy Distillation for Diffusion Language Modelscited by 1
  12. 2026Reinforcement Learning from Denoising Feedbackcited by 0
  13. 2026Read the Trace, Steer the Path: Trajectory-Aware Reinforcement Learning for Diffusion Language Modelscited by 0
  14. 2026SLIM-RL: Risk-Budgeted Random-Masking RL for Diffusion LLMs Without Trajectory Slicingcited by 0
  15. 2026Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMscited by 0
  16. 2026Conditional Generation of Creative Chess Puzzles with Diffusion Modelscited by 0
  17. 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Modelscited by 0
  18. 2026Exploring More, Reasoning Better: Stepwise Risk-Sensitive GRPO for Diffusion Language Modelscited by 0
  19. 2026Towards Unbiased On-Policy Distillation for Block Diffusion Language Modelscited by 0
  20. 2026JustOPD: On-Policy Distillation from Autoregressive to Diffusion Language Modelscited by 0
  21. 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMscited by 0
  22. 2026ForkGRPO: Efficient Exploration and Fine-Grained Credit Assignment for Diffusion Language Modelscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization
  2. 2025Boundary-Guided Policy Optimization for Memory-efficient RL of Diffusion Large Language Models
  3. 2026LFPO: Likelihood-Free Policy Optimization for Masked Diffusion Models
  4. 2025Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization
  5. 2025Learning Unmasking Policies for Diffusion Language Models
  6. 2025Reasoning in Diffusion Large Language Models is Concentrated in Dynamic Confusion Zones
  7. 2025d-TreeRPO: Towards More Reliable Policy Optimization for Diffusion Language Models
  8. 2025Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Models