Instance-optimal PAC Algorithms for Contextual Bandits
Zhaoqi Li, Lillian J. Ratliff, Houssam Nassif, Kevin Jamieson, Lalit Jain
Abstract
In the stochastic contextual bandit setting, regret-minimizing algorithms have been extensively researched, but their instance-minimizing best-arm identification counterparts remain seldom studied. In this work, we focus on the stochastic bandit problem in the - setting: given a policy class the goal of the learner is to return a policy whose expected reward is within of the optimal policy with probability greater than . We characterize the first PAC sample complexity of contextual bandits through a quantity , and provide matching upper and lower bounds in terms of for the agnostic and linear contextual best-arm identification settings. We show that no algorithm can be simultaneously minimax-optimal for regret minimization and instance-dependent PAC for best-arm identification. Our main result is a new instance-optimal and computationally efficient algorithm that relies on a polynomial number of calls to an argmax oracle.
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 papers9
- Instance-Dependent Near-Optimal Policy Identification in Linear MDPs via Online Experiment DesignAndrew Wagenmaker, Kevin JamiesonNeurIPS 2022 · 38 citations
- Proportional Response: Contextual Bandits for Simple and Cumulative Regret MinimizationSanath Kumar Krishnamurthy, Ruohan Zhan, Susan Athey, Emma BrunskillNeurIPS 2023 · 15 citations
- Darwin: Flexible Learning-based CDN CachingJiayi Chen, Nihal Sharma, Tarannum Khan, Shu Liu et al.SIGCOMM 2023 · 13 citations
- Multi-task Representation Learning for Pure Exploration in Linear BanditsYihan Du, Longbo Huang, Wen SunICML 2023 · 6 citations
- Experiment Planning with Function ApproximationAldo Pacchiano, Jonathan Lee, Emma BrunskillNeurIPS 2023 · 6 citations
Builds on5
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Gamification of Pure Exploration for Linear BanditsRémy Degenne, Pierre Ménard, Xuedong Shang, Michal ValkoICML 2020 · 86 citations
- An Asymptotically Optimal Primal-Dual Incremental Algorithm for Contextual Linear BanditsAndrea Tirinzoni, Matteo Pirotta, Marcello Restelli, Alessandro LazaricNeurIPS 2020 · 37 citations
- Improved Confidence Bounds for the Linear Logistic Model and Applications to BanditsKwang-Sung Jun, Lalit Jain, Houssam Nassif, Blake MasonICML 2021 · 30 citations
- Design of Experiments for Stochastic Contextual Linear BanditsAndrea Zanette, Kefan Dong, Jonathan N. Lee, Emma BrunskillNeurIPS 2021 · 24 citations
Related papers
- Low-Rank Bandits via Tight Two-to-Infinity Singular Subspace RecoveryYassir Jedra, William Réveillard, Stefan Stojanovic, Alexandre ProutièreICML 2024 · 3 citations
- Bandits with many optimal armsRianne de Heide, James Cheshire, Pierre Ménard, Alexandra CarpentierNeurIPS 2021 · 28 citations
- Multi-Armed Bandits with Bounded Arm-Memory: Near-Optimal Guarantees for Best-Arm Identification and Regret MinimizationArnab Maiti, Vishakha Patil, Arindam KhanNeurIPS 2021 · 19 citations
- An Optimal Elimination Algorithm for Learning a Best ArmAvinatan Hassidim, Ron Kupfer, Yaron SingerNeurIPS 2020 · 17 citations
- Near Instance-Optimal PAC Reinforcement Learning for Deterministic MDPsAndrea Tirinzoni, Aymen Al Marjani, Emilie KaufmannNeurIPS 2022 · 20 citations
