arXiv:2506.16507cs.LG2025-06被引 17

用因果规则对抗奖励欺骗,让模型更懂真实质量而非表面特征。

Robust Reward Modeling via Causal Rubrics

  • 通过因果与中性增强对齐模型,强制区分真实质量与表面线索。
  • 在RewardBench上平均准确率提升5.4%,特定任务最高增益达13.2%。
  • 无需预先知道干扰因素,仅靠大模型提问即可生成有效增强数据。

奖励模型(RMs)是通过人类反馈对齐大语言模型的核心组件,但常因奖励欺骗而失效。它们易受响应长度、格式等表面或虚假属性影响,将训练数据中的相关性误认为真实因果驱动因素(如事实性、相关性)。标准训练目标难以解耦这些因素,导致奖励模型脆弱且策略错位。本文提出Crome(因果鲁棒奖励建模),基于显式因果模型的新框架。Crome在训练中引入两类合成增强:(1) 因果增强,即在特定因果属性上差异的样本对,以强化对每个因果属性的敏感性;(2) 中性增强,即主要在虚假属性上变化但标签相同的样本对,以强制对虚假属性的不变性。关键创新在于,这些增强仅通过查询一个权威大模型获取因果规则后,对答案进行干预生成,无需事先知晓虚假因素。实验证明,Crome显著优于标准基线,在RewardBench上平均准确率提升最高达5.4%,特定类别最高增益13.2%和7.2%。其鲁棒性还体现在Best-of-N推理中,随N增大持续提升,覆盖RewardBench(聊天、困难聊天、安全、推理)、WildGuardTest(安全导向)和GSM8k(推理专项)等多类基准。

原文摘要 · Abstract (English)

Reward models (RMs) are fundamental to aligning Large Language Models (LLMs) via human feedback, yet they often suffer from reward hacking. They tend to latch on to superficial or spurious attributes, such as response length or formatting, mistaking these cues learned from correlations in training data for the true causal drivers of quality (e.g., factuality, relevance). This occurs because standard training objectives struggle to disentangle these factors, leading to brittle RMs and misaligned policies. We introduce Crome (Causally Robust Reward Modeling), a novel framework grounded in an explicit causal model designed to mitigate reward hacking. Crome employs the following synthetic targeted augmentations during training: (1) Causal Augmentations, which are pairs that differ along specific causal attributes, to enforce sensitivity along each causal attribute individually, and (2) Neutral Augmentations, which are tie-label pairs varying primarily in spurious attributes, to enforce invariance along spurious attributes. Notably, our augmentations are produced without any knowledge of spurious factors, via answer interventions only along causal rubrics, that are identified by querying an oracle LLM. Empirically, Crome significantly outperforms standard baselines on RewardBench, improving average accuracy by up to 5.4% and achieving gains of up to 13.2% and 7.2% in specific categories. The robustness of Crome is further testified by the consistent gains obtained in a Best-of-N inference setting across increasing N, across various benchmarks, including the popular RewardBench (covering chat, chat-hard, safety, and reasoning tasks), the safety-focused WildGuardTest, and the reasoning-specific GSM8k.

奖励建模因果推理大模型对齐鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。