动态调整奖励模型的偏好阈值,提升主观任务中的评分准确性。
Act-Adaptive Margin: Dynamically Calibrating Reward Models for Subjective Ambiguity
- 根据模型内部知识动态调节偏好差距,无需人工标注。
- 在角色扮演任务中提升奖励模型性能4.85%。
- 适合需要精准主观评价的对齐训练场景。
当前强化学习多聚焦于数学、编程等可验证领域,但在角色扮演等主观任务中,基于布拉德利-特瑞模型的奖励建模因偏好模糊而进展受限。本文提出AAM(Act-Adaptive Margin),通过利用模型内部参数知识动态校准偏好边际,有效生成上下文相关的偏好差距,无需额外人工标注。该方法显著提升了奖励模型对主观奖励的建模能力,实现了生成理解与偏好评分的更好融合。在RewardBench、JudgeBench及挑战性角色扮演任务上的实验表明,AAM使布拉德利-特瑞奖励模型在通用任务中提升2.95%,在主观角色扮演任务中提升4.85%。此外,使用AAM增强的奖励模型进行下游对齐训练,可在CharacterEval和Charm上达到顶尖效果。代码与数据集见https://github.com/calubkk/AAM。
原文摘要 · Abstract (English)
Currently, most reinforcement learning tasks focus on domains like mathematics and programming, where verification is relatively straightforward. However, in subjective tasks such as role-playing, alignment techniques struggle to make progress, primarily because subjective reward modeling using the Bradley-Terry model faces significant challenges when dealing with ambiguous preferences. To improve reward modeling in subjective tasks, this paper proposes AAM (\textbf{\underline{A}}ct-\textbf{\underline{A}}daptive \textbf{\underline{M}}argin), which enhances reward modeling by dynamically calibrating preference margins using the model's internal parameter knowledge. We design two versions of AAM that efficiently generate contextually-appropriate preference gaps without additional human annotation. This approach fundamentally improves how reward models handle subjective rewards by better integrating generative understanding with preference scoring. To validate AAM's effectiveness in subjective reward modeling, we conduct evaluations on RewardBench, JudgeBench, and challenging role-playing tasks. Results show that AAM significantly improves subjective reward modeling performance, enhancing Bradley-Terry reward models by 2.95\% in general tasks and 4.85\% in subjective role-playing tasks. Furthermore, reward models trained with AAM can help downstream alignment tasks achieve better results. Our test results show that applying rewards generated by AAM-Augmented RM to preference learning techniques (e.g., GRPO) achieves state-of-the-art results on CharacterEval and Charm. Code and dataset are available at https://github.com/calubkk/AAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。