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

RLHF中探索驱动的策略优化:高效数据利用的理论洞见

Exploration-Driven Policy Optimization in RLHF: Theoretical Insights on Efficient Data Utilization

University of Illinois Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);NVIDIA Research(英伟达研究院);Technion(以色列理工学院)

Yihan Du, Anna Winnicki, Gal Dalal, Shie Mannor, R. Srikant

AI总结:

本研究针对RLHF少量反馈即获优效的理论解释不足问题,提出基于策略优化的PO-RLHF算法,通过轨迹级椭圆势分析给出低查询复杂度性能界,为其高效数据利用提供理论支撑。

摘要:

基于人类反馈的强化学习(Reinforcement Learning from Human Feedback, RLHF)在仅依赖少量人类反馈的情况下已取得令人瞩目的实证成功,但目前对这一现象的理论解释仍十分有限。此外,尽管基于策略的算法近期实证表现亮眼,多数最新研究仍聚焦于基于价值的算法。本研究提出一种基于策略优化的RLHF算法(PO-RLHF),该算法以广泛使用的策略覆盖-策略梯度(Policy Cover-Policy Gradient, PC-PG)算法为基础,而PC-PG算法需假设奖励函数已知。PO-RLHF无需假设奖励函数已知,而是通过基于轨迹的比较反馈来推断奖励函数。我们给出了低查询复杂度下PO-RLHF的性能界,为“少量人类反馈即可让RLHF取得良好性能”的现象提供了解释思路。该研究的核心创新点是轨迹级椭圆势分析,该分析可在仅提供比较反馈(而非数值奖励观测)的情况下,对奖励估计误差进行界定。我们针对线性函数逼近和神经函数逼近两种场景,分别提出并分析了PG-RLHF和NN-PG-RLHF算法。

英文摘要:

Reinforcement Learning from Human Feedback (RLHF) has achieved impressive empirical successes while relying on a small amount of human feedback. However, there is limited theoretical justification for this phenomenon. Additionally, most recent studies focus on value-based algorithms despite the recent empirical successes of policy-based algorithms. In this work, we consider an RLHF algorithm based on policy optimization (PO-RLHF). The algorithm is based on the popular Policy Cover-Policy Gradient (PC-PG) algorithm, which assumes knowledge of the reward function. In PO-RLHF, knowledge of the reward function is not assumed, and the algorithm uses trajectory-based comparison feedback to infer the reward function. We provide performance bounds for PO-RLHF with low query complexity, which provides insight into why a small amount of human feedback may be sufficient to achieve good performance with RLHF. A key novelty is a trajectory-level elliptical potential analysis, which bounds the reward estimation error when comparison feedback (rather than numerical reward observation) is given. We provide and analyze algorithms PG-RLHF and NN-PG-RLHF for two settings: linear and neural function approximation, respectively.

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

↑