Does Stochastic Gradient really succeed for bandits?
Dorian Baudry, Emmeran Johnson, Simon Vary, Ciara Pike-Burke, Patrick Rebeschini
Abstract
Recent works of Mei et al. [1,2] have deepened the theoretical understanding of the Stochastic Gradient Bandit (SGB) policy, showing that using a constant learning rate guarantees asymptotic convergence to the optimal policy, and that sufficiently small learning rates can yield logarithmic regret. However, whether logarithmic regret holds beyond small learning rates remains unclear. In this work, we take a step towards characterizing the regret regimes of SGB as a function of its learning rate. For two-armed bandits, we identify a sharp threshold, scaling with the suboptimality gap ∆, below which SGB achieves logarithmic regret on all instances, and above which it can incur polynomial regret on some instances. This result highlights the necessity of knowing (or estimating) ∆ to ensure logarithmic regret with a constant learning rate. For general K-armed bandits, we further show the learning rate must additionally scale inversely with K to avoid polynomial regret. We introduce novel techniques to derive regret upper bounds for SGB, laying the groundwork for future advances in the theory of gradient-based bandit algorithms.
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.
Builds on13
- On the Global Convergence Rates of Softmax Policy Gradient MethodsJincheng Mei, Chenjun Xiao, Csaba Szepesvári, Dale SchuurmansICML 2020 · 349 citations
- On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient MethodJunyu Zhang, Chengzhuo Ni, Zheng Yu, Csaba Szepesvári et al.NeurIPS 2021 · 87 citations
- Escaping the Gravitational Pull of SoftmaxJincheng Mei, Chenjun Xiao, Bo Dai, Lihong Li et al.NeurIPS 2020 · 56 citations
- From Dirichlet to Rubin: Optimistic Exploration in RL without BonusesDaniil Tiapkin, Denis Belomestny, Eric Moulines, Alexey Naumov et al.ICML 2022 · 24 citations
- Differentiable Meta-Learning of Bandit PoliciesCraig Boutilier, Chih-Wei Hsu, Branislav Kveton, Martin Mladenov et al.NeurIPS 2020 · 23 citations
Related papers
- Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning ratesJincheng Mei, Bo Dai, Alekh Agarwal, Sharan Vaswani et al.NeurIPS 2024 · 5 citations
- A Closer Look at the Worst-case Behavior of Multi-armed Bandit AlgorithmsAnand Kalvit, Assaf ZeeviNeurIPS 2021 · 48 citations
- Bandits with many optimal armsRianne de Heide, James Cheshire, Pierre Ménard, Alexandra CarpentierNeurIPS 2021 · 28 citations
- REINFORCE Converges to Optimal Policies with Any Learning RateSamuel Robertson, Thang Chu, Bo Dai, Dale Schuurmans et al.NeurIPS 2025 · 2 citations
- Tight last-iterate convergence rates for no-regret learning in multi-player gamesNoah Golowich, Sarath Pattathil, Constantinos DaskalakisNeurIPS 2020 · 100 citations
