0

VisionReward: Fine-Grained Multi-Dimensional Human Preference Learning for Image and Video Generation

A reward model with multi-dimensional analysis aligns visual generation models with human preferences, surpassing existing methods in image and video scoring.

Year
2024
Venue
arXiv 2024
Authors
21
Hosting
Abstract onlyARXIV-DEFAULT

Cite

Notes

Only stored in your browser.

Attribution

Abstract & full text
arxiv.org/abs/2412.21059ARXIV-DEFAULT
TL;DR
Semantic Scholar
Attribution policy →

Abstract

We present a general strategy to aligning visual generation models -- both image and video generation -- with human preference. To start with, we build VisionReward -- a fine-grained and multi-dimensional reward model. We decompose human preferences in images and videos into multiple dimensions, each represented by a series of judgment questions, linearly weighted and summed to an interpretable and accurate score. To address the challenges of video quality assessment, we systematically analyze various dynamic features of videos, which helps VisionReward surpass VideoScore by 17.2% and achieve top performance for video preference prediction. Based on VisionReward, we develop a multi-objective preference learning algorithm that effectively addresses the issue of confounding factors within preference data. Our approach significantly outperforms existing image and video scoring methods on both machine metrics and human evaluation. All code and datasets are provided at https://github.com/THUDM/VisionReward.

Authors

21