用句子级奖励模型提升大模型对齐效果,更精准捕捉人类偏好。
Sentence-level Reward Model can Generalize Better for Aligning LLM from Human Preference
- 将回复按句子分割,对每句打分,实现中间粒度奖励建模。
- 在RewardBench上比响应级模型高2.7%,AlpacaEval上超越所有基线。
- 适合需要精细偏好对齐的生成任务,如对话系统优化。
从人类偏好数据集学习奖励模型,并通过强化学习优化语言模型,已成为对齐大模型与人类偏好的核心范式。奖励模型性能直接影响对齐效果。以往方法多为粗粒度,需生成完整回复才能获得奖励,导致奖励稀疏,不利于下游强化学习。虽有研究尝试构建词元级奖励模型,但缺乏显式语义信息,难以准确分配每个词元的贡献。本文提出对每句话赋分,引入中间粒度奖励模型。通过将完整回复切分为句子,并对每句起止位置的奖励输出进行微分操作,有效建模句子级奖励。同时设计新型注意力机制,将所有句子得分聚合为响应级分数,支持使用Bradley-Terry模型训练。在常见基准上,本方法在RewardBench上比响应级模型提升2.7%,在AlpacaEval上全面超越所有基线。
原文摘要 · Abstract (English)
Learning reward models from human preference datasets and subsequently optimizing language models via reinforcement learning has emerged as a fundamental paradigm for aligning LLMs with human preferences. The performance of the reward model plays a crucial role in the effectiveness of alignment. Previous reward models operate at a coarse-grained level, requiring the generation of a complete response to obtain a reward value. The sparse reward may present challenges for downstream reinforcement learning. While recent efforts have attempted to learn token-level reward models, the lack of explicit semantic information makes it difficult to model the credit of every individual token. In this paper, we propose assigning scores to every sentence, introducing an intermediate-grained reward model. By segmenting the complete response into sentences and applying differential operations to reward output at the start and end positions of each sentence, we can effectively model the rewards of sentences. Moreover, a novel attention mechanism is introduced to aggregate the scores of all sentences into a response-level score, which allows it to be trained using the Bradley-Terry model. On common benchmarks, our method outperforms the response-level reward model by 2.7% on RewardBench (for reward modeling evaluation) and surpasses all baselines on AlpacaEval (for alignment evaluation).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。