Experiment Planning with Function Approximation
Aldo Pacchiano, Jonathan Lee, Emma Brunskill
Abstract
We study the problem of experiment planning with function approximation in contextual bandit problems. In settings where there is a significant overhead to deploying adaptive algorithms-for example, when the execution of the data collection policies is required to be distributed, or a human in the loop is needed to implement these policies-producing in advance a set of policies for data collection is paramount. We study the setting where a large dataset of contexts but not rewards is available and may be used by the learner to design an effective data collection strategy. Although when rewards are linear this problem has been well studied [53] , results are still missing for more complex reward models. In this work we propose two experiment planning strategies compatible with function approximation. The first is an eluder planning and sampling procedure that can recover optimality guarantees depending on the eluder dimension [42] of the reward function class. For the second, we show that a uniform sampler achieves competitive optimality rates in the setting where the number of actions is small. We finalize our results introducing a statistical gap fleshing out the fundamental differences between planning and adaptive learning and provide results for planning with model selection.
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 b5496732-af0d-49c0-802c-1b674e7cd79dCited by top-tier papers3
- Transductive Active Learning: Theory and ApplicationsJonas Hübotter, Bhavya Sukhija, Lenart Treven, Yarden As et al.NeurIPS 2024 · 24 citations
- Active Evaluation Acquisition for Efficient LLM BenchmarkingYang Li, Jie Ma, Miguel Ballesteros, Yassine Benajiba et al.ICML 2025
- Second Order Bounds for Contextual Bandits with Function ApproximationAldo PacchianoICLR 2025
Builds on20
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
- Beyond UCB: Optimal and Efficient Contextual Bandits with Regression OraclesDylan J. Foster, Alexander RakhlinICML 2020 · 241 citations
- Reward-Free Exploration for Reinforcement LearningChi Jin, Akshay Krishnamurthy, Max Simchowitz, Tiancheng YuICML 2020 · 226 citations
- Bilinear Classes: A Structural Framework for Provable Generalization in RLSimon S. Du, Sham M. Kakade, Jason D. Lee, Shachar Lovett et al.ICML 2021 · 207 citations
- Learning with Good Feature Representations in Bandits and in RL with a Generative ModelTor Lattimore, Csaba Szepesvári, Gellért WeiszICML 2020 · 181 citations
Related papers
- How Does Variance Shape the Regret in Contextual Bandits?Zeyu Jia, Jian Qian, Alexander Rakhlin, Chen-Yu WeiNeurIPS 2024 · 13 citations
- Design of Experiments for Stochastic Contextual Linear BanditsAndrea Zanette, Kefan Dong, Jonathan N. Lee, Emma BrunskillNeurIPS 2021 · 24 citations
- Offline Neural Contextual Bandits: Pessimism, Optimization and GeneralizationThanh Nguyen-Tang, Sunil Gupta, A. Tuan Nguyen, Svetha VenkateshICLR 2022 · 35 citations
- Contextual Bandits with Smooth Regret: Efficient Learning in Continuous Action SpacesYinglun Zhu, Paul MineiroICML 2022 · 19 citations
- On the Interplay Between Misspecification and Sub-optimality Gap in Linear Contextual BanditsWeitong Zhang, Jiafan He, Zhiyuan Fan, Quanquan GuICML 2023 · 6 citations
