mlsanityData to 7 Oct 2026

Data-regularized Reinforcement Learning for Diffusion Models at Scale

arXiv 2512.04332 ↗First posted 3 Dec 2025cs.LGcited by 13

On the map: Diffusion models → Diffusion and flow policies for robotics

Abstract (arXiv)

Aligning generative diffusion models with human preferences via reinforcement learning (RL) is critical yet challenging. Most existing algorithms are often vulnerable to reward hacking, such as quality degradation, over-stylization, or reduced diversity. Our analysis demonstrates that this can be attributed to the inherent limitations of their regularization, which provides unreliable penalties. We introduce Data-regularized Diffusion Reinforcement Learning (DDRL), a novel framework that uses the forward KL divergence to anchor the policy to an off-policy data distribution. Theoretically, DDRL enables robust, unbiased integration of RL with standard diffusion training. Empirically, this translates into a simple yet effective algorithm that combines reward maximization with diffusion loss minimization. With over a million GPU hours of experiments and ten thousand double-blind human evaluations, we demonstrate on high-resolution video generation tasks that DDRL significantly improves rewards while alleviating the reward hacking seen in baselines, achieving the highest human preference and establishing a robust and scalable paradigm for diffusion post-training.

Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Placed in Diffusion and flow policies for robotics, one of 14 directions in Diffusion models. Builds on 37 of its 44 references that are in the map. First posted on arXiv 3 Dec 2025.

Builds on

37 of 44 references.

Closest ancestors · 6

Same direction, newest first

  1. 2025MIRA: Towards Mitigating Reward Hacking in Inference-Time Alignment of T2I Diffusion Models2 Oct
  2. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Process19 Sep
  3. 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
  4. 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
  5. 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
  6. 2023Training Diffusion Models with Reinforcement Learning22 May

Other references · 31

  1. 2023GPT-4 Technical Reportcited by 18,132
  2. 2022Training language models to follow instructions with human feedbackcited by 15,712
  3. 2020Denoising Diffusion Implicit Modelscited by 9,655
  4. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  6. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  7. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
  8. 2022Classifier-Free Diffusion Guidancecited by 5,717
  9. 2022Flow Matching for Generative Modelingcited by 5,426
  10. 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,151
  11. 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
  12. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
19 more
  1. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
  2. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
  3. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  4. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
  5. 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,388
  6. 2023Consistency Modelscited by 1,778
  7. 2023VBench: Comprehensive Benchmark Suite for Video Generative Modelscited by 1,755
  8. 2016Concrete Problems in AI Safetycited by 1,582
  9. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
  10. 2025Cosmos World Foundation Model Platform for Physical AIcited by 833
  11. 2025Improving Video Generation with Human Feedbackcited by 267
  12. 2025Seedance 1.0: Exploring the Boundaries of Video Generation Modelscited by 247
  13. 2025World Simulation with Video Foundation Models for Physical AIcited by 183
  14. 2025Cosmos-Reason1: From Physical Common Sense To Embodied Reasoningcited by 159
  15. 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 103
  16. 2023Goodhart's Law in Reinforcement Learningcited by 37
  17. 2025Towards a Unified View of Large Language Model Post-Trainingcited by 36
  18. 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
  19. 2025Beyond Two-Stage Training: Cooperative SFT and RL for LLM Reasoningcited by 28

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 13.

13 papers.

Sort:

By area:

Similar, not cited

Close in wording, no citation either way

  1. 2024Large-scale Reinforcement Learning for Diffusion Models
  2. 2024Forward KL Regularized Preference Optimization for Aligning Diffusion Policies
  3. 2024Scores as Actions: a framework of fine-tuning diffusion models by continuous-time reinforcement learning
  4. 2025Beyond Reward Margin: Rethinking and Resolving Likelihood Displacement in Diffusion Models via Video Generation
  5. 2023Score Regularized Policy Optimization through Diffusion Behavior
  6. 2026Selective Timestep Weighting and Advantage-Based Replay for Sample-Efficient Diffusion RLHF
  7. 2026CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think
  8. 2025Diffusion Models for Reinforcement Learning: Foundations, Taxonomy, and Development