mlsanityData to 14 Oct 2026

Beyond On-Policy Exploration: Integrating External Policy Rollouts for Reinforcement Learning in Diffusion Language Models

arXiv 2608.01717 ↗First posted 3 Aug 2026cs.LGcited by 0

On the map: Diffusion models → Discrete and text diffusion

Abstract (arXiv)

Recent reinforcement learning methods for diffusion large language models (dLLMs) commonly rely on on-policy rollouts generated by the target dLLM itself. When successful on-policy rollouts are scarce, however, on-policy training may receive little positive reward and make only limited progress. To mitigate this problem, we explore incorporating higher-reward rollouts generated by a stronger external policy alongside on-policy rollouts from the target dLLM. However, directly incorporating these external rollouts introduces two practical challenges: differences in rollout length and instability when jointly processing rewards from on-policy and external rollouts. To address these challenges, we propose External Rollout Integration with Length Control and Source-Specific Processing (ERILS), which controls external-rollout length and processes the rewards of on-policy and external rollouts separately. Experiments on Sudoku, Countdown, and MATH500 under zero-shot evaluation show that ERILS improves multi-sample performance across all three tasks, with the largest gains on Sudoku. On Sudoku, ERILS achieves 98.4% best-of-4 completion accuracy, compared with 40.3% for the strongest baseline. ERILS also maintains approximately 90% deterministic single-completion accuracy on Sudoku across generation lengths of 128, 256, and 512 tokens. Our component analysis further shows that length-controlled external rollouts are more effective than uncontrolled external rollouts, and that source-specific reward processing avoids the training collapse observed with joint reward processing. These results show that rollout construction and reward processing are important design dimensions when integrating external rollouts into dLLM reinforcement learning.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Discrete and text diffusion, one of 14 directions in Diffusion models. Builds on 27 of its 43 references that are in the map. First posted on arXiv 3 Aug 2026.

Builds on

27 of 43 references.

Closest ancestors · 17

Same direction, newest first

  1. 2025SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models10 Oct
  2. 2025Improving Reasoning for Diffusion Language Models via Group Diffusion Policy Optimization9 Oct
  3. 2025DiFFPO: Training Diffusion LLMs to Reason Fast and Furious via Reinforcement Learning2 Oct
  4. 2025Inpainting-Guided Policy Optimization for Diffusion Large Language Models12 Sep
  5. 2025Blockwise SFT for Diffusion Language Models: Reconciling Bidirectional Attention and Autoregressive Decoding27 Aug
  6. 2025Dream 7B: Diffusion Large Language Models21 Aug
  7. 2025Diffusion LLMs Can Do Faster-Than-AR Inference via Discrete Diffusion Forcing8 Aug
  8. 2025wd1: Weighted Policy Optimization for Reasoning in Diffusion Language Models7 Jul
  9. 2025DiffuCoder: Understanding and Improving Masked Diffusion Models for Code Generation25 Jun
  10. 2025LongLLaDA: Unlocking Long Context Capabilities in Diffusion LLMs17 Jun
  11. 2025FlashDLM: Accelerating Diffusion Language Model Inference via Efficient KV Caching and Guided Diffusion27 May
  12. 2025dKV-Cache: The Cache for Diffusion Language Models21 May
5 more
  1. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Models12 Mar
  2. 2025Large Language Diffusion Models14 Feb
  3. 2024Scaling Diffusion Language Models via Adaptation from Autoregressive Models23 Oct
  4. 2024Simplified and Generalized Masked Diffusion for Discrete Data6 Jun
  5. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution25 Oct

Other references · 10

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  2. 2023Let's Verify Step by Stepcited by 3,789
  3. 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
  4. 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
  5. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
  6. 2025Fast-dLLM: Training-free Acceleration of Diffusion LLM by Enabling KV Cache and Parallel Decodingcited by 420
  7. 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 285
  8. 2025Learning to Reason under Off-Policy Guidancecited by 281
  9. 2025d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learningcited by 186
  10. 2025dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Cachingcited by 169

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Rollout-Time Intervention: Unlocking Agentic Reinforcement Learning Beyond Policy Limits
  2. 2026Rollout Efficiency in Reinforcement Learning for Reasoning Large Language Models: A Taxonomy and Future Directions
  3. 2025APRIL: Active Partial Rollouts in Reinforcement Learning to Tame Long-tail Generation
  4. 2026ROSS: Relearning from Self-Generated Rollouts through Selective Supervision
  5. 2026Generate, Filter, Control, Replay: A Comprehensive Survey of Rollout Strategies for LLM Reinforcement Learning
  6. 2026Single-Rollout Asynchronous Optimization for Agentic Reinforcement Learning
  7. 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning
  8. 2026RollVerify: Bridging Efficiency and Accuracy in Long-Tail Rollout Reinforcement Learning