Stochastic Gradient Succeeds for Bandits
Not on the map. Why?
Abstract (arXiv)
We show that the stochastic gradient bandit algorithm converges to a globally optimal policy at an O(1/t) rate, even with a constant step size. Remarkably, global convergence of the stochastic gradient bandit algorithm has not been previously established, even though it is an old algorithm known to be applicable to bandits. The new result is achieved by establishing two novel technical findings: first, the noise of the stochastic updates in the gradient bandit algorithm satisfies a strong ``growth condition'' property, where the variance diminishes whenever progress becomes small, implying that additional noise control via diminishing step sizes is unnecessary; second, a form of ``weak exploration'' is automatically achieved through the stochastic gradient updates, since they prevent the action probabilities from decaying faster than O(1/t), thus ensuring that every action is sampled infinitely often with probability 1. These two findings can be used to show that the stochastic gradient update is already ``sufficient'' for bandits in the sense that exploration versus exploitation is automatically balanced in a manner that ensures almost sure convergence to a global optimum. These novel theoretical findings are further verified by experimental results.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 24 of its 49 references that are in the map. First posted on arXiv 27 Feb 2024.
Builds on
24 of 49 references.
References · 24
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
- 2015Trust Region Policy Optimizationcited by 2,580
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
- 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 264
- 2019Global Convergence of Policy Gradient Methods to (Almost) Locally Optimal Policiescited by 124
- 2020Sample Efficient Reinforcement Learning with REINFORCEcited by 86
- 2021A general sample complexity analysis of vanilla policy gradientcited by 76
- 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
12 more
- 2017Boltzmann Exploration Done Rightcited by 69
- 2018Variance Reduction for Policy Gradient with Action-Dependent Factorized Baselinescited by 68
- 2013Fast Convergence of Stochastic Gradient Descent under a Strong Growth Conditioncited by 54
- 2018The Mirage of Action-Dependent Baselines in Reinforcement Learningcited by 48
- 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 48
- 2021Leveraging Non-uniformity in First-order Non-convex Optimizationcited by 44
- 2020Beyond variance reduction: Understanding the true impact of baselines on policy optimizationcited by 32
- 2018Variance Reduction for Reinforcement Learning in Input-Driven Environmentscited by 30
- 2021Beyond Exact Gradients: Convergence of Stochastic Soft-Max Policy Gradient Methods with Entropy Regularizationcited by 25
- 2023The Role of Baselines in Policy Gradient Optimizationcited by 25
- 2021Understanding the Effect of Stochasticity in Policy Optimizationcited by 18
- 2022Block Policy Mirror Descentcited by 11
Built on by
Citing papers per year: 2024: 1; 2025: 1; 2026: 8.
10 papers.
- 2024Towards Principled, Practical Policy Gradient for Bandits and Tabular MDPscited by 7
- 2026IsoCompute Playbook: Optimally Scaling Sampling Compute for LLM RLcited by 6
- 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradientscited by 5
- 2025Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning ratescited by 4
- 2026A Diffusion Analysis of Policy Gradient for Stochastic Banditscited by 4
- 2026How Log-Barrier Helps Exploration in Policy Optimizationcited by 1
- 2026A Lyapunov Analysis of Softmax Policy Gradient for Stochastic Banditscited by 1
- 2026Softmax gradient policy for variance minimization and risk-averse multi armed banditscited by 0
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learningcited by 0
- 2026Fine-Tuning on Self-Generated and Reward-Weighted Data: Learning Dynamics, Convergence Rates, and Benefits of Off-Policynesscited by 0
- 2026Fine-Tuning on Self-Generated and Reward-Weighted Data: Learning Dynamics, Convergence Rates, and Benefits of Off-Policyness29 Sep 2026
- 2026Shared Actors Need Not Share Critics: Effects of Value Mismatch in Parallel Reinforcement Learning27 Aug 2026
- 2026Softmax gradient policy for variance minimization and risk-averse multi armed bandits31 Mar 2026
- 2026A Lyapunov Analysis of Softmax Policy Gradient for Stochastic Bandits27 Mar 2026
- 2026How Log-Barrier Helps Exploration in Policy Optimization16 Mar 2026
- 2026IsoCompute Playbook: Optimally Scaling Sampling Compute for LLM RL12 Mar 2026
- 2026A Diffusion Analysis of Policy Gradient for Stochastic Bandits10 Mar 2026
- 2026Why GRPO Needs Normalization: A Local-Curvature Perspective on Adaptive Gradients30 Jan 2026
- 2025Small steps no more: Global convergence of stochastic gradient bandits for arbitrary learning rates11 Feb 2025
- 2024Towards Principled, Practical Policy Gradient for Bandits and Tabular MDPs21 May 2024