提出新算法,实现对大模型强化学习的对数级误差控制。
Logarithmic Regret for Online KL-Regularized Reinforcement Learning
- 基于乐观估计与KL正则化设计在线算法
- 理论证明误差随时间呈对数增长,优于传统方法
- 适合需要高效微调大模型的研究者参考
近期在人类反馈强化学习(RLHF)中的进展表明,KL-正则化在提升大语言模型(LLMs)微调效率方面起关键作用。尽管其在实践中表现优异,但其与标准强化学习的理论差异仍不明确。现有理论分析或退化为传统强化学习设置,或依赖强覆盖假设。本文提出一种基于乐观思想的KL-正则化在线上下文赌博机算法,并首次提供其后悔率的全新分析。通过巧妙利用KL正则化带来的良性优化结构及乐观奖励估计,算法实现了$ig(η ext{log}(N_{ ext{R}}T)ig)ig)d_{ ext{R}}$的对数后悔界,其中$η$为正则化参数,$N_{ ext{R}}$为奖励函数类的基数,$T$为轮次数,$d_{ ext{R}}$为奖励函数类的复杂度。进一步地,通过引入新的转移步骤分解方法,将该算法扩展至强化学习场景,同样获得类似对数后悔界。
原文摘要 · Abstract (English)
Recent advances in Reinforcement Learning from Human Feedback (RLHF) have shown that KL-regularization plays a pivotal role in improving the efficiency of RL fine-tuning for large language models (LLMs). Despite its empirical advantage, the theoretical difference between KL-regularized RL and standard RL remains largely under-explored. While there is a recent line of work on the theoretical analysis of KL-regularized objective in decision making \citep{xiong2024iterative, xie2024exploratory,zhao2024sharp}, these analyses either reduce to the traditional RL setting or rely on strong coverage assumptions. In this paper, we propose an optimism-based KL-regularized online contextual bandit algorithm, and provide a novel analysis of its regret. By carefully leveraging the benign optimization landscape induced by the KL-regularization and the optimistic reward estimation, our algorithm achieves an $\mathcal{O}\big(η\log (N_{\mathcal R} T)\cdot d_{\mathcal R}\big)$ logarithmic regret bound, where $η, N_{\mathcal R},T,d_{\mathcal R}$ denote the KL-regularization parameter, the cardinality of the reward function class, number of rounds, and the complexity of the reward function class. Furthermore, we extend our algorithm and analysis to reinforcement learning by developing a novel decomposition over transition steps and also obtain a similar logarithmic regret bound.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。