GDSD: Reinforcement Learning as Guided Denoiser Self-Distillation for Diffusion Language Models
On the map: Diffusion models → Discrete and text diffusion
Abstract (arXiv)
Reinforcement learning (RL) can be used to improve the policy (denoiser) of diffusion large language models (dLLMs), while being hindered by the intractability of the policy likelihood. A dominant and efficient family of methods replaces the likelihood in standard RL with its evidence lower bound (ELBO), estimated from randomly masked sequences. Despite being well aligned with pre-training, these approaches introduce bias through training--inference mismatch by using the ELBO as a likelihood surrogate, which can degrade performance. In this work, we propose Guided Denoiser Self-Distillation (GDSD) to directly distill the denoiser of dLLMs from an advantage-guided self-teacher, derived from the closed-form optimum of reverse-KL regularized RL. GDSD matches the dLLM's denoiser logits to the teacher's via a normalization-free objective, which reduces RL to likelihood-free self-distillation and thus bypasses the TIM biases. Recent ELBO-based methods emerge as instances of applying different distillation divergences, but with diagnosable pathologies that GDSD avoids. On planning, math, and coding benchmarks with LLaDA-8B and Dream-7B, GDSD consistently outperforms prior state-of-the-art ELBO-based methods with a more stable training reward dynamics, achieving test-accuracy improvements of up to +19.6%. These results suggest that direct denoiser self-distillation, without relying on an ELBO likelihood surrogate, can provide a more stable and effective RL procedure for dLLMs. Code is available at https://github.com/GaryBall/GDSD.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 56 of its 65 references that are in the map. First posted on arXiv 28 May 2026.
Builds on
56 of 65 references.
Closest ancestors · 18
Same direction, newest first
- 2026Stabilizing Reinforcement Learning for Diffusion Language Models6 Mar
- 2026LFPO: Likelihood-Free Policy Optimization for Masked Diffusion Models2 Mar
- 2026DUEL: Exact Likelihood for Masked Diffusion via Deterministic Unmasking2 Mar
- 2026LLaDA2.1: Speeding Up Text Diffusion via Token Editing9 Feb
- 2025dUltra: Ultra-Fast Diffusion Language Models via Reinforcement Learning24 Dec
- 2025LLaDA2.0: Scaling Up Diffusion Language Models to 100B10 Dec
- 2025SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models10 Oct
- 2025d2: Improving Reasoning in Diffusion Language Models via Trajectory Likelihood Estimation25 Sep
- 2025Revolutionizing Reinforcement Learning Framework for Diffusion Large Language Models8 Sep
- 2025Dream 7B: Diffusion Large Language Models21 Aug
- 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
- 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation25 Jun
6 more
- 2025Reinforcing the Diffusion Chain of Lateral Thought with Diffusion Language Models15 May
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
- 2025Large Language Diffusion Models14 Feb
- 2024Simple and Effective Masked Diffusion Language Models11 Jun
- 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Data6 Jun
Other references · 38
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 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
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
- 2023Let's Verify Step by Stepcited by 3,789
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2023Diffusion Policy: Visuomotor Policy Learning via Action Diffusioncited by 2,838
- 2015Trust Region Policy Optimizationcited by 2,580
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
26 more
- 2022Planning with Diffusion for Flexible Behavior Synthesiscited by 1,061
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2022Diffusion Policies as an Expressive Policy Class for Offline Reinforcement Learningcited by 562
- 2023IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policiescited by 325
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
- 2024Diffusion Policy Policy Optimizationcited by 280
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 250
- 2025Mercury: Ultra-Fast Language Models Based on Diffusioncited by 235
- 2024Self-Play Preference Optimization for Language Model Alignmentcited by 232
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 186
- 2023Contrastive Energy Prediction for Exact Energy-Guided Diffusion Sampling in Offline Reinforcement Learningcited by 149
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
- 2025Flow Matching Policy Gradientscited by 99
- 2025ACECODER: Acing Coder RL via Automated Test-Case Synthesiscited by 93
- 2025Energy-Weighted Flow Matching for Offline Reinforcement Learningcited by 73
- 2023Score Regularized Policy Optimization through Diffusion Behaviorcited by 71
- 2025Defeating the Training-Inference Mismatch via FP16cited by 66
- 2023Fine-Tuning Language Models with Advantage-Induced Policy Alignmentcited by 48
- 2025What Makes a Good Diffusion Planner for Decision Making?cited by 38
- 2025Principled RL for Diffusion LLMs Emerges from a Sequence-Level Perspectivecited by 34
- 2025Enhancing Reasoning for Diffusion LLMs via Distribution Matching Policy Optimizationcited by 20
- 2026Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Designcited by 17
- 2025Generalizing from SIMPLE to HARD Visual Reasoning: Can We Mitigate Modality Imbalance in VLMs?cited by 14
- 2025Safe and Stable Control via Lyapunov-Guided Diffusion Modelscited by 10
- 2025RSPO: Regularized Self-Play Alignment of Large Language Modelscited by 5
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Temporal Self-Distillation: Faster Inference in Discrete Diffusion Language Modelscited by 1
- 2026Semantic DLM+: Improving Diffusion Language Models through Bias-variance Trade-off in Transition Kernel Designcited by 0
- 2026CanvasAnneal: Curriculum Reinforcement Learning for Diffusion Language Modelscited by 0
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Modelscited by 0
- 2026RL Post-Training Diffusion Language Models Along Their Own Generation Pathscited by 0
- 2026Outcome-Routed Distillation and Reinforcement Learning for Diffusion Language Modelscited by 0
- 2026Outcome-Routed Distillation and Reinforcement Learning for Diffusion Language Models13 Oct 2026
- 2026RL Post-Training Diffusion Language Models Along Their Own Generation Paths12 Oct 2026
- 2026Fork-dLLM: Avoiding the Flexibility Trap in Diffusion Language Models30 Sep 2026
- 2026Temporal Self-Distillation: Faster Inference in Discrete Diffusion Language Models14 Sep 2026
- 2026CanvasAnneal: Curriculum Reinforcement Learning for Diffusion Language Models11 Sep 2026
- 2026Semantic DLM+: Improving Diffusion Language Models through Bias-variance Trade-off in Transition Kernel Design13 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Reinforcement Learning for Diffusion LLMs with Entropy-Guided Step Selection and Stepwise Advantages
- 2026dOPSD: On-Policy Self-Distillation for Diffusion Language Models
- 2026Reinforcement Learning from Denoising Feedback
- 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization
- 2025MDPO: Overcoming the Training-Inference Divide of Masked Diffusion Language Models
- 2025Learning Unmasking Policies for Diffusion Language Models
- 2026Learning from the Self-future: On-policy Self-distillation for dLLMs
- 2025d-TreeRPO: Towards More Reliable Policy Optimization for Diffusion Language Models