VideoAWR: Reinforcing Video Flow Models with Decoupled-Timestep Advantage-Weighted Rectification
Not on the map. Why?
Abstract (arXiv)
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 pdfstartlink attr/Border [0 0 0] user/Subtype /Link /A << /S /URI /URI (https://github.com/Fuzm7/FlowAWR) >>% textcolorpearDarkVideoAWRpdfendlink.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 40 references that are in the map. First posted on arXiv 9 Oct 2026.
Builds on
27 of 40 references.
References · 27
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,255
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2018Neural Ordinary Differential Equationscited by 3,779
- 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
- 2024CogVideoX: Text-to-Video Diffusion Models with An Expert Transformercited by 2,261
- 2023VBench: Comprehensive Benchmark Suite for Video Generative Modelscited by 1,822
- 2023Training Diffusion Models with Reinforcement Learningcited by 982
- 2019Advantage-Weighted Regression: Simple and Scalable Off-Policy Reinforcement Learningcited by 782
- 2022On Distillation of Guided Diffusion Modelscited by 725
15 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
- 2025DiffusionNFT: Online Diffusion Reinforcement with Forward Processcited by 230
- 2025HPSv3: Towards Wide-Spectrum Human Preference Scorecited by 211
- 2025MixGRPO: Unlocking Flow-based GRPO Efficiency with Mixed ODE-SDEcited by 178
- 2024VideoDPO: Omni-Preference Alignment for Video Diffusion Generationcited by 108
- 2024Video Diffusion Alignment via Reward Gradientscited by 107
- 2025TempFlow-GRPO: When Timing Matters for GRPO in Flow Modelscited by 97
- 2025LongCat-Video Technical Reportcited by 85
- 2025Diffusion Guidance Is a Controllable Policy Improvement Operatorcited by 58
- 2024OnlineVPO: Align Video Diffusion Model with Online Video-Centric Preference Optimizationcited by 41
- 2025Diffusion Models without Classifier-free Guidancecited by 33
- 2026FlowAWR: Online Adaptive Flow Reinforcement via Advantage-Weighted Rectificationcited by 6
- 2026Flash-GRPO: Efficient Alignment for Video Diffusion via One-Step Policy Optimizationcited by 4
Built on by
Not cited yet.