提出一种新方法,让奖励模型更保守,防止人工智能作弊。
Learning a Pessimistic Reward Model in RLHF
- 用悲观奖励微调技术学习更稳健的奖励模型。
- 在无需正则化情况下,仍能获得高性能策略且避免奖励作弊。
- 适合追求高鲁棒性奖励模型的研究者或工业应用。
本文提出一种名为PET的新方法,用于在离线强化学习与人类反馈(RLHF)中学习对抗奖励作弊的悲观奖励模型。传统奖励建模方法依赖不完美的奖励模型,并通过KL正则化缓解奖励作弊问题,但该方法仍易受攻击,且会排除与数据分布差异大的策略。相比之下,我们证明,在使用PET微调的悲观奖励模型上优化策略时,无需任何正则化即可防止奖励作弊。我们在标准的TL;DR摘要数据集上测试,发现可直接训练出高质量策略,其与数据分布的KL散度较大,但实际表现优异。结果表明,构建悲观奖励模型以抵御奖励作弊是可行的,智能体可贪婪搜索高悲观奖励策略而不受奖励作弊影响。
原文摘要 · Abstract (English)
This work proposes `PET', a novel pessimistic reward fine-tuning method, to learn a pessimistic reward model robust against reward hacking in offline reinforcement learning from human feedback (RLHF). Traditional reward modeling techniques in RLHF train an imperfect reward model, on which a KL regularization plays a pivotal role in mitigating reward hacking when optimizing a policy. Such an intuition-based method still suffers from reward hacking, and the policies with large KL divergence from the dataset distribution are excluded during learning. In contrast, we show that when optimizing a policy on a pessimistic reward model fine-tuned through PET, reward hacking can be prevented without relying on any regularization. We test our methods on the standard TL;DR summarization dataset. We find that one can learn a high-quality policy on our pessimistic reward without using any regularization. Such a policy has a high KL divergence from the dataset distribution while having high performance in practice. In summary, our work shows the feasibility of learning a pessimistic reward model against reward hacking. The agent can greedily search for the policy with a high pessimistic reward without suffering from reward hacking.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。