Omni-Reward

📌 核心贡献

提出首个全模态(文本、图像、视频、音频、3D)奖励建模框架,包括覆盖 5 种模态 9 类任务的 Omni-RewardBench 基准、317K 偏好对的 Omni-RewardData 数据集,以及支持自由格式偏好描述的判别式(BT)和生成式(R1)奖励模型。

📖 摘要

Reward models (RMs) play a critical role in aligning AI behaviors with human preferences, yet they face two fundamental challenges: (1) Modality Imbalance, where most RMs are mainly focused on text and image modalities, offering limited support for video, audio, and other modalities; and (2) Preference Rigidity, where training on fixed binary preference pairs fails to capture the complexity and diversity of personalized preferences. To address the above challenges, we propose Omni-Reward, a step toward generalist omni-modal reward modeling with support for free-form preferences, consisting of: (1) Evaluation: We introduce Omni-RewardBench, the first omni-modal RM benchmark with free-form preferences, covering nine tasks across five modalities including text, image, video, audio, and 3D; (2) Data: We construct Omni-RewardData, a multimodal preference dataset comprising 248K general preference pairs and 69K instruction-tuning pairs for training generalist omni-modal RMs; (3) Model: We propose Omni-RewardModel, which includes both discriminative and generative RMs, and achieves strong performance on Omni-RewardBench as well as other widely used reward modeling benchmarks.

🔍 关键信息

字段内容
机构Institute of Automation, Chinese Academy of Sciences / University of Chinese Academy of Sciences
发表2025-10-27
分类cs.CL, cs.AI, cs.CV
链接arXiv

📝 我的笔记

方法总览

Omni-RewardModel 架构概览:判别式(BT)和生成式(R1)两种变体

动机与问题定义

现有奖励模型存在两个根本限制:

  1. 模态不平衡(Modality Imbalance):绝大多数 RM 聚焦于文本和图像模态,对视频、音频、3D 等模态支持有限。
  2. 偏好刚性(Preference Rigidity):基于固定二元偏好对训练,无法捕捉真实场景中个性化偏好的复杂性和多样性。用户的偏好标准往往是隐式的、多维度的。

Omni-Reward 旨在构建一个统一的全模态奖励建模框架,同时支持自由格式偏好描述(free-form preferences),使奖励模型能根据用户指定的评估准则进行自适应评分。

任务形式化

每个样本表示为五元组 :

  • :输入 prompt
  • :两个候选回复
  • :自由格式评估准则(free-form criterion)
  • :偏好选择

两种评估设定:

  • w/o Ties:
  • w/ Ties:

Omni-RewardBench:全模态基准

Omni-RewardBench 中九种奖励建模任务的示意

覆盖 5 种模态、9 类任务,共 3,725 个高质量人工标注偏好对:

任务描述输入 → 输出数据来源
T2T文本生成指令 → 文本RMB, RPR (13 LLMs)
TI2T图像理解指令 + 图像 → 文本VL-Feedback, MIA-Bench (14 MLLMs)
TV2T视频理解指令 + 视频 → 文本VCGBench-Diverse (4 MLLMs)
TA2T音频理解指令 + 音频 → 文本OpenAQA (4 MLLMs)
T2I图像生成文本 → 图像Rapidata, HPDv2 (27 models)
T2V视频生成文本 → 视频GenAI-Bench (8 models)
T2A音频生成文本 → 音频Audio-alpaca, Tango
T23D3D 生成文本 → 3D3DRewardDB, mvdream-sd2.1
TI2I图像编辑源图 + 指令 → 编辑图GenAI-Bench

标注质量控制:

  • 3 名计算机博士生独立标注
  • 丢弃 23% 含无效准则的样本、15% 标注冲突的样本

Omni-RewardData:训练数据

通用偏好数据(248K 对):

任务子集数量
T2TSkywork-Reward-Preference50,000
T2TOmni-Skywork-Reward-Preference*16,376
T2TOmni-UltraFeedback*7,901
T2IHPDv250,000
T2IEvalMuse2,944
T2IOmni-HPDv2*8,959
T2IOmni-Open-Image-Preferences*8,105
TI2TRLAIF-V83,124
TI2TOmniAlign-V-DPO50,000
TI2TOmni-RLAIF-V*15,867
TI2TOmni-VLFeedback*12,311
T2VVideoDPO10,000
T2VVisionRewardDB-Video1,795

带 * 的是新构造的含自由格式偏好描述的子集,由 GPT-4o 生成准则 。

指令微调数据(69K 对): T2T (24K) + TI2T (28K) + T2I (17K),用于提升模型对自由格式偏好的理解和遵循能力。

模型架构

判别式模型(Omni-RewardModel-BT)

  • 基座模型: MiniCPM-o-2.6
  • 冻结组件: 视觉编码器、音频编码器
  • 可训练组件: 语言模型解码器 + 值头(value head)
  • 用户偏好注入方式: 将自由格式偏好描述 作为 system message 输入

Bradley-Terry 损失函数:

其中 为标量奖励函数, 为自由格式评估准则, 为被选回复, 为被拒回复。

生成式模型(Omni-RewardModel-R1)

  • 基座模型: Qwen2.5-VL-7B-Instruct
  • 生成过程: 给定 ,模型先输出 Chain-of-Thought 解释 ,再输出偏好预测
  • 训练框架: GRPO 强化学习
  • 训练数据: 仅 10K 样本(约占 Omni-RewardData 的 3%)
  • 奖励信号: 对比 与 ground-truth
  • 无需蒸馏: 不依赖大模型蒸馏

关键结果

不同模型在 Omni-RewardBench 九个任务上的性能对比

Omni-RewardBench 主实验(w/ Ties)

模型T2TTI2TTV2TTA2TT2IT2VT2AT23DTI2IOverall
GPT-4o78.1861.7469.3062.7559.3365.0344.5370.8669.8764.62
Claude-3.5-Sonnet76.7461.5567.04-61.6964.27-68.5465.9466.54
Gemma-3-27B-it77.2261.1767.04-59.1461.44-63.9165.9465.12
Qwen2.5-VL-32B74.8260.2363.88-60.5162.38-62.5869.4364.83
UnifiedReward1.559.4754.1769.30-58.3569.57-61.5945.4159.69
Omni-RewardModel-R171.2256.0663.88-61.6958.22-63.9146.2960.18
Omni-RewardModel-BT75.3060.2368.8570.5958.3564.0863.9967.8858.9565.36

Omni-RewardModel-BT 是唯一支持全部 5 种模态 9 类任务的开源模型,Overall 65.36% 超过大部分开源模型,接近 Claude-3.5-Sonnet (66.54%)。

Omni-RewardBench(w/o Ties)

模型Overall
Claude-3.5-Sonnet76.66
Gemma-3-27B-it75.27
GPT-4o-mini74.42
Qwen2.5-VL-72B74.02
Omni-RewardModel-BT73.68
Omni-RewardModel-R169.71
UnifiedReward1.568.79

VL-RewardBench 结果

模型GeneralHallucinationReasoningOverall AccMacro Acc
Skywork-VL-Reward66.080.061.073.169.0
UnifiedReward60.678.460.566.166.5
IXC-2.5-Reward84.762.562.965.870.0
GPT-4o49.167.670.565.862.4
Omni-RewardModel-R171.990.259.069.673.7
Omni-RewardModel-BT81.594.260.476.378.7

Omni-RewardModel-BT 在 VL-RewardBench 上 Macro Acc 达到 78.7%,显著超越所有基线(包括 IXC-2.5-Reward 70.0%),尤其在 Hallucination 检测(94.2%)上表现突出。

Multimodal RewardBench 结果

模型OverallGeneralKnowledgeReasoningSafetyVQACorrectnessMath
Claude 3.5 Sonnet72.062.667.873.968.665.176.885.6
Gemini 1.5 Pro72.063.567.766.368.955.594.587.2
GPT-4o71.562.669.072.067.662.174.887.2
Omni-RewardModel-BT70.571.358.466.771.048.579.385.1

BT 模型在 Multimodal RewardBench 上 70.5%,接近顶级闭源模型水平,且在 General (71.3%) 和 Safety (71.0%) 上超越 GPT-4o。

Ablation 分析

训练数据组合的影响(w/ Ties):

训练数据T2TTI2TT2IT2VT23DOverall
MiniCPM-o-2.6 基线61.3951.8947.3539.7037.0946.67
仅 T2T 数据74.3054.7345.3843.8649.6757.13
仅 TI2T 数据74.5459.6241.4548.7751.0058.84
仅 T2I & T2V 数据52.2845.8358.9364.8467.5557.50
全部数据75.3060.2358.3564.0867.8865.36
仅偏好数据(无 SFT)54.9249.8059.1461.0664.9058.67

关键发现:

  • 混合全部模态数据训练 (65.36%) 显著优于任何单模态子集 (~57-59%)
  • 指令微调数据至关重要:移除 69K SFT 数据后性能从 65.36% 降至 58.67%(下降 6.69%)
  • 理解类任务(T2T, TI2T)的训练数据对生成类任务有正向迁移
  • 跨模态任务间的 Pearson 相关系数:理解类任务内部 0.8-0.9,生成类任务内部 0.7-0.8

总结评价

优势:

  1. 首个覆盖 5 种模态 9 类任务的统一 RM 框架,填补了 audio / 3D 模态的奖励建模空白
  2. 自由格式偏好描述使 RM 可适应个性化评估需求,突破了传统二元偏好的局限
  3. BT 模型在多个公开基准上接近或超越闭源模型,R1 模型用仅 3% 数据就达到了不错的性能
  4. Omni-RewardBench 设计严谨,3,725 个样本经过严格质控

局限:

  1. BT 模型基于 MiniCPM-o-2.6,模型规模和能力受基座限制
  2. 音频和 3D 模态的训练数据量远少于文本和图像,性能差距仍然显著
  3. R1 模型不支持 audio 模态(受限于 Qwen2.5-VL 的模态支持)
  4. 不同任务间性能方差高达 28.37%,全模态均衡性仍有提升空间

🔗 相关论文

基于/改进自: UnifiedReward

同方向: UnifiedReward-Flex, UnifiedReward-Think, VL-RewardBench, VisionReward, VideoScore, IXC-2.5-Reward, Skywork-VL-Reward