mlsanityData to 14 Oct 2026

Stochastic Gradient Succeeds for Bandits

arXiv 2402.17235 ↗First posted 27 Feb 2024cs.LGcited by 10

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

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  4. 2015Trust Region Policy Optimizationcited by 2,580
  5. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
  6. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
  7. 2021A Survey of Uncertainty in Deep Neural Networkscited by 680
  8. 2020On the Global Convergence Rates of Softmax Policy Gradient Methodscited by 264
  9. 2019Global Convergence of Policy Gradient Methods to (Almost) Locally Optimal Policiescited by 124
  10. 2020Sample Efficient Reinforcement Learning with REINFORCEcited by 86
  11. 2021A general sample complexity analysis of vanilla policy gradientcited by 76
  12. 2021On the Convergence and Sample Efficiency of Variance-Reduced Policy Gradient Methodcited by 70
12 more
  1. 2017Boltzmann Exploration Done Rightcited by 69
  2. 2018Variance Reduction for Policy Gradient with Action-Dependent Factorized Baselinescited by 68
  3. 2013Fast Convergence of Stochastic Gradient Descent under a Strong Growth Conditioncited by 54
  4. 2018The Mirage of Action-Dependent Baselines in Reinforcement Learningcited by 48
  5. 2021Softmax Policy Gradient Methods Can Take Exponential Time to Convergecited by 48
  6. 2021Leveraging Non-uniformity in First-order Non-convex Optimizationcited by 44
  7. 2020Beyond variance reduction: Understanding the true impact of baselines on policy optimizationcited by 32
  8. 2018Variance Reduction for Reinforcement Learning in Input-Driven Environmentscited by 30
  9. 2021Beyond Exact Gradients: Convergence of Stochastic Soft-Max Policy Gradient Methods with Entropy Regularizationcited by 25
  10. 2023The Role of Baselines in Policy Gradient Optimizationcited by 25
  11. 2021Understanding the Effect of Stochasticity in Policy Optimizationcited by 18
  12. 2022Block Policy Mirror Descentcited by 11

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 1; 2025: 1; 2026: 8.

10 papers.