arXiv:2605.09214cs.LGcs.AI2026-05

首次实现前向KL正则化离线上下文老虎机的快速收敛率

Fast Rates for Offline Contextual Bandits with Forward-KL Regularization under Single-Policy Concentrability

  • 基于新颖的悲观原则分析框架,统一处理表格与函数逼近场景
  • 在单策略集中条件下,达成首个$ ilde{O}(ε^{-1})$上界,优于传统$ε^{-2}$
  • 证明率最优下界,揭示低正则化时恢复无正则化慢速率现象

KL正则化广泛应用于强化学习算法中,分为反向和前向KL两种形式。近期研究已证明反向KL正则化可实现$ε^{-1}$型快速率,而前向KL正则化现有分析或不适用,或仅得$ ilde{O}(ε^{-2})$慢速率。本文首次通过简化分析,对前向KL正则化的离线上下文老虎机问题给出$ ilde{O}(ε^{-1})$上界,涵盖表格与一般函数逼近设置,均在单策略集中性假设下成立。我们提出基于凸分析的统一框架,创新性地运用悲观原则,完全规避了以往依赖中值定理的证明路径,该方法本身具有独立意义。此外,我们构建了率最优下界,验证了上界的紧致性;同时表明,在低正则化区域,前向KL正则化样本复杂度会退化为未正则化情形的慢速率,与反向KL正则化行为一致。

原文摘要 · Abstract (English)

\emph{Kullback-Leibler} (KL) regularization is ubiquitous in reinforcement learning algorithms in the form of \emph{reverse} or \emph{forward} KL. Recent studies have demonstrated $ε^{-1}$-type fast rates for decision making under reverse KL regularization, in contrast to the standard $ε^{-2}$-type sample complexity. However, for forward-KL-regularized objectives, existing statistical analyses are either not applicable or result in $\tilde{O}(ε^{-2})$ slow rates. We take the first step towards addressing this problem via a streamlined analysis of forward-KL-regularized offline CBs. We give the first $\tilde{O}(ε^{-1})$ upper bounds in tabular and general function approximation settings, both under notions of \emph{single-policy concentrability}. In particular, our convex-analytical pipeline unifies these settings by exploiting the pessimism principle in a novel way and completely bypasses the proof routines in previous works based on the mean value theorem, which might be of independent interest. Moreover, we provide rate-optimal lower bounds, manifesting the tightness of our upper bounds in terms of statistical rates. Our lower bounds also demonstrate that the forward-KL-regularized sample complexity recovers the unregularized slow rate in the low-regularization regime, similarly to the reverse-KL regularization.

强化学习在线决策统计学习正则化

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