Data-regularized Reinforcement Learning for Diffusion Models at Scale
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
- 2025MIRA: Towards Mitigating Reward Hacking in Inference-Time Alignment of T2I Diffusion Models2 Oct
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Process19 Sep
- 2025DanceGRPO: Unleashing GRPO on Visual Generation12 May
- 2025Flow-GRPO: Training Flow Matching Models via Online RL8 May
- 2023Diffusion Model Alignment Using Direct Preference Optimization21 Nov
- 2023Training Diffusion Models with Reinforcement Learning22 May
Other references · 31
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2022Classifier-Free Diffusion Guidancecited by 5,717
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,151
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesiscited by 4,094
19 more
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,536
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,388
- 2023Consistency Modelscited by 1,778
- 2023VBench: Comprehensive Benchmark Suite for Video Generative Modelscited by 1,755
- 2016Concrete Problems in AI Safetycited by 1,582
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 950
- 2025Cosmos World Foundation Model Platform for Physical AIcited by 833
- 2025Improving Video Generation with Human Feedbackcited by 267
- 2025Seedance 1.0: Exploring the Boundaries of Video Generation Modelscited by 247
- 2025World Simulation with Video Foundation Models for Physical AIcited by 183
- 2025Cosmos-Reason1: From Physical Common Sense To Embodied Reasoningcited by 159
- 2024TFG: Unified Training-Free Guidance for Diffusion Modelscited by 103
- 2023Goodhart's Law in Reinforcement Learningcited by 37
- 2025Towards a Unified View of Large Language Model Post-Trainingcited by 36
- 2025KL-Regularized Reinforcement Learning is Designed to Mode Collapsecited by 29
- 2025Beyond Two-Stage Training: Cooperative SFT and RL for LLM Reasoningcited by 28
Built on by
Citing papers per year: 2026: 13.
13 papers.
Sort:
By area:
- 2026Reward Hacking in the Era of Large Models: Mechanisms, Emergent Misalignment, Challengescited by 30
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Modelscited by 23
- 2026Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Designcited by 17
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPOcited by 6
- 2026Beyond the Dirac Delta: Mitigating Diversity Collapse in Reinforcement Fine-Tuning for Versatile Image Generationcited by 5
- 2026Diffusion Reinforcement Learning via Centered Reward Distillationcited by 3
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shapingcited by 3
- 2026Efficient Adjoint Matching for Fine-tuning Diffusion Modelscited by 2
- 2026Advances in GRPO for Generation Models: A Surveycited by 1
- 2026Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Modelscited by 1
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matchingcited by 1
- 2026Guidance Contrastive Token Credit Assignment for Discrete Policy Optimizationcited by 0
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matching24 Aug 2026
- 2026Causal-rCM: A Unified Teacher-Forcing and Self-Forcing Open Recipe for Autoregressive Diffusion Distillation in Streaming Video Generation and Interactive World Models24 Jun 2026
- 2026The Reward Was in Your Data All Along: Correcting Flow Matching with Discriminator-Guided RL17 Jun 2026
- 2026Guidance Contrastive Token Credit Assignment for Discrete Policy Optimization28 May 2026
- 2026Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models26 May 2026
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPO14 May 2026
- 2026Efficient Adjoint Matching for Fine-tuning Diffusion Models12 May 2026
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping11 May 2026
- 2026Reward Hacking in the Era of Large Models: Mechanisms, Emergent Misalignment, Challenges15 Apr 2026
- 2026Diffusion Reinforcement Learning via Centered Reward Distillation14 Mar 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2026Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Design4 Feb 2026
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPOcited by 6
- 2026Diffusion Reinforcement Learning via Centered Reward Distillationcited by 3
- 2026Efficient Adjoint Matching for Fine-tuning Diffusion Modelscited by 2
- 2026Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Modelscited by 1
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matchingcited by 1
- 2026The Reward Was in Your Data All Along: Correcting Flow Matching with Discriminator-Guided RLcited by 0
- 2026Scaling Reinforcement Learning for Diffusion Models via Velocity Matching24 Aug 2026
- 2026The Reward Was in Your Data All Along: Correcting Flow Matching with Discriminator-Guided RL17 Jun 2026
- 2026Beyond Pairwise Preferences: Listwise Reward-Aware Alignment for Diffusion Models26 May 2026
- 2026RAVEN: Real-time Autoregressive Video Extrapolation with Consistency-model GRPO14 May 2026
- 2026Efficient Adjoint Matching for Fine-tuning Diffusion Models12 May 2026
- 2026Diffusion Reinforcement Learning via Centered Reward Distillation14 Mar 2026
- 2026Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Designcited by 17
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shapingcited by 3
- 2026Advances in GRPO for Generation Models: A Surveycited by 1
- 2026Guidance Contrastive Token Credit Assignment for Discrete Policy Optimizationcited by 0
- 2026Guidance Contrastive Token Credit Assignment for Discrete Policy Optimization28 May 2026
- 2026Power Reinforcement Post-Training of Text-to-Image Models with Super-Linear Advantage Shaping11 May 2026
- 2026Advances in GRPO for Generation Models: A Survey21 Feb 2026
- 2026Rethinking the Design Space of Reinforcement Learning for Diffusion Models: On the Importance of Likelihood Estimation Beyond Loss Design4 Feb 2026
Similar, not cited
Close in wording, no citation either way
- 2024Large-scale Reinforcement Learning for Diffusion Models
- 2024Forward KL Regularized Preference Optimization for Aligning Diffusion Policies
- 2024Scores as Actions: a framework of fine-tuning diffusion models by continuous-time reinforcement learning
- 2025Beyond Reward Margin: Rethinking and Resolving Likelihood Displacement in Diffusion Models via Video Generation
- 2023Score Regularized Policy Optimization through Diffusion Behavior
- 2026Selective Timestep Weighting and Advantage-Based Replay for Sample-Efficient Diffusion RLHF
- 2026CRAFT: Aligning Diffusion Models with Fine-Tuning Is Easier Than You Think
- 2025Diffusion Models for Reinforcement Learning: Foundations, Taxonomy, and Development