VideoAWR: Reinforcing Video Flow Models with Decoupled-Timestep Advantage-Weighted Rectification
cs.CV
Submitted: 2026-10-09
Updated: 2026-10-09
Code: https://github.com/Fuzm7/FlowAWR
License: http://creativecommons.org/licenses/by/4.0/
The gist: Regression-based RL for flow models mandates sampling without classifier-free guidance (CFG) to align the rectification target with the reference marginal velocity.
Abstract
Regression-based RL for flow models mandates sampling without classifier-free guidance (CFG) to align the rectification target with the reference marginal velocity. Extending Advantage-Weighted Rectification (AWR), which dynamically scales this regression target based on reward, to video generation encounters a cold-start pathology: unguided rollouts collapse into low-fidelity regions where intra-group reward gaps remain wide but no longer track perceptual quality, rendering the optimization target arbitrary. Conversely, utilizing guided rollouts to restore fidelity, while keeping the regression target tied to the guided sampling distribution, provably triggers a geometric guidance blow-up under exponential-moving-average (EMA) updates. We propose VideoAWR, which resolves this tension by untying sampling from the optimization anchor. An EMA reference policy generates guided rollouts to restore reward separability, while the trainable policy is rectified against the unguided conditional anchor. We prove this decoupled objective is mathematically exact: the centered advantage cancels the anchor in expectation, filtering out the reward-neutral guidance offset and retaining only the reward-correlated covariance increment. Consequently, inference runs single-branch and CFG-free, yet inference-time CFG preserves its original semantics. Furthermore, we introduce Decoupled Timestep Rectification, supervising each rollout at an independently sampled timestep without bias to improve sample efficiency. Experiments on Wan2.1-1.3B and 14B demonstrate stable optimization and superior visual and motion quality in both CFG-free and CFG-enabled regimes. Code is available at attr /Border [0 0 0] user /Subtype /Link /A << /S /URI /URI (https://github.com/Fuzm7/FlowAWR) >> % VideoAWR.
Sources
- Diffusion Guidance Is a Controllable Policy Improvement Operator
- FlowAWR: Online Adaptive Flow Reinforcement via Advantage-Weighted Rectification
- G$^2$RPO-A: Guided Group Relative Policy Optimization with Adaptive Guidance
- Flash-GRPO: Efficient Alignment for Video Diffusion via One-Step Policy Optimization
- TempFlow-GRPO: When Timing Matters for GRPO in Flow Models
- Classifier-Free Diffusion Guidance
- HunyuanVideo: A Systematic Framework For Large Video Generative Models
- MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDE
- VideoDPO: Omni-Preference Alignment for Video Diffusion Generation
- Training language models to follow instructions with human feedback
- Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learning
- Video Diffusion Alignment via Reward Gradients
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Diffusion Models without Classifier-free Guidance
- LongCat-Video Technical Report
- Wan: Open and Advanced Large-Scale Video Generative Models
- Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesis
- DanceGRPO: Unleashing GRPO on Visual Generation
- CogVideoX: Text-to-Video Diffusion Models with An Expert Transformer
- OnlineVPO: Align Video Diffusion Model with Online Video-Centric Preference Optimization
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models