Thompson Sampling for Robust Transfer in Multi-Task Bandits
Zhi Wang, Chicheng Zhang, Kamalika Chaudhuri
Abstract
We study the problem of online multi-task learning where the tasks are performed within similar but not necessarily identical multi-armed bandit environments. In particular, we study how a learner can improve its overall performance across multiple related tasks through robust transfer of knowledge. While an upper confidence bound (UCB)-based algorithm has recently been shown to achieve nearly-optimal performance guarantees in a setting where all tasks are solved concurrently, it remains unclear whether Thompson sampling (TS) algorithms, which have superior empirical performance in general, share similar theoretical properties. In this work, we present a TS-type algorithm for a more general online multi-task learning protocol, which extends the concurrent setting. We provide its frequentist analysis and prove that it is also nearly-optimal using a novel concentration inequality for multi-task data aggregation at random stopping times. Finally, we evaluate the algorithm on synthetic data and show that the TS-type algorithm enjoys superior empirical performance in comparison with the UCB-based algorithm and a baseline algorithm that performs TS for each individual task without transfer.
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 papers4
- Online Clustering of Bandits with Misspecified User ModelsZhiyong Wang, Jize Xie, Xutong Liu, Shuai Li et al.NeurIPS 2023 · 16 citations
- Transportability for Bandits with Data from Different EnvironmentsAlexis Bellot, Alan Malek, Silvia ChiappaNeurIPS 2023 · 11 citations
- Global Perception Based Autoregressive Neural ProcessesJinyang TaiICCV 2023 · 1 citation
- Global Human-guided Counterfactual Explanations for Molecular Properties via Reinforcement LearningDanqing Wang, Antonis Antoniades, Kha-Dinh Luong, Edwin Zhang et al.KDD 2024
Builds on5
- Meta-Thompson SamplingBranislav Kveton, Mikhail Konobeev, Manzil Zaheer, Chih-Wei Hsu et al.ICML 2021 · 74 citations
- No Regrets for Learning the Prior in BanditsSoumya Basu, Branislav Kveton, Manzil Zaheer, Csaba SzepesváriNeurIPS 2021 · 39 citations
- MOTS: Minimax Optimal Thompson SamplingTianyuan Jin, Pan Xu, Jieming Shi, Xiaokui Xiao et al.ICML 2021 · 37 citations
- Metadata-based Multi-Task Bandits with Bayesian Hierarchical ModelsRunzhe Wan, Lin Ge, Rui SongNeurIPS 2021 · 33 citations
- Provably efficient multi-task reinforcement learning with model transferChicheng Zhang, Zhi WangNeurIPS 2021 · 20 citations
Related papers
- Latent Bandits RevisitedJoey Hong, Branislav Kveton, Manzil Zaheer, Yinlam Chow et al.NeurIPS 2020 · 55 citations
- Improved Bayes Regret Bounds for Multi-Task Hierarchical Bayesian Bandit AlgorithmsJiechao Guan, Hui XiongNeurIPS 2024 · 3 citations
- Lenient Regret for Multi-Armed BanditsNadav Merlis, Shie MannorAAAI 2021 · 10 citations
- Offline-to-Online Hyperparameter Transfer for Stochastic BanditsDravyansh Sharma, Arun SuggalaAAAI 2025 · 8 citations
- Thompson Sampling for (Combinatorial) Pure ExplorationSiwei Wang, Jun ZhuICML 2022 · 8 citations
