Unified Reward Model for Multimodal Understanding and Generation

Yibin Wang1,2, Yuhang Zang3, Hao Li1,2,4, Cheng Jin1, Jiaqi Wang2,3,
1Fudan University,
2Shanghai Innovation Institute,
3Shanghai AI Lab,
4Shanghai Academy of Artificial Intelligence for Science
Corresponding Author
data-overview

Overview of our UnifiedReward for multimodal understanding and generation alignment, including three steps: (1) Unified Reward Model Training, (2) Preference Data Construction, and (3) Generation/Understand Model Alignment.

Abstract

Recent advances in human preference alignment have significantly enhanced multimodal generation and understanding. A key approach is training reward models to guide preference optimization. However, existing models are often task-specific, limiting their adaptability across diverse visual applications. We also argue that jointly learning to assess multiple tasks may foster a synergistic effect, where improved image understanding enhances image generation assessment, and refined image evaluation benefits video quality assessment through better frame analysis. To this end, this paper proposes UnifiedReward, the first unified reward model for multimodal understanding and generation assessment, enabling both pairwise ranking and pointwise scoring, which can be employed for vision model preference alignment. Specifically, (1) we first develop UnifiedReward on our constructed large-scale human preference dataset, including both image and video generation/understanding tasks. (2) Then, it is utilized to automatically construct high-quality preference pair data based on the vision models, fine-gradually filtering their outputs through pair ranking and point sifting. (3) Finally, these data are used for their preference alignment through direct preference optimization (DPO). Our experimental results demonstrate that joint learning to assess diverse visual tasks can lead to substantial mutual benefits and we apply our pipeline to both image and video understanding/generation tasks, significantly improving the performance in each domain.

Compared with Current Reward Models

pipeline

Comparison of Our Reward Method with Recent Approaches.
UnifiedReward is capable of assessing both image and video understanding and generation. “Pair” and “Point” refer to “Pair Ranking” and “Point Scoring”, respectively.

Method Overview

pipeline

Method Overview. The pipeline of UnifiedReward consists of three key stages:
(1) Unified Reward Model Training: We train a unified reward model to evaluate both multimodal generation and understanding tasks using pointwise scoring and pairwise ranking strategy.
(2) Preference Data Construction: We use the trained reward model to construct high-quality preference data through three steps: (a) data generation from VLM/Diffusion, (b) pairwise ranking to divide the chosen and rejected outputs, and (c) pointwise filtering to refine the chosen and rejected samples.
(3) Generation/Understanding Model Alignment: The constructed preference data is then used to fine-tune VLM/Diffusion via Direct Preference Optimization, aligning their outputs with human preferences.

Our Constructed Unified Preference Dataset

pipeline

Visualization of Statistical Results.
This figure presents the distribution of our constructed unified preference dataset, along with the pairwise and pointwise distributions for each task.

Reward Model Quantitative Comparison

pipeline

DPO Quantitative Comparison

pipeline

Qualitative Comparison

pipeline
pipeline

BibTeX

@article{UnifiedReward,
      title={Unified Reward Model for Multimodal Understanding and Generation.},
      author={Wang, Yibin and Zang, Yuhang, and Li, Hao and Jin, Cheng and Wang Jiaqi},
      journal={arXiv preprint arXiv:2503.05236},
      year={2025}
    }