Near-Optimal Regret for KL-Regularized Multi-Armed Bandits
Kaixuan Ji, Qingyue Zhao, Heyang Zhao, Qiwei Di, Quanquan Gu
摘要
Recent studies have shown that reinforcement learning with KL-regularized objectives can enjoy faster rates of convergence or logarithmic regret, in contrast to the classical -type regret in the unregularized setting. However, the statistical efficiency of online learning with respect to KL-regularized objectives remains far from completely characterized, even when specialized to multi-armed bandits (MABs). We address this problem for MABs via a sharp analysis of KL-UCB (Zhao et al., 2025b) using a novel peeling argument, which yields a KL-regularized regret upper bound: the first high-probability regret bound with linear dependence on . Here, is the time horizon, is the number of arms, is the regularization intensity, and hides all logarithmic factors except those involving . The near-tightness of our analysis is certified by the first non-constant lower bound , which follows from subtle hard-instance constructions and a tailored decomposition of the Bayes prior. Moreover, in the low-regularization regime (i.e., large ), we show that the KL-regularized regret for MABs is -independent and scales as . Overall, our results provide a thorough understanding of KL-regularized MABs across all regimes of and yield nearly optimal bounds in terms of , , and .
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
它引用的顶会 Paper16
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida 等NeurIPS 2022 · 被引用 24,707 次
- Direct Preference Optimization: Your Language Model is Secretly a Reward ModelRafael Rafailov, Archit Sharma, Eric Mitchell, Christopher D. Manning 等NeurIPS 2023 · 被引用 10,924 次
- Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-constraintWei Xiong, Hanze Dong, Chenlu Ye, Ziqi Wang 等ICML 2024 · 被引用 346 次
- Provably Efficient Exploration in Policy OptimizationQi Cai, Zhuoran Yang, Chi Jin, Zhaoran WangICML 2020 · 被引用 304 次
- Leverage the Average: an Analysis of KL Regularization in Reinforcement LearningNino Vieillard, Tadashi Kozuno, Bruno Scherrer, Olivier Pietquin 等NeurIPS 2020 · 被引用 106 次
相关 Paper
- Logarithmic Regret for Online KL-Regularized Reinforcement LearningHeyang Zhao, Chenlu Ye, Wei Xiong, Quanquan Gu 等ICML 2025
- Sharp Analysis for KL-Regularized Contextual Bandits and RLHFHeyang Zhao, Chenlu Ye, Quanquan Gu, Tong ZhangNeurIPS 2025 · 被引用 33 次
- Towards a Sharp Analysis of Offline Policy Learning for -Divergence-Regularized Contextual BanditsQingyue Zhao, Kaixuan Ji, Heyang Zhao, Tong Zhang 等ICLR 2026 · 被引用 9 次
- Precise Asymptotics and Refined Regret of Variance-Aware UCBYingying Fan, Yuxuan Han, Jinchi Lv, Xiaocong Xu 等NeurIPS 2025 · 被引用 5 次
- IMED-RL: Regret optimal learning of ergodic Markov decision processesFabien Pesquerel, Odalric-Ambrym MaillardNeurIPS 2022 · 被引用 12 次
