arXiv:2409.17401cs.LGstat.ML2024-09ICLR被引 14

无需奖励模型,直接优化策略,让大模型更高效地学习人类偏好。

Zeroth-Order Policy Gradient for Reinforcement Learning from Human Feedback without Reward Inference

  • 基于人类偏好估计局部价值差,用零阶梯度近似策略梯度。
  • 在随机环境中,算法收敛速度为多项式级别,性能优于DPO和PPO。
  • 适用于非确定性任务和复杂偏好模型,适合追求简洁高效的RLHF研究者。

从人类反馈中进行强化学习(RLHF)的关键步骤是奖励推断(从人类偏好中学习奖励模型)。实践中,该流程面临分布偏移、奖励模型过拟合及问题设定错误等根本挑战。替代方案是不进行奖励推断的直接策略优化,如直接偏好优化(DPO),其流程更简单且在大语言模型应用中表现良好。然而,DPO依赖最优策略与奖励函数之间的闭式表达,仅适用于赌博设置或确定性马尔可夫决策过程(MDP)。本文提出两种无需奖励推断的通用强化学习算法,适用于非带状任务和一般偏好模型,超越了布拉德利-特里模型。核心思想是从人类偏好中估计局部值函数差异,并使用零阶梯度近似器逼近策略梯度。我们证明了算法在策略梯度迭代次数、轨迹样本数及每轮偏好查询数上的多项式收敛率。数值实验在随机环境中验证了所提算法的有效性,性能优于主流的RLHF基线如DPO和PPO。结果表明,无需奖励推断即可高效求解通用的强化学习人类反馈问题。

原文摘要 · Abstract (English)

Reward inference (learning a reward model from human preferences) is a critical intermediate step in the Reinforcement Learning from Human Feedback (RLHF) pipeline for fine-tuning Large Language Models (LLMs). In practice, RLHF faces fundamental challenges such as distribution shift, reward model overfitting, and problem misspecification. An alternative approach is direct policy optimization without reward inference, such as Direct Preference Optimization (DPO), which provides a much simpler pipeline and has shown empirical success in LLM applications. However, DPO utilizes the closed-form expression between the optimal policy and the reward function, which is only suitable under the bandit setting or deterministic MDPs. This paper develops two RLHF algorithms without reward inference for general RL problems beyond bandits and deterministic MDPs, and general preference models beyond the Bradley-Terry model. The key idea is to estimate the local value function difference from human preferences and then approximate the policy gradient with a zeroth-order gradient approximator. For both algorithms, we establish polynomial convergence rates in terms of the number of policy gradient iterations, the number of trajectory samples, and human preference queries per iteration. Numerical experiments in stochastic environments validate the performance of our proposed algorithms, outperforming popular RLHF baselines such as DPO and PPO. Our paper shows there exist provably efficient methods to solve general RLHF problems without reward inference.

强化学习人类反馈策略优化大模型

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