用演示数据直接学奖励模型,无需人类偏好标注。
From Demonstrations to Rewards: Alignment Without Explicit Human Preferences
- 基于逆强化学习,从演示数据直接推导奖励模型。
- 在多个基准上性能媲美依赖偏好数据的顶尖方法。
- 适合只有演示数据、缺乏偏好标注的场景使用。
大模型对齐人类偏好的挑战之一在于数据需求和现有方法的技术复杂性。主流方法如RLHF需多步流程,分别依赖演示数据和偏好数据。传统上,人类偏好通过奖励模型建模,指导强化学习阶段以生成符合偏好的策略。本文提出新视角:基于逆强化学习,仍以奖励最大化为目标,但直接从演示数据学习奖励模型,无需依赖偏好数据。该方法在仅有演示数据时依然有效,突破了当前RLHF的局限,揭示了演示数据的更大潜力。我们在HuggingFace Open LLM Leaderboard、MT-Bench及公开奖励基准上进行广泛评估,结果表明,该方法在仅使用演示数据的情况下,性能可媲美最先进的对齐方法。
原文摘要 · Abstract (English)
One of the challenges of aligning large models with human preferences lies in both the data requirements and the technical complexities of current approaches. Predominant methods, such as RLHF, involve multiple steps, each demanding distinct types of data, including demonstration data and preference data. In RLHF, human preferences are typically modeled through a reward model, which serves as a proxy to guide policy learning during the reinforcement learning stage, ultimately producing a policy aligned with human preferences. However, in this paper, we propose a fresh perspective on learning alignment based on inverse reinforcement learning principles, where the optimal policy is still derived from reward maximization. However, instead of relying on preference data, we directly learn the reward model from demonstration data. This new formulation offers the flexibility to be applied even when only demonstration data is available, a capability that current RLHF methods lack, and it also shows that demonstration data offers more utility than what conventional wisdom suggests. Our extensive evaluation, based on public reward benchmark, HuggingFace Open LLM Leaderboard and MT-Bench, demonstrates that our approach compares favorably to state-of-the-art methods that rely solely on demonstration data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。