Universal and data-adaptive algorithms for model selection in linear contextual bandits
Vidya K. Muthukumar, Akshay Krishnamurthy
Abstract
Model selection in contextual bandits is an important complementary problem to regret minimization with respect to a fixed model class. We consider the simplest non-trivial instance of model-selection: distinguishing a simple multi-armed bandit problem from a linear contextual bandit problem. Even in this instance, current state-of-the-art methods explore in a suboptimal manner and require strong"feature-diversity"conditions. In this paper, we introduce new algorithms that a) explore in a data-adaptive manner, and b) provide model selection guarantees of the form with no feature diversity conditions whatsoever, where denotes the dimension of the linear model and denotes the total number of rounds. The first algorithm enjoys a"best-of-both-worlds"property, recovering two prior results that hold under distinct distributional assumptions, simultaneously. The second removes distributional assumptions altogether, expanding the scope for tractable model selection. Our approach extends to model selection among nested linear contextual bandits under some additional assumptions.
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 papers1
Ask how each one uses itBuilds on5
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Model Selection in Contextual Stochastic Bandit ProblemsAldo Pacchiano, My Phan, Yasin Abbasi-Yadkori, Anup Rao et al.NeurIPS 2020 · 107 citations
- Dynamic Balancing for Model Selection in Bandits and RLAshok Cutkosky, Christoph Dann, Abhimanyu Das, Claudio Gentile et al.ICML 2021 · 40 citations
- Leveraging Good Representations in Linear Contextual BanditsMatteo Papini, Andrea Tirinzoni, Marcello Restelli, Alessandro Lazaric et al.ICML 2021 · 35 citations
- The Pareto Frontier of model selection for general Contextual BanditsTeodor Vanislavov Marinov, Julian ZimmertNeurIPS 2021 · 31 citations
Related papers
- Best of Both Worlds Model SelectionAldo Pacchiano, Christoph Dann, Claudio GentileNeurIPS 2022 · 12 citations
- Feature and Parameter Selection in Stochastic Linear BanditsAhmadreza Moradipari, Berkay Turan, Yasin Abbasi-Yadkori, Mahnoosh Alizadeh et al.ICML 2022 · 6 citations
- Lasso Bandit with Compatibility Condition on Optimal ArmHarin Lee, Taehyun Hwang, Min-hwan OhICLR 2025
- Exploration via Feature Perturbation in Contextual BanditsSeouh-won Yi, Min-hwan OhNeurIPS 2025
- An Asymptotically Optimal Primal-Dual Incremental Algorithm for Contextual Linear BanditsAndrea Tirinzoni, Matteo Pirotta, Marcello Restelli, Alessandro LazaricNeurIPS 2020 · 37 citations
