Lune

ICML2026Top-tier venue

Near-Optimal Regret for KL-Regularized Multi-Armed Bandits

Kaixuan Ji, Qingyue Zhao, Heyang Zhao, Qiwei Di, Quanquan Gu

2026Year
3Citations

Abstract

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 T\sqrt{T}-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 O~(ηKlog⁡2T)\tilde{O}(\eta K\log^2T) KL-regularized regret upper bound: the first high-probability regret bound with linear dependence on KK. Here, TT is the time horizon, KK is the number of arms, η−1\eta^{-1} is the regularization intensity, and O~\tilde{O} hides all logarithmic factors except those involving log⁡T\log T. The near-tightness of our analysis is certified by the first non-constant lower bound Ω(ηKlog⁡T)\Omega(\eta K \log T), which follows from subtle hard-instance constructions and a tailored decomposition of the Bayes prior. Moreover, in the low-regularization regime (i.e., large η\eta), we show that the KL-regularized regret for MABs is η\eta-independent and scales as Θ~(KT)\tilde{\Theta}(\sqrt{KT}). Overall, our results provide a thorough understanding of KL-regularized MABs across all regimes of η\eta and yield nearly optimal bounds in terms of KK, η\eta, and TT.

Ask about this paper

Your agent reads all of it.

Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.

Questions to start from

Your agent calls

Luneget_paper_fulltext

Ask in Lune

Free to start. No credit card required.

lune papers fulltext b09159f7-bade-40e4-875b-52edc3c28cbb

Builds on16

Related papers

Dusk over the sea between two cliffs drawn in fine vertical lines