arXiv:2411.08302cs.CLcs.AI2024-11EMNLP被引 10

让大模型每一步生成都获得精准反馈,提升语言细节表现。

RED: Unleashing Token-Level Rewards from Holistic Feedback via Reward Redistribution

  • 用现成奖励模型给每个词分配具体奖励,实现细粒度指导。
  • 在多个数据集上显著提升模型生成质量,效果优于传统方法。
  • 无需修改奖励模型或增加训练步骤,计算开销极低。

基于人类反馈的强化学习(RLHF)为对齐大语言模型与人类偏好提供了有效途径。现有方法通常使用序列到单值的奖励模型,在生成完成后给出单一、稀疏且延迟的奖励,难以捕捉单个词汇对最终结果的贡献。为此,我们提出RED方法,一种新颖的奖励重分配机制,利用现成奖励模型对每个生成词进行评估并分配具体奖励。该细粒度奖励使模型更深入理解语言细微差别,从而带来更精确的性能提升。本方法无需修改奖励模型或引入额外训练步骤,计算成本几乎可忽略。在多个数据集和任务上的实验结果表明,该方法显著优于现有基线。

原文摘要 · Abstract (English)

Reinforcement learning from human feedback (RLHF) offers a promising approach to aligning large language models (LLMs) with human preferences. Typically, a reward model is trained or supplied to act as a proxy for humans in evaluating generated responses during the reinforcement training phase. However, current reward models operate as sequence-to-one models, allocating a single, sparse, and delayed reward to an entire output sequence. This approach may overlook the significant contributions of individual tokens toward the desired outcome. To this end, we propose a more fine-grained, token-level guidance approach for RL training. Specifically, we introduce RED, a novel reward redistribition method that evaluates and assigns specific credit to each token using an off-the-shelf reward model. Utilizing these fine-grained rewards enhances the model's understanding of language nuances, leading to more precise performance improvements. Notably, our method does not require modifying the reward model or introducing additional training steps, thereby incurring minimal computational costs. Experimental results across diverse datasets and tasks demonstrate the superiority of our approach.

强化学习奖励模型语言生成

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