Efficient First-Order Contextual Bandits: Prediction, Allocation, and Triangular Discrimination
Dylan J. Foster, Akshay Krishnamurthy
Abstract
A recurring theme in statistical learning, online learning, and beyond is that faster convergence rates are possible for problems with low noise, often quantified by the performance of the best hypothesis; such results are known as first-order or small-loss guarantees. While first-order guarantees are relatively well understood in statistical and online learning, adapting to low noise in contextual bandits (and more broadly, decision making) presents major algorithmic challenges. In a COLT 2017 open problem, Agarwal, Krishnamurthy, Langford, Luo, and Schapire asked whether first-order guarantees are even possible for contextual bandits and -- if so -- whether they can be attained by efficient algorithms. We give a resolution to this question by providing an optimal and efficient reduction from contextual bandits to online regression with the logarithmic (or, cross-entropy) loss. Our algorithm is simple and practical, readily accommodates rich function classes, and requires no distributional assumptions beyond realizability. In a large-scale empirical evaluation, we find that our approach typically outperforms comparable non-first-order methods. On the technical side, we show that the logarithmic loss and an information-theoretic quantity called the triangular discrimination play a fundamental role in obtaining first-order guarantees, and we combine this observation with new refinements to the regression oracle reduction framework of Foster and Rakhlin. The use of triangular discrimination yields novel results even for the classical statistical learning model, and we anticipate that it will find broader use.
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 papers31
- Is Behavior Cloning All You Need? Understanding Horizon in Imitation LearningDylan J. Foster, Adam Block, Dipendra MisraNeurIPS 2024 · 112 citations
- REBEL: Reinforcement Learning via Regressing Relative RewardsZhaolin Gao, Jonathan D. Chang, Wenhao Zhan, Owen Oertell et al.NeurIPS 2024 · 82 citations
- First-Order Regret in Reinforcement Learning with Linear Function Approximation: A Robust Estimation ApproachAndrew J. Wagenmaker, Yifang Chen, Max Simchowitz, Simon S. Du et al.ICML 2022 · 49 citations
- Contextual Bandits with Large Action Spaces: Made PracticalYinglun Zhu, Dylan J. Foster, John Langford, Paul MineiroICML 2022 · 34 citations
- The Benefits of Being Distributional: Small-Loss Bounds for Reinforcement LearningKaiwen Wang, Kevin Zhou, Runzhe Wu, Nathan Kallus et al.NeurIPS 2023 · 31 citations
Builds on5
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Adapting to Misspecification in Contextual BanditsDylan J. Foster, Claudio Gentile, Mehryar Mohri, Julian ZimmertNeurIPS 2020 · 111 citations
- Tight First- and Second-Order Regret Bounds for Adversarial Linear BanditsShinji Ito, Shuichi Hirahara, Tasuku Soma, Yuichi YoshidaNeurIPS 2020 · 24 citations
- Tight Bounds on Minimax Regret under Logarithmic Loss via Self-ConcordanceBlair L. Bilodeau, Dylan J. Foster, Daniel M. RoyICML 2020 · 18 citations
- Online and Distribution-Free Robustness: Regression and Contextual Bandits with Huber ContaminationSitan Chen, Frederic Koehler, Ankur Moitra, Morris YauFOCS 2021 · 14 citations
Related papers
- Efficient Contextual Bandits with Uninformed Feedback GraphsMengxiao Zhang, Yuheng Zhang, Haipeng Luo, Paul MineiroICML 2024 · 5 citations
- Contextual Bandits with Online Neural RegressionRohan Deb, Yikun Ban, Shiliang Zuo, Jingrui He et al.ICLR 2024 · 8 citations
- An Improved Relaxation for Oracle-Efficient Adversarial Contextual BanditsKiarash Banihashem, MohammadTaghi Hajiaghayi, Suho Shin, Max SpringerNeurIPS 2023 · 3 citations
- Online Linear Classification with Massart NoiseIlias Diakonikolas, Vasilis Kontonis, Christos Tzamos, Nikos ZarifisICML 2025
- High-dimensional Linear Bandits with KnapsacksWanteng Ma, Dong Xia, Jiashuo JiangICML 2024 · 1 citation
