English

Bayesian Bandit Algorithms with Approximate Inference in Stochastic Linear Bandits

Machine Learning 2025-05-23 v3 Machine Learning

Abstract

Bayesian bandit algorithms with approximate Bayesian inference have been widely used in real-world applications. Despite the superior practical performance, their theoretical justification is less investigated in the literature, especially for contextual bandit problems. To fill this gap, we propose a theoretical framework to analyze the impact of approximate inference in stochastic linear bandits and conduct frequentist regret analysis on two Bayesian bandit algorithms, Linear Thompson Sampling (LinTS) and the extension of Bayesian Upper Confidence Bound, namely Linear Bayesian Upper Confidence Bound (LinBUCB). We demonstrate that when applied in approximate inference settings, LinTS and LinBUCB can universally preserve their original rates of regret upper bound but with a sacrifice of larger constant terms. These results hold for general Bayesian inference approaches, assuming the inference error measured by two different α\alpha-divergences is bounded. Additionally, by introducing a new definition of well-behaved distributions, we show that LinBUCB expedites the regret rate of LinTS from O~(d3/2T)\tilde{O}(d^{3/2}\sqrt{T}) to O~(dT)\tilde{O}(d\sqrt{T}), matching the minimax optimal rate. To our knowledge, this work provides the first regret bounds in the setting of stochastic linear bandits with bounded approximate inference errors.

Keywords

Cite

@article{arxiv.2406.14071,
  title  = {Bayesian Bandit Algorithms with Approximate Inference in Stochastic Linear Bandits},
  author = {Ziyi Huang and Henry Lam and Haofeng Zhang},
  journal= {arXiv preprint arXiv:2406.14071},
  year   = {2025}
}
R2 v1 2026-06-28T17:13:03.522Z