arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.07533cs.AI
收藏

联合奖励建模:将思维链内化以实现高效的视觉奖励模型

Joint Reward Modeling: Internalizing Chain-of-Thought for Efficient Visual Reward Models

Harbin Institute of Technology, Shenzhen(哈尔滨工业大学(深圳));Kuaishou Technology(快手科技)
Yankai Yang, Yancheng Long, Hongyang Wei, Wei Chen, Tianke Zhang, Kaiyu Jiang, Haonan Fan, Changyi Liu, Jiankang Chen, Kaiyu Tang, Bin Wen, Fan Yang, Tingting G… 展开作者

Yankai Yang, Yancheng Long, Hongyang Wei, Wei Chen, Tianke Zhang, Kaiyu Jiang, Haonan Fan, Changyi Liu, Jiankang Chen, Kaiyu Tang, Bin Wen, Fan Yang, Tingting Gao, Han Li, Shuo Yang

AI总结:

提出联合奖励建模(JRM),通过共享视觉-语言骨干联合优化偏好学习和语言建模,将生成模型的语义推理能力内化到判别表示中,实现快速准确评估,在MMRB2和EditReward-Bench上取得最优结果。

摘要:

奖励模型对于从人类反馈中进行强化学习至关重要,因为它们决定了生成模型的对齐质量和可靠性。对于图像编辑等复杂任务,奖励模型需要捕捉全局语义一致性和超越局部相似性的隐式逻辑约束。现有的奖励建模方法存在明显局限性。判别式奖励模型与人类偏好对齐良好,但由于推理监督有限,难以处理复杂语义。生成式奖励模型提供更强的语义理解和推理能力,但在推理时成本高昂,且难以直接与人类偏好对齐。为此,我们提出联合奖励建模(JRM),在共享的视觉-语言骨干上联合优化偏好学习和语言建模。该方法将生成模型的语义和推理能力内化到高效的判别表示中,从而实现快速准确的评估。JRM在MMRB2和EditReward-Bench上取得了最先进的结果,并显著提升了下游在线强化学习的稳定性和性能。这些结果表明,联合训练有效地弥合了奖励建模中效率与语义理解之间的差距。

英文摘要:

Reward models are critical for reinforcement learning from human feedback, as they determine the alignment quality and reliability of generative models. For complex tasks such as image editing, reward models are required to capture global semantic consistency and implicit logical constraints beyond local similarity. Existing reward modeling approaches have clear limitations. Discriminative reward models align well with human preferences but struggle with complex semantics due to limited reasoning supervision. Generative reward models offer stronger semantic understanding and reasoning, but they are costly at inference time and difficult to align directly with human preferences. To this end, we propose Joint Reward Modeling (JRM), which jointly optimizes preference learning and language modeling on a shared vision-language backbone. This approach internalizes the semantic and reasoning capabilities of generative models into efficient discriminative representations, enabling fast and accurate evaluation. JRM achieves state-of-the-art results on MMRB2 and EditReward-Bench, and significantly improves stability and performance in downstream online reinforcement learning. These results show that joint training effectively bridges efficiency and semantic understanding in reward modeling.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑