新框架统一生成式奖励建模与策略优化,提升大模型对齐效果。
A Unified Pairwise Framework for RLHF: Bridging Generative Reward Modeling and Policy Optimization
- 用生成式建模统一处理奖励与策略优化,避免传统方法的偏差。
- 在多个基准上超越传统方法,显著提升模型对齐表现。
- 适合追求高精度对齐的大模型训练团队使用。
基于人类反馈的强化学习(RLHF)已成为大语言模型后训练阶段对齐人类偏好的重要范式,通常包含两阶段:先在人类偏好数据上训练奖励模型,再用强化学习算法优化语言模型。然而现有方法存在两大局限:其一,依赖布拉德利-特里模型对单个响应进行成对比较并赋予标量奖励,但不同上下文下提示-响应对的固有差异对奖励模型的校准能力提出严峻挑战;其二,奖励模型常从生成式基础模型(如预训练或监督微调模型)初始化,而奖励任务本质为判别任务,导致任务不匹配。本文提出 Pairwise-RL 框架,通过结合生成式奖励建模与成对近端策略优化(PPO)算法,将奖励模型训练与其在强化学习中的应用统一于一致的成对范式中,利用生成式建模技术增强奖励模型性能与得分校准能力。实验表明,Pairwise-RL 在内部评估数据集和标准公开基准上均优于传统 RLHF 框架,证明其在提升对齐效果与模型行为方面的有效性。
原文摘要 · Abstract (English)
Reinforcement Learning from Human Feedback (RLHF) has emerged as a important paradigm for aligning large language models (LLMs) with human preferences during post-training. This framework typically involves two stages: first, training a reward model on human preference data, followed by optimizing the language model using reinforcement learning algorithms. However, current RLHF approaches may constrained by two limitations. First, existing RLHF frameworks often rely on Bradley-Terry models to assign scalar rewards based on pairwise comparisons of individual responses. However, this approach imposes significant challenges on reward model (RM), as the inherent variability in prompt-response pairs across different contexts demands robust calibration capabilities from the RM. Second, reward models are typically initialized from generative foundation models, such as pre-trained or supervised fine-tuned models, despite the fact that reward models perform discriminative tasks, creating a mismatch. This paper introduces Pairwise-RL, a RLHF framework that addresses these challenges through a combination of generative reward modeling and a pairwise proximal policy optimization (PPO) algorithm. Pairwise-RL unifies reward model training and its application during reinforcement learning within a consistent pairwise paradigm, leveraging generative modeling techniques to enhance reward model performance and score calibration. Experimental evaluations demonstrate that Pairwise-RL outperforms traditional RLHF frameworks across both internal evaluation datasets and standard public benchmarks, underscoring its effectiveness in improving alignment and model behavior.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。