arXiv:2501.12735cs.LG2025-01ICLR被引 32

让大模型在线探索新回答,提升指令遵循能力。

Online Preference Alignment for Language Models via Count-based Exploration

  • 用计数奖励鼓励模型探索未见的问答对
  • 在指令遵循任务上性能显著提升
  • 适合需要持续优化的交互式大模型

基于人类反馈的强化学习(RLHF)在对齐大语言模型与人类偏好方面展现出巨大潜力。现有方法依赖固定数据集进行偏好对齐,存在数据覆盖不足、泛化能力差的问题。因此,在线RLHF更受期待,可使模型通过迭代收集提示-响应对,在初始数据分布外持续探索。本文研究在线RLHF中的核心问题:如何有效探索。在线性奖励假设下,理论证明乐观奖励加置信度上限(UCB)项可实现高效策略。为此,我们将目标重构为带探索项的直接偏好优化,将UCB项转化为基于计数的探索奖励。进一步提出实用算法COPO,利用简单的硬币翻转计数模块估算已收集数据中提示-响应对的伪计数。COPO在迭代中平衡探索与偏好优化,扩大了探索空间和整体数据覆盖范围。在Zephyr和Llama-3模型上进行的在线RLHF实验表明,该方法在指令遵循和标准学术基准测试中均显著提升性能。

原文摘要 · Abstract (English)

Reinforcement Learning from Human Feedback (RLHF) has shown great potential in fine-tuning Large Language Models (LLMs) to align with human preferences. Existing methods perform preference alignment from a fixed dataset, which can be limited in data coverage, and the resulting reward model is hard to generalize in out-of-distribution responses. Thus, online RLHF is more desirable to empower the LLM to explore outside the support of the initial dataset by iteratively collecting the prompt-response pairs. In this paper, we study the fundamental problem in online RLHF, i.e. \emph{how to explore} for LLM. We give a theoretical motivation in linear reward assumption to show that an optimistic reward with an upper confidence bound (UCB) term leads to a provably efficient RLHF policy. Then, we reformulate our objective to direct preference optimization with an exploration term, where the UCB-term can be converted to a count-based exploration bonus. We further propose a practical algorithm, named \emph{Count-based Online Preference Optimization (COPO)}, which leverages a simple coin-flip counting module to estimate the pseudo-count of a prompt-response pair in previously collected data. COPO encourages LLMs to balance exploration and preference optimization in an iterative manner, which enlarges the exploration space and the entire data coverage of iterative LLM policies. We conduct online RLHF experiments on Zephyr and Llama-3 models. The results on instruction-following and standard academic benchmarks show that COPO significantly increases performance.

在线学习探索机制大模型对齐

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。