用自生成的逐词奖励,让大模型更精准地学习人类偏好。
T-REG: Preference Optimization with Token-Level Reward Regularization
- 让大模型通过对比提示自行生成逐词奖励,替代外部标注。
- 在Alpaca Eval 2和Arena-Hard上分别提升3.8%和4.4%。
- 适合追求高对齐效果且无标注资源的研究者。
基于人类反馈的强化学习(RLHF)对齐大语言模型与人类价值观至关重要。传统方法仅对完整响应给予单一稀疏奖励,难以定位贡献关键部分。现有逐词奖励方法依赖训练好的信用分配模型或AI标注员,存在质量风险。本文提出逐词奖励正则化(T-REG),结合序列级与逐词级奖励进行偏好优化。利用大模型的自修正能力,通过对比提示让模型自动生成逐词奖励,作为奖励正则化项,引导模型更合理地分配序列级奖励。实验表明,在Alpaca Eval 2和Arena-Hard基准上,该方法分别优于基线3.8%和4.4%。代码与模型将开源。
原文摘要 · Abstract (English)
Reinforcement learning from human feedback (RLHF) has been crucial in aligning large language models (LLMs) with human values. Traditionally, RLHF involves generating responses to a query and using a reward model to assign a reward to the entire response. However, this approach faces challenges due to its reliance on a single, sparse reward, which makes it challenging for the model to identify which parts of the sequence contribute most significantly to the final reward. Recent methods have attempted to address this limitation by introducing token-level rewards. However, these methods often rely on either a trained credit assignment model or AI annotators, raising concerns about the quality and reliability of the rewards. In this paper, we propose token-level reward regularization (T-REG), a novel approach that leverages both sequence-level and token-level rewards for preference optimization. Harnessing the self-refinement capabilities of LLMs, our method uses contrastive prompting to enable LLMs to self-generate token-level rewards. These self-generated rewards then act as reward regularization, guiding the model to more effectively distribute sequence-level rewards across tokens. This facilitates better token-level credit assignment and enhances alignment performance. Experiments on the instruction following benchmarks, including Alpaca Eval 2 and Arena-Hard, show that our method consistently outperforms baseline methods by up to 3.8% and 4.4%, respectively. We will release the code and models at https://github.com/wzhouad/T-REG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。