Video Generation Models Are Good Latent Reward Models
By: Xiaoyue Mi , Wenqing Yu , Jiesong Lian and more
Potential Business Impact:
Makes AI videos better and faster to create.
Reward feedback learning (ReFL) has proven effective for aligning image generation with human preferences. However, its extension to video generation faces significant challenges. Existing video reward models rely on vision-language models designed for pixel-space inputs, confining ReFL optimization to near-complete denoising steps after computationally expensive VAE decoding. This pixel-space approach incurs substantial memory overhead and increased training time, and its late-stage optimization lacks early-stage supervision, refining only visual quality rather than fundamental motion dynamics and structural coherence. In this work, we show that pre-trained video generation models are naturally suited for reward modeling in the noisy latent space, as they are explicitly designed to process noisy latent representations at arbitrary timesteps and inherently preserve temporal information through their sequential modeling capabilities. Accordingly, we propose Process Reward Feedback Learning~(PRFL), a framework that conducts preference optimization entirely in latent space, enabling efficient gradient backpropagation throughout the full denoising chain without VAE decoding. Extensive experiments demonstrate that PRFL significantly improves alignment with human preferences, while achieving substantial reductions in memory consumption and training time compared to RGB ReFL.
Similar Papers
Improving Video Generation with Human Feedback
CV and Pattern Recognition
Makes videos look smoother and match your words.
Taming Camera-Controlled Video Generation with Verifiable Geometry Reward
CV and Pattern Recognition
Makes AI videos move cameras more accurately.
Goal-Driven Reward by Video Diffusion Models for Reinforcement Learning
Machine Learning (CS)
Teaches robots goals using movie clips.