Thompson Sampling for High-Dimensional Sparse Linear Contextual Bandits
Sunrit Chakraborty, Saptarshi Roy, Ambuj Tewari
Abstract
We consider the stochastic linear contextual bandit problem with high-dimensional features. We analyze the Thompson sampling algorithm using special classes of sparsity-inducing priors (e.g., spike-and-slab) to model the unknown parameter and provide a nearly optimal upper bound on the expected cumulative regret. To the best of our knowledge, this is the first work that provides theoretical guarantees of Thompson sampling in high-dimensional and sparse contextual bandits. For faster computation, we use variational inference instead of Markov Chain Monte Carlo (MCMC) to approximate the posterior distribution. Extensive simulations demonstrate the improved performance of our proposed algorithm over existing ones.
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.
Cited by top-tier papers5
- Linear Bandits with Partially Observable FeaturesWonyoung Kim, Sungwoo Park, Garud Iyengar, Assaf Zeevi et al.ICML 2025 · 3 citations
- Sparsity-Agnostic Linear Bandits with Adaptive AdversariesTianyuan Jin, Kyoungseok Jang, Nicolò Cesa-BianchiNeurIPS 2024 · 2 citations
- On the necessity of adaptive regularisation: Optimal anytime online learning on ℓp-ballsEmmeran Johnson, David Martínez-Rubio, Ciara Pike-Burke, Patrick RebeschiniNeurIPS 2025 · 1 citation
- Sparse Optimistic Information Directed SamplingLudovic Schwartz, Hamish Flynn, Gergely NeuNeurIPS 2025 · 1 citation
- Lasso Bandit with Compatibility Condition on Optimal ArmHarin Lee, Taehyun Hwang, Min-hwan OhICLR 2025
Builds on2
Related papers
- VITS : Variational Inference Thompson Sampling for contextual banditsPierre Clavier, Tom Huix, Alain Oliviero DurmusICML 2024 · 6 citations
- Langevin Monte Carlo for Contextual BanditsPan Xu, Hongkai Zheng, Eric V. Mazumdar, Kamyar Azizzadenesheli et al.ICML 2022 · 34 citations
- Neural Thompson SamplingWeitong Zhang, Dongruo Zhou, Lihong Li, Quanquan GuICLR 2021 · 152 citations
- Spike and slab variational Bayes for high dimensional logistic regressionKolyan Ray, Botond Szabó, Gabriel ClaraNeurIPS 2020 · 35 citations
- Sparsity-Agnostic Lasso BanditMin-hwan Oh, Garud Iyengar, Assaf ZeeviICML 2021 · 54 citations
