DiffusionART: Efficient, Stable and Scalable Reinforcement of Diffusion Models
On the map: LLM reasoning → Diffusion reasoning and test-time scaling also fits: Reinforcement learning for reasoning
also in 1 more
Abstract (arXiv)
Online reinforcement learning (RL) fine-tunes diffusion models with black-box rewards, but reverse-process methods such as Flow-GRPO are limited in efficiency by poor credit assignment, while forward-process methods such as DiffusionNFT are efficient yet unstable without carefully tuned exponential-moving-average (EMA) off-policy training. We show that NFT-style objectives contain a hidden forward--reverse consistency force that joint positive--negative training cancels only partially, and that EMA stabilizes training through restoration rather than off-policy rollouts. We introduce DiffusionART (Diffusion Reinforcement with Aggregation-First Reward Tilting), which aggregates each prompt's rewards into a shared reward-improving direction for reverse-process states, removing the consistency force by construction, and adds a trust-region-masked second update for drift-adaptive restoration without EMA tuning. It requires no differentiable reward model, remains effective with fewer samples per prompt, and supports pairwise reward models (DiffusionART-Pairwise). Across image and video generation, DiffusionART trains efficiently and stably without per-task tuning. In particular, on HunyuanVideo-13B with equally weighted visual and motion quality rewards, DiffusionART reaches the late-stage mean reward levels of DiffusionNFT and Flow-GRPO-Fast with 12.02 and 11.72 times fewer GPU hours, respectively, while achieving 28.8% and 53.0% higher mean training rewards.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion reasoning and test-time scaling, one of 17 directions in LLM reasoning. Builds on 26 of its 30 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
26 of 30 references.
References · 26
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
- 2024HunyuanVideo: A Systematic Framework For Large Video Generative Modelscited by 1,660
- 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
- 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
- 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
- 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
14 more
- 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
- 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 413
- 2025Improving Video Generation with Human Feedbackcited by 284
- 2025Seedance 1.0: Exploring the Boundaries of Video Generation Modelscited by 259
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Modelscited by 97
- 2025Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Modelscited by 67
- 2025Coefficients-Preserving Sampling for Reinforcement Learning with Flow Matchingcited by 56
- 2025GRPO-Guard: Mitigating Implicit Over-Optimization in Flow Matching via Regulated Clippingcited by 49
- 2025Tilt Matching for Scalable Sampling and Fine-Tuningcited by 22
- 2026Flow-DPPO: Divergence Proximal Policy Optimization for Flow Matching Modelscited by 6
- 2026Diffusion Reinforcement Learning via Centered Reward Distillationcited by 5
- 2026Qwen-Image-2.0-RL Technical Reportcited by 5
- 2026Designing Reinforcement Learning for Diffusion Models: A Unified Path-Space Viewcited by 3
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026DiffusionOPD: A Unified Perspective of On-Policy Distillation in Diffusion Models
- 2026Explicit Critic Guidance for Aligning Diffusion Models
- 2026Reverse Flow Matching: A Unified Framework for Online Reinforcement Learning with Diffusion and Flow Policies
- 2026Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion
- 2026On-Policy Self-Distillation in Diffusion Models
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
- 2024Avoiding mode collapse in diffusion models fine-tuned with reinforcement learning
- 2026iADD: Improving Alignment and Diversity in Diffusion Policy Optimization