arXiv:2409.00162cs.CLcs.AI2024-09AAAI被引 23

用语言反馈替代打分,让大模型更懂人类意图。

Sequence to Sequence Reward Modeling: Improving RLHF by Language Feedback

  • 用序列生成方式建模语言反馈,无需额外标注
  • 在3个任务上平均胜率提升至76.9%,减少拒答和长文本偏差
  • 适用于2B与7B模型,对分布外提示仍有效

大语言模型(LLM)与人类意图对齐仍是关键挑战。强化学习从人类反馈(RLHF)通过人类偏好训练奖励模型(RM),并微调LLM以最大化RM反馈。尽管有效,但易陷入局部优化,导致RM无法准确反映人类偏好,使LLM产生意外泛化,偏离对齐目标。为此,我们提出一种新的序列到序列(seq2seq)奖励建模方法,核心思想是直接学习语言反馈而非标量反馈,无需额外标注。将奖励建模目标从二元最大似然估计(MLE)替换为序列MLE,实现更丰富、细粒度的语言反馈,且不增加模型或训练阶段。实验表明,该方法显著降低单轮安全对话中的拒答现象,缓解文本摘要中的长响应偏差。在2B与7B LLM上跨3个NLP任务的分析显示,平均胜率达到76.9%。进一步验证其在分布外提示下仍能提升RLHF性能。

原文摘要 · Abstract (English)

Aligning the behavior of Large language models (LLMs) with human intentions and values remains a critical challenge. Reinforcement learning from human feedback (RLHF) aligns LLMs by training a reward model (RM) on human preferences and fine-tuning the LLMs to maximize RM feedback. Despite its effectiveness and popularity, RLHF is prone to biased local optimization. It means RM fails to provide feedback that accurately aligns with human preference, causing LLMs to explore unexpected generalizations, and failing to achieve alignment objectives. To mitigate this issue, we propose a novel \textit{sequence-to-sequence (seq2seq) reward modeling} method. Its key insight is that learning from language feedback rather than scalar feedback improves RLHF without additional annotations. We replaced the reward modeling target from binary maximum likelihood estimation (MLE) with sequence MLE. This method enables richer and fine-grained language feedback without additional annotations, models, or training stages. Our experiments demonstrated its effectiveness, specifically, reducing the refusal-to-response paradigm in single-turn safety dialogues and the long-response bias in text summarization tasks. We provide further analysis that seq2seq RM improves RLHF performance across 2B and 7B LLMs on 3 NLP tasks, achieving an average win rate of 76.9\%. We further show that seq2seq RM can still improve the performance of RLHF under out-of-distribution prompts.

RLHF奖励建模语言反馈对齐

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