Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning rates
Jincheng Mei, Bo Dai, Alekh Agarwal, Sharan Vaswani, Anant Raj, Csaba Szepesvári, Dale Schuurmans
Abstract
We provide a new understanding of the stochastic gradient bandit algorithm by showing that it converges to a globally optimal policy almost surely using any constant learning rate. This result demonstrates that the stochastic gradient algorithm continues to balance exploration and exploitation appropriately even in scenarios where standard smoothness and noise control assumptions break down. The proofs are based on novel findings about action sampling rates and the relationship between cumulative progress and noise, and extend the current understanding of how simple stochastic gradient methods behave in bandit settings.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 2296ed1d-387a-4174-ba25-c9b46b4ee2e8Cited by top-tier papers2
- Does Stochastic Gradient really succeed for bandits?Dorian Baudry, Emmeran Johnson, Simon Vary, Ciara Pike-Burke et al.NeurIPS 2025 · 3 citations
- REINFORCE Converges to Optimal Policies with Any Learning RateSamuel Robertson, Thang Chu, Bo Dai, Dale Schuurmans et al.NeurIPS 2025 · 2 citations
Builds on9
- Training language models to follow instructions with human feedbackLong Ouyang, Jeffrey Wu, Xu Jiang, Diogo Almeida et al.NeurIPS 2022 · 24,707 citations
- On the Global Convergence Rates of Softmax Policy Gradient MethodsJincheng Mei, Chenjun Xiao, Csaba Szepesvári, Dale SchuurmansICML 2020 · 349 citations
- Sample Efficient Reinforcement Learning with REINFORCEJunzi Zhang, Jongho Kim, Brendan O'Donoghue, Stephen P. BoydAAAI 2021 · 162 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
Related papers
- Stochastic Gradient Succeeds for BanditsJincheng Mei, Zixin Zhong, Bo Dai, Alekh Agarwal et al.ICML 2023 · 6 citations
- Ordering-based Conditions for Global Convergence of Policy Gradient MethodsJincheng Mei, Bo Dai, Alekh Agarwal, Mohammad Ghavamzadeh et al.NeurIPS 2023 · 4 citations
- Understanding the Effect of Stochasticity in Policy OptimizationJincheng Mei, Bo Dai, Chenjun Xiao, Csaba Szepesvári et al.NeurIPS 2021 · 24 citations
- The Role of Baselines in Policy Gradient OptimizationJincheng Mei, Wesley Chung, Valentin Thomas, Bo Dai et al.NeurIPS 2022 · 34 citations
- Convergence and Optimality of Policy Gradient Methods in Weakly Smooth SettingsMatthew Shunshi Zhang, Murat A. Erdogdu, Animesh GargAAAI 2022 · 6 citations
