Dynamic Balancing for Model Selection in Bandits and RL
Ashok Cutkosky, Christoph Dann, Abhimanyu Das, Claudio Gentile, Aldo Pacchiano, Manish Purohit
Abstract
We propose a framework for model selection by combining base algorithms in stochastic bandits and reinforcement learning. We require a candidate regret bound for each base algorithm that may or may not hold. We select base algorithms to play in each round using a "balancing condition" on the candidate regret bounds. Our approach simultaneously recovers previous worst-case regret bounds, while also obtaining much smaller regret in natural scenarios when some base learners significantly exceed their candidate bounds. Our framework is relevant in many settings, including linear bandits and MDPs with nested function classes, linear bandits with unknown misspecification, and tuning confidence parameters of algorithms such as LinUCB. Moreover, unlike recent efforts in model selection for linear stochastic bandits, our approach can be extended to consider adversarial rather than stochastic contexts.
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 papers17
- Unpacking Reward Shaping: Understanding the Benefits of Reward Engineering on Sample ComplexityAbhishek Gupta, Aldo Pacchiano, Yuexiang Zhai, Sham M. Kakade et al.NeurIPS 2022 · 115 citations
- Model Selection in Contextual Stochastic Bandit ProblemsAldo Pacchiano, My Phan, Yasin Abbasi-Yadkori, Anup Rao et al.NeurIPS 2020 · 107 citations
- Learning in POMDPs is Sample-Efficient with Hindsight ObservabilityJonathan Lee, Alekh Agarwal, Christoph Dann, Tong ZhangICML 2023 · 25 citations
- Reinforcement Learning Can Be More Efficient with Multiple RewardsChristoph Dann, Yishay Mansour, Mehryar MohriICML 2023 · 24 citations
- Best of Both Worlds Model SelectionAldo Pacchiano, Christoph Dann, Claudio GentileNeurIPS 2022 · 12 citations
Builds on3
- Learning Near Optimal Policies with Low Inherent Bellman ErrorAndrea Zanette, Alessandro Lazaric, Mykel J. Kochenderfer, Emma BrunskillICML 2020 · 238 citations
- Adapting to Misspecification in Contextual BanditsDylan J. Foster, Claudio Gentile, Mehryar Mohri, Julian ZimmertNeurIPS 2020 · 111 citations
- Model Selection in Contextual Stochastic Bandit ProblemsAldo Pacchiano, My Phan, Yasin Abbasi-Yadkori, Anup Rao et al.NeurIPS 2020 · 107 citations
Related papers
- Sparsity-Agnostic Linear Bandits with Adaptive AdversariesTianyuan Jin, Kyoungseok Jang, Nicolò Cesa-BianchiNeurIPS 2024 · 2 citations
- Universal and data-adaptive algorithms for model selection in linear contextual banditsVidya K. Muthukumar, Akshay KrishnamurthyICML 2022 · 5 citations
- Corruption-Robust Linear Bandits: Minimax Optimality and Gap-Dependent MisspecificationHaolin Liu, Artin Tajdini, Andrew Wagenmaker, Chen-Yu WeiNeurIPS 2024 · 8 citations
- Feature and Parameter Selection in Stochastic Linear BanditsAhmadreza Moradipari, Berkay Turan, Yasin Abbasi-Yadkori, Mahnoosh Alizadeh et al.ICML 2022 · 6 citations
- On the Interplay Between Misspecification and Sub-optimality Gap in Linear Contextual BanditsWeitong Zhang, Jiafan He, Zhiyuan Fan, Quanquan GuICML 2023 · 6 citations
