用博弈论方法给文本生成中的每个词分配奖励,让大模型更快更准地学习人类偏好。
SCAR: Shapley Credit Assignment for More Efficient RLHF
- 基于沙普利值原理,将整体奖励拆分到每个词或词段上
- 在情感控制、摘要生成等任务中收敛速度提升,最终得分更高
- 无需额外训练模型或人工标注,适合追求高效对齐的LLM研究者
强化学习从人类反馈(RLHF)是使大语言模型符合人类偏好的常用方法,但常因奖励信号稀疏而难以有效分配责任。传统方法仅对整个生成序列给出单一评分,无法判断具体哪些词或片段导致结果好坏。为此,我们提出沙普利信用分配奖励(SCAR),利用合作博弈论中的沙普利值,根据各词或文本片段的边际贡献,将总奖励合理分配至每个组成部分。该方法生成密集奖励信号,且无需训练辅助批判模型或在生成中间阶段依赖细粒度人工标注。理论上,SCAR保持原最优策略;实验上,在情感控制、文本摘要和指令微调等多种任务中,相比标准RLHF及基于注意力的密集奖励基线,SCAR收敛更快、最终奖励分数更高。结果表明,SCAR为RLHF中的信用分配提供了更高效且理论严谨的方法,显著提升大模型对齐效率。
原文摘要 · Abstract (English)
Reinforcement Learning from Human Feedback (RLHF) is a widely used technique for aligning Large Language Models (LLMs) with human preferences, yet it often suffers from sparse reward signals, making effective credit assignment challenging. In typical setups, the reward model provides a single scalar score for an entire generated sequence, offering little insight into which token or span-level decisions were responsible for the outcome. To address this, we propose Shapley Credit Assignment Rewards (SCAR), a novel method that leverages Shapley values in cooperative game theory. SCAR distributes the total sequence-level reward among constituent tokens or text spans based on their principled marginal contributions. This creates dense reward signals, crucially, without necessitating the training of auxiliary critique models or recourse to fine-grained human annotations at intermediate generation stages. Unlike prior dense reward methods, SCAR offers a game-theoretic foundation for fair credit attribution. Theoretically, we demonstrate that SCAR preserves the original optimal policy, and empirically, across diverse tasks including sentiment control, text summarization, and instruction tuning, we show that SCAR converges significantly faster and achieves higher final reward scores compared to standard RLHF and attention-based dense reward baselines. Our findings suggest that SCAR provides a more effective and theoretically sound method for credit assignment in RLHF, leading to more efficient alignment of LLMs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。