Graphical Models Meet Bandits: A Variational Thompson Sampling Approach
Tong Yu, Branislav Kveton, Zheng Wen, Ruiyi Zhang, Ole J. Mengshoel
Abstract
We propose a novel framework for structured bandits, which we call an influence diagram bandit. Our framework uses a graphical model to capture complex statistical dependencies between actions, latent variables, and observations; and thus unifies and extends many existing models, such as combinatorial semi-bandits, cascading bandits, and low-rank bandits. We develop novel online learning algorithms that learn to act efficiently in our models. The key idea is to track a structured posterior distribution of model parameters, either exactly or approximately. To act, we sample model parameters from their posterior and then use the structure of the influence diagram to find the most optimistic action under the sampled parameters. We empirically evaluate our algorithms in three structured bandit problems, and show that they perform as well as or better than problem-specific state-of-the-art baselines.
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
- An Analysis of Ensemble SamplingChao Qin, Zheng Wen, Xiuyuan Lu, Benjamin Van RoyNeurIPS 2022 · 30 citations
- Differentiable Meta-Learning of Bandit PoliciesCraig Boutilier, Chih-Wei Hsu, Branislav Kveton, Martin Mladenov et al.NeurIPS 2020 · 23 citations
- Deep Hierarchy in BanditsJoey Hong, Branislav Kveton, Sumeet Katariya, Manzil Zaheer et al.ICML 2022 · 21 citations
- VITS : Variational Inference Thompson Sampling for contextual banditsPierre Clavier, Tom Huix, Alain Oliviero DurmusICML 2024 · 6 citations
- Diffusion Models Meet Contextual BanditsImad AoualiNeurIPS 2025
Related papers
- Structural Causal Bandits under Markov EquivalenceMin Woo Park, Andy Arditi, Elias Bareinboim, Sanghack LeeNeurIPS 2025 · 3 citations
- Approximate Allocation Matching for Structural Causal Bandits with Unobserved ConfoundersLai Wei, Muhammad Qasim Elahi, Mahsa Ghasemi, Murat KocaogluNeurIPS 2023 · 10 citations
- Combinatorial Causal BanditsShi Feng, Wei ChenAAAI 2023 · 16 citations
- Latent Bandits RevisitedJoey Hong, Branislav Kveton, Manzil Zaheer, Yinlam Chow et al.NeurIPS 2020 · 55 citations
- Cascading Reinforcement LearningYihan Du, R. Srikant, Wei ChenICLR 2024 · 2 citations
