Video Diffusion Alignment via Reward Gradient

Mihir Prabhudesai*         Russell Mendonca*         Zheyang Qin*         Katerina Fragkiadaki         Deepak Pathak
Carnegie Mellon University

Abstract

We have made significant progress towards building foundational video diffusion models. As these models are trained using large-scale unsupervised data, it has become crucial to adapt these models to specific downstream tasks, such as video-text alignment or ethical video generation. Adapting these models via supervised fine-tuning requires collecting target datasets of videos, which is challenging and tedious. In this work, we instead utilize pre-trained reward models that are learned via preferences on top of powerful discriminative models. These models contain dense gradient information with respect to generated RGB pixels, which is critical to be able to learn efficiently in complex search spaces, such as videos. We show that our approach can enable alignment of video diffusion for aesthetic generations, similarity between text context and video, as well long horizon video generations that are 3X longer than the training sequence length. We show our approach can learn much more efficiently in terms of reward queries and compute than previous gradient-free approaches for video generation.

Aesthetic and HPS Reward

PickScore Reward

HPS Reward

Object Removal Reward

Removing books with the YOLOS object detection model.