Offline Neural Contextual Bandits: Pessimism, Optimization and Generalization
Thanh Nguyen-Tang, Sunil Gupta, A. Tuan Nguyen, Svetha Venkatesh
Abstract
Offline policy learning (OPL) leverages existing data collected a priori for policy optimization without any active exploration. Despite the prevalence and recent interest in this problem, its theoretical and algorithmic foundations in function approximation settings remain under-developed. In this paper, we consider this problem on the axes of distributional shift, optimization, and generalization in offline contextual bandits with neural networks. In particular, we propose a provably efficient offline contextual bandit with neural network function approximation that does not require any functional assumption on the reward. We show that our method provably generalizes over unseen contexts under a milder condition for distributional shift than the existing OPL works. Notably, unlike any other OPL method, our method learns from the offline data in an online manner using stochastic gradient descent, allowing us to leverage the benefits of online learning into an offline setting. Moreover, we show that our method is more computationally efficient and has a better dependence on the effective dimension of the neural network than an online counterpart. Finally, we demonstrate the empirical effectiveness of our method in a range of synthetic and real-world OPL problems.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c3aeee05-830c-49c2-9095-f442b9c6ecd2Cited by top-tier papers14
- Reward-Directed Conditional Diffusion: Provable Distribution Estimation and Reward ImprovementHui Yuan, Kaixuan Huang, Chengzhuo Ni, Minshuo Chen et al.NeurIPS 2023 · 59 citations
- On the Convergence and Sample Complexity Analysis of Deep Q-Networks with ε-Greedy ExplorationShuai Zhang, Hongkang Li, Meng Wang, Miao Liu et al.NeurIPS 2023 · 57 citations
- Data-Driven Offline Decision-Making via Invariant Representation LearningHan Qi, Yi Su, Aviral Kumar, Sergey LevineNeurIPS 2022 · 43 citations
- On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function ApproximationThanh Nguyen-Tang, Ming Yin, Sunil Gupta, Svetha Venkatesh et al.AAAI 2023 · 24 citations
- PAC-Bayesian Offline Contextual Bandits With GuaranteesOtmane Sakhi, Pierre Alquier, Nicolas ChopinICML 2023 · 23 citations
Builds on13
- Bridging Offline Reinforcement Learning and Imitation Learning: A Tale of PessimismParia Rashidinejad, Banghua Zhu, Cong Ma, Jiantao Jiao et al.NeurIPS 2021 · 373 citations
- Neural Contextual Bandits with UCB-based ExplorationDongruo Zhou, Lihong Li, Quanquan GuICML 2020 · 329 citations
- Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret BoundLin Yang, Mengdi WangICML 2020 · 308 citations
- Finite Depth and Width Corrections to the Neural Tangent KernelBoris Hanin, Mihai NicaICLR 2020 · 169 citations
- Minimax-Optimal Off-Policy Evaluation with Linear Function ApproximationYaqi Duan, Zeyu Jia, Mengdi WangICML 2020 · 161 citations
Related papers
- Optimal Regret for Policy Optimization in Contextual BanditsOrin Levy, Yishay MansourICML 2026 · 1 citation
- Instabilities of Offline RL with Pre-Trained Neural RepresentationRuosong Wang, Yifan Wu, Ruslan Salakhutdinov, Sham M. KakadeICML 2021 · 46 citations
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Offline Contextual Bandits with Overparameterized ModelsDavid Brandfonbrener, William F. Whitney, Rajesh Ranganath, Joan BrunaICML 2021 · 12 citations
- Reward-Biased Maximum Likelihood Estimation for Neural Contextual Bandits: A Distributional Learning PerspectiveYu-Heng Hung, Ping-Chun HsiehAAAI 2023 · 2 citations
