arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.14760cs.LGcs.CL
收藏

面向分布外偏好学习的奖励建模泛化

Generalizing Reward Modeling for Out-of-Distribution Preference Learning

SI-TECH Information Technology(SI-TECH信息技术公司)

Chen Jia

AI总结:

本文提出一种基于元学习和双层优化的通用奖励建模方法,用于在有限偏好反馈下实现分布外偏好学习,并在多个文本生成任务和留出领域上验证了其优于强基线的性能。

摘要:

基于大语言模型(LLMs)的偏好学习(PL)旨在使大语言模型的生成内容与人类偏好对齐。以往关于基于人类反馈的强化学习(RLHF)的工作已在分布内偏好学习中展现出有前景的结果。然而,由于获取人类反馈较为困难,为每一个遇到的分布单独训练奖励模型具有很大的挑战性。因此,分布外(OOD)偏好学习对于在有限偏好反馈下增强大语言模型的泛化能力具有实际价值。本工作通过元学习方法优化一个通用奖励模型来解决分布外偏好学习问题。在元训练阶段,利用双层优化算法学习一个能够引导策略学习、使其在不同分布上与人类偏好对齐的奖励模型。当遇到测试分布时,元测试过程使用学习到的奖励模型进行正则化策略优化以实现偏好学习。我们在合理假设下从理论上证明了该双层优化算法的收敛速率。此外,我们在两个文本生成任务上跨20个留出领域进行了实验,并在多种评估指标上优于多个强基线。

英文摘要:

Preference learning (PL) with large language models (LLMs) aims to align the LLMs' generations with human preferences. Previous work on reinforcement learning from human feedback (RLHF) has demonstrated promising results in in-distribution PL. However, due to the difficulty of obtaining human feedback, discretely training reward models for every encountered distribution is challenging. Thus, out-of-distribution (OOD) PL is practically useful for enhancing the generalization ability of LLMs with limited preference feedback. This work addresses OOD PL by optimizing a general reward model through a meta-learning approach. During meta-training, a bilevel optimization algorithm is utilized to learn a reward model capable of guiding policy learning to align with human preferences across various distributions. When encountering a test distribution, the meta-test procedure conducts regularized policy optimization using the learned reward model for PL. We theoretically demonstrate the convergence rate of the bilevel optimization algorithm under reasonable assumptions. Additionally, we conduct experiments on two text generation tasks across 20 held-out domains and outperform a variety of strong baselines across various evaluation metrics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑