English

Unified Personalized Reward Model for Vision Generation

Computer Vision and Pattern Recognition 2026-02-11 v2

Abstract

Recent advancements in multimodal reward models (RMs) have significantly propelled the development of visual generation. Existing frameworks typically adopt Bradley-Terry-style preference modeling or leverage generative VLMs as judges, and subsequently optimize visual generation models via reinforcement learning. However, current RMs suffer from inherent limitations: they often follow a one-size-fits-all paradigm that assumes a monolithic preference distribution or relies on fixed evaluation rubrics. As a result, they are insensitive to content-specific visual cues, leading to systematic misalignment with subjective and context-dependent human preferences. To this end, inspired by human assessment, we propose UnifiedReward-Flex, a unified personalized reward model for vision generation that couples reward modeling with flexible and context-adaptive reasoning. Specifically, given a prompt and the generated visual content, it first interprets the semantic intent and grounds on visual evidence, then dynamically constructs a hierarchical assessment by instantiating fine-grained criteria under both predefined and self-generated high-level dimensions. Our training pipeline follows a two-stage process: (1) we first distill structured, high-quality reasoning traces from advanced closed-source VLMs to bootstrap SFT, equipping the model with flexible and context-adaptive reasoning behaviors; (2) we then perform direct preference optimization (DPO) on carefully curated preference pairs to further strengthen reasoning fidelity and discriminative alignment. To validate the effectiveness, we integrate UnifiedReward-Flex into the GRPO framework for image and video synthesis, and extensive results demonstrate its superiority.

Keywords

Cite

@article{arxiv.2602.02380,
  title  = {Unified Personalized Reward Model for Vision Generation},
  author = {Yibin Wang and Yuhang Zang and Feng Han and Jiazi Bu and Yujie Zhou and Cheng Jin and Jiaqi Wang},
  journal= {arXiv preprint arXiv:2602.02380},
  year   = {2026}
}

Comments

Website: https://codegoat24.github.io/UnifiedReward/flex

R2 v1 2026-07-01T09:32:23.162Z