mlsanityData to 14 Oct 2026

DiffusionART: Efficient, Stable and Scalable Reinforcement of Diffusion Models

arXiv 2610.12572 ↗First posted 8 Oct 2026cs.LGcited by 0

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

  1. 2017Proximal Policy Optimization Algorithmscited by 15,215
  2. 2022Flow Matching for Generative Modelingcited by 5,702
  3. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
  4. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
  5. 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
  6. 2024HunyuanVideo: A Systematic Framework For Large Video Generative Modelscited by 1,660
  7. 2023ImageReward: Learning and Evaluating Human Preferences for Text-to-Image Generationcited by 1,045
  8. 2023Training Diffusion Models with Reinforcement Learningcited by 982
  9. 2023Pick-a-Pic: An Open Dataset of User Preferences for Text-to-Image Generationcited by 977
  10. 2023GenEval: An Object-Focused Framework for Evaluating Text-to-Image Alignmentcited by 972
  11. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 890
  12. 2023Diffusion Model Alignment Using Direct Preference Optimizationcited by 790
14 more
  1. 2025Flow-GRPO: Training Flow Matching Models via Online RLcited by 661
  2. 2025DanceGRPO: Unleashing GRPO on Visual Generationcited by 413
  3. 2025Improving Video Generation with Human Feedbackcited by 284
  4. 2025Seedance 1.0: Exploring the Boundaries of Video Generation Modelscited by 259
  5. 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
  6. 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Modelscited by 97
  7. 2025Advantage Weighted Matching: Aligning RL with Pretraining in Diffusion Modelscited by 67
  8. 2025Coefficients-Preserving Sampling for Reinforcement Learning with Flow Matchingcited by 56
  9. 2025GRPO-Guard: Mitigating Implicit Over-Optimization in Flow Matching via Regulated Clippingcited by 49
  10. 2025Tilt Matching for Scalable Sampling and Fine-Tuningcited by 22
  11. 2026Flow-DPPO: Divergence Proximal Policy Optimization for Flow Matching Modelscited by 6
  12. 2026Diffusion Reinforcement Learning via Centered Reward Distillationcited by 5
  13. 2026Qwen-Image-2.0-RL Technical Reportcited by 5
  14. 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

  1. 2026DiffusionOPD: A Unified Perspective of On-Policy Distillation in Diffusion Models
  2. 2026Explicit Critic Guidance for Aligning Diffusion Models
  3. 2026Reverse Flow Matching: A Unified Framework for Online Reinforcement Learning with Diffusion and Flow Policies
  4. 2026Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion
  5. 2026On-Policy Self-Distillation in Diffusion Models
  6. 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewards
  7. 2024Avoiding mode collapse in diffusion models fine-tuned with reinforcement learning
  8. 2026iADD: Improving Alignment and Diversity in Diffusion Policy Optimization