提出新方法让大模型更准确理解人类反馈,避免奖励漏洞。
Approximated Variational Bayesian Inverse Reinforcement Learning for Large Language Model Alignment
- 用贝叶斯逆强化学习建模每条反馈的直接奖励和中间奖励
- 在多个任务上优于现有对齐方法,提升奖励建模与泛化能力
- 适合需要精准对齐人类偏好的大模型训练场景
大语言模型对齐对于生成有益且无害的内容至关重要。现有方法依赖基于偏好的人类反馈数据来学习奖励函数并使模型与反馈对齐,但这些方法仅关注选择与拒绝示范之间的奖励差异,而非直接建模每个示范的真实奖励。此外,它们假设奖励仅在句子末尾获得,忽略了中间奖励的建模。这导致反馈数据中的训练信号利用不足,限制了奖励表示与泛化能力,可能引发奖励黑客问题。本文将大模型对齐建模为贝叶斯逆强化学习(BIRL)问题,提出一种新型训练目标——近似变分对齐(AVA),通过近似变分奖励模仿学习(AVRIL)实现对齐。BIRL框架支持中间奖励建模与单个示范的直接奖励建模,提升了反馈数据中训练信号的利用率。实验表明,AVA在奖励建模、强化学习微调和直接优化方面均优于现有方法。
原文摘要 · Abstract (English)
The alignment of large language models (LLMs) is crucial for generating helpful and harmless content. Existing approaches leverage preference-based human feedback data to learn the reward function and align the LLM with the feedback data. However, these approaches focus on modeling the reward difference between the chosen and rejected demonstrations, rather than directly modeling the true reward from each demonstration. Moreover, these approaches assume that the reward is only obtained at the end of the sentence, which overlooks the modeling of intermediate rewards. These issues lead to insufficient use of training signals in the feedback data, limiting the representation and generalization ability of the reward and potentially resulting in reward hacking. In this paper, we formulate LLM alignment as a Bayesian Inverse Reinforcement Learning (BIRL) problem and propose a novel training objective, Approximated Variational Alignment (AVA), to perform LLM alignment through Approximated Variational Reward Imitation Learning (AVRIL). The BIRL formulation facilitates intermediate reward modeling and direct reward modeling on each single demonstration, which enhances the utilization of training signals in the feedback data. Experiments show that AVA outperforms existing LLM alignment approaches in reward modeling, RL fine-tuning, and direct optimization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。