Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspective
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
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,391
32 more
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
- 2022MaskGIT: Masked Generative Image Transformercited by 1,099
- 2025Large Language Diffusion Modelscited by 912
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 626
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
- 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 381
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
- 2025Train for the Worst, Plan for the Best: Understanding Token Ordering in Masked Diffusionscited by 214
- 2024Scaling up Masked Diffusion Models on Textcited by 199
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generationcited by 187
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 186
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 178
- 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 169
- 2025LLaDA-V: Large Language Diffusion Models with Visual Instruction Tuningcited by 148
- 2025dKV-Cache: The Cache for Diffusion Language Modelscited by 143
- 2025Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decodingcited by 101
- 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 93
- 2025LaViDa: A Large Diffusion Language Model for Multimodal Understandingcited by 84
- 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusioncited by 77
- 2024Amortizing intractable inference in diffusion models for vision, language, and controlcited by 72
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Modelscited by 71
- 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Modelscited by 54
- 2025Diffusion Language Models are Super Data Learnerscited by 48
- 2025LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMscited by 40
- 2025Fine-Tuning Discrete Diffusion Models with Policy Gradient Methodscited by 38
- 2025On a few pitfalls in KL divergence gradient estimation for RLcited by 29
- 2025Target Concrete Score Matching: A Holistic Framework for Discrete Diffusioncited by 25
- 2025Effective and Efficient Masked Image Generation Modelscited by 13
Built on by
Citing papers per year: 2026: 34.
34 papers.
Sort:
By area:
- 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editingcited by 54
- 2026Improved Large Language Diffusion Modelscited by 14
- 2026LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Modelcited by 10
- 2026Stabilizing Reinforcement Learning for Diffusion Language Modelscited by 10
- 2026Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Modelscited by 9
- 2026DLLM-Searcher: Adapting Diffusion Large Language Model for Search Agentscited by 8
- 2026LaDi-RL: Latent Diffusion Reasoning Prevents Entropy Collapse in Reinforcement Learningcited by 7
- 2026LaViDa-R1: Advancing Reasoning for Unified Multimodal Diffusion Language Modelscited by 7
- 2026Trace-Based On-Policy Distillation for Masked Diffusion Language Modelscited by 7
- 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learningcited by 6
- 2026GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Modelscited by 6
- 2026Learning from the Self-future: On-policy Self-distillation for dLLMscited by 5
22 more
- 2026DreamReasoner-8B: Block-Size Curriculum Learning for Diffusion Reasoning Modelscited by 5
- 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantagescited by 4
- 2026Beyond Mode-Seeking RL: Trajectory-Balance Post-Training for Diffusion Language Modelscited by 3
- 2026Mask-Aware Policy Gradients for Diffusion Language Modelscited by 3
- 2026dFlowGRPO: Rate-Aware Policy Optimization for Discrete Flow Modelscited by 2
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLMcited by 2
- 2026A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Modelscited by 2
- 2026Relative Score Policy Optimization for Diffusion Language Modelscited by 1
- 2026Self-Distilled Trajectory-Aware Boltzmann Modeling: Bridging the Training-Inference Discrepancy in Diffusion Language Modelscited by 1
- 2026ForkLeft: Entropy-First Rollouts for Prefix-Aligned Autoregressive-to-Diffusion Distillationcited by 1
- 2026Hesitation-Aware On-Policy Distillation for Diffusion Language Modelscited by 1
- 2026Reinforcement Learning from Denoising Feedbackcited by 0
- 2026Read the Trace, Steer the Path: Trajectory-Aware Reinforcement Learning for Diffusion Language Modelscited by 0
- 2026SLIM-RL: Risk-Budgeted Random-Masking RL for Diffusion LLMs Without Trajectory Slicingcited by 0
- 2026Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMscited by 0
- 2026Conditional Generation of Creative Chess Puzzles with Diffusion Modelscited by 0
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Modelscited by 0
- 2026Exploring More, Reasoning Better: Stepwise Risk-Sensitive GRPO for Diffusion Language Modelscited by 0
- 2026Towards Unbiased On-Policy Distillation for Block Diffusion Language Modelscited by 0
- 2026JustOPD: On-Policy Distillation from Autoregressive to Diffusion Language Modelscited by 0
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMscited by 0
- 2026ForkGRPO: Efficient Exploration and Fine-Grained Credit Assignment for Diffusion Language Modelscited by 0
- 2026ForkGRPO: Efficient Exploration and Fine-Grained Credit Assignment for Diffusion Language Models14 Oct 2026
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMs13 Oct 2026
- 2026JustOPD: On-Policy Distillation from Autoregressive to Diffusion Language Models12 Oct 2026
- 2026Towards Unbiased On-Policy Distillation for Block Diffusion Language Models4 Oct 2026
- 2026Exploring More, Reasoning Better: Stepwise Risk-Sensitive GRPO for Diffusion Language Models30 Sep 2026
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models30 Sep 2026
- 2026Conditional Generation of Creative Chess Puzzles with Diffusion Models29 Sep 2026
- 2026Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMs28 Sep 2026
- 2026Hesitation-Aware On-Policy Distillation for Diffusion Language Models27 Sep 2026
- 2026ForkLeft: Entropy-First Rollouts for Prefix-Aligned Autoregressive-to-Diffusion Distillation26 Sep 2026
- 2026Trace-Based On-Policy Distillation for Masked Diffusion Language Models18 Jul 2026
- 2026Mask-Aware Policy Gradients for Diffusion Language Models16 Jul 2026
22 more
- 2026A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Models16 Jul 2026
- 2026SLIM-RL: Risk-Budgeted Random-Masking RL for Diffusion LLMs Without Trajectory Slicing30 Jun 2026
- 2026Improved Large Language Diffusion Models24 Jun 2026
- 2026DreamReasoner-8B: Block-Size Curriculum Learning for Diffusion Reasoning Models17 Jun 2026
- 2026Learning from the Self-future: On-policy Self-distillation for dLLMs16 Jun 2026
- 2026Read the Trace, Steer the Path: Trajectory-Aware Reinforcement Learning for Diffusion Language Models3 Jun 2026
- 2026GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models28 May 2026
- 2026Reinforcement Learning from Denoising Feedback25 May 2026
- 2026Beyond Mode-Seeking RL: Trajectory-Balance Post-Training for Diffusion Language Models13 May 2026
- 2026Self-Distilled Trajectory-Aware Boltzmann Modeling: Bridging the Training-Inference Discrepancy in Diffusion Language Models12 May 2026
- 2026Relative Score Policy Optimization for Diffusion Language Models11 May 2026
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLM10 May 2026
- 2026dFlowGRPO: Rate-Aware Policy Optimization for Discrete Flow Models10 May 2026
- 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantages13 Mar 2026
- 2026Stabilizing Reinforcement Learning for Diffusion Language Models6 Mar 2026
- 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learning4 Mar 2026
- 2026LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Model1 Mar 2026
- 2026LaViDa-R1: Advancing Reasoning for Unified Multimodal Diffusion Language Models15 Feb 2026
- 2026Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Models10 Feb 2026
- 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editing9 Feb 2026
- 2026DLLM-Searcher: Adapting Diffusion Large Language Model for Search Agents3 Feb 2026
- 2026LaDi-RL: Latent Diffusion Reasoning Prevents Entropy Collapse in Reinforcement Learning2 Feb 2026
- 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editingcited by 54
- 2026Improved Large Language Diffusion Modelscited by 14
- 2026LLaDA-o: An Effective and Length-Adaptive Omni Diffusion Modelcited by 10
- 2026Where-to-Unmask: Ground-Truth-Guided Unmasking Order Learning for Masked Diffusion Language Modelscited by 9
- 2026Trace-Based On-Policy Distillation for Masked Diffusion Language Modelscited by 7
- 2026GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Modelscited by 6
- 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantagescited by 4
- 2026Beyond Mode-Seeking RL: Trajectory-Balance Post-Training for Diffusion Language Modelscited by 3
- 2026A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Modelscited by 2
- 2026Self-Distilled Trajectory-Aware Boltzmann Modeling: Bridging the Training-Inference Discrepancy in Diffusion Language Modelscited by 1
- 2026Hesitation-Aware On-Policy Distillation for Diffusion Language Modelscited by 1
- 2026Reinforcement Learning from Denoising Feedbackcited by 0
12 most cited of 19 All 19 (JSON)
- 2026ForkGRPO: Efficient Exploration and Fine-Grained Credit Assignment for Diffusion Language Models14 Oct 2026
- 2026JustOPD: On-Policy Distillation from Autoregressive to Diffusion Language Models12 Oct 2026
- 2026Towards Unbiased On-Policy Distillation for Block Diffusion Language Models4 Oct 2026
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models30 Sep 2026
- 2026Conditional Generation of Creative Chess Puzzles with Diffusion Models29 Sep 2026
- 2026Low-Confidence Remasking Traps Flexibility: Realizing Arbitrary-Order Potential for Diverse Rollouts in Diffusion LLMs28 Sep 2026
- 2026Hesitation-Aware On-Policy Distillation for Diffusion Language Models27 Sep 2026
- 2026Trace-Based On-Policy Distillation for Masked Diffusion Language Models18 Jul 2026
- 2026A Continuous-Time Reinforcement Learning Framework for Fine-Tuning Discrete Diffusion Models16 Jul 2026
- 2026Improved Large Language Diffusion Models24 Jun 2026
- 2026Read the Trace, Steer the Path: Trajectory-Aware Reinforcement Learning for Diffusion Language Models3 Jun 2026
- 2026GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models28 May 2026
12 newest of 19 All 19 (JSON)
- 2026Stabilizing Reinforcement Learning for Diffusion Language Modelscited by 10
- 2026LaDi-RL: Latent Diffusion Reasoning Prevents Entropy Collapse in Reinforcement Learningcited by 7
- 2026LaViDa-R1: Advancing Reasoning for Unified Multimodal Diffusion Language Modelscited by 7
- 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learningcited by 6
- 2026DreamReasoner-8B: Block-Size Curriculum Learning for Diffusion Reasoning Modelscited by 5
- 2026Mask-Aware Policy Gradients for Diffusion Language Modelscited by 3
- 2026Relative Score Policy Optimization for Diffusion Language Modelscited by 1
- 2026ForkLeft: Entropy-First Rollouts for Prefix-Aligned Autoregressive-to-Diffusion Distillationcited by 1
- 2026Exploring More, Reasoning Better: Stepwise Risk-Sensitive GRPO for Diffusion Language Modelscited by 0
- 2026Exploring More, Reasoning Better: Stepwise Risk-Sensitive GRPO for Diffusion Language Models30 Sep 2026
- 2026ForkLeft: Entropy-First Rollouts for Prefix-Aligned Autoregressive-to-Diffusion Distillation26 Sep 2026
- 2026Mask-Aware Policy Gradients for Diffusion Language Models16 Jul 2026
- 2026DreamReasoner-8B: Block-Size Curriculum Learning for Diffusion Reasoning Models17 Jun 2026
- 2026Relative Score Policy Optimization for Diffusion Language Models11 May 2026
- 2026Stabilizing Reinforcement Learning for Diffusion Language Models6 Mar 2026
- 2026LightningRL: Breaking the Accuracy-Parallelism Trade-off of Block-wise dLLMs via Reinforcement Learning4 Mar 2026
- 2026LaViDa-R1: Advancing Reasoning for Unified Multimodal Diffusion Language Models15 Feb 2026
- 2026LaDi-RL: Latent Diffusion Reasoning Prevents Entropy Collapse in Reinforcement Learning2 Feb 2026
- 2026dFlowGRPO: Rate-Aware Policy Optimization for Discrete Flow Modelscited by 2
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLMcited by 2
- 2026SLIM-RL: Risk-Budgeted Random-Masking RL for Diffusion LLMs Without Trajectory Slicingcited by 0
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMscited by 0
- 2026A Few More Tokens Ahead: Reinforcing Parallelism in Diffusion LLMs13 Oct 2026
- 2026SLIM-RL: Risk-Budgeted Random-Masking RL for Diffusion LLMs Without Trajectory Slicing30 Jun 2026
- 2026TAD: Temporal-Aware Trajectory Self-Distillation for Fast and Accurate Diffusion LLM10 May 2026
- 2026dFlowGRPO: Rate-Aware Policy Optimization for Discrete Flow Models10 May 2026
Similar, not cited
Close in wording, no citation either way
- 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization
- 2025Boundary-Guided Policy Optimization for Memory-efficient RL of Diffusion Large Language Models
- 2026LFPO: Likelihood-Free Policy Optimization for Masked Diffusion Models
- 2025Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimization
- 2025Learning Unmasking Policies for Diffusion Language Models
- 2025Reasoning in Diffusion Large Language Models is Concentrated in Dynamic Confusion Zones
- 2025d-TreeRPO: Towards More Reliable Policy Optimization for Diffusion Language Models
- 2025Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Models