Instance-Dependent Near-Optimal Policy Identification in Linear MDPs via Online Experiment Design
Andrew Wagenmaker, Kevin Jamieson
Abstract
While much progress has been made in understanding the minimax sample complexity of reinforcement learning (RL) -- the complexity of learning on the"worst-case"instance -- such measures of complexity often do not capture the true difficulty of learning. In practice, on an"easy"instance, we might hope to achieve a complexity far better than that achievable on the worst-case instance. In this work we seek to understand the"instance-dependent"complexity of learning near-optimal policies (PAC RL) in the setting of RL with linear function approximation. We propose an algorithm, Pedel, which achieves a fine-grained instance-dependent measure of complexity, the first of its kind in the RL with function approximation setting, thereby capturing the difficulty of learning on each particular problem instance. Through an explicit example, we show that Pedel yields provable gains over low-regret, minimax-optimal algorithms and that such algorithms are unable to hit the instance-optimal rate. Our approach relies on a novel online experiment design-based procedure which focuses the exploration budget on the"directions"most relevant to learning a near-optimal policy, and may be of independent interest.
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 papers19
- Leveraging Offline Data in Online Reinforcement LearningAndrew Wagenmaker, Aldo PacchianoICML 2023 · 47 citations
- Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLAndrew Wagenmaker, Kevin Huang, Liyiming Ke, Kevin Jamieson et al.NeurIPS 2024 · 45 citations
- Optimal Exploration for Model-Based RL in Nonlinear SystemsAndrew Wagenmaker, Guanya Shi, Kevin JamiesonNeurIPS 2023 · 29 citations
- Pessimism for Offline Linear Contextual Bandits using Confidence SetsGene Li, Cong Ma, Nati SrebroNeurIPS 2022 · 20 citations
- Refined Regret for Adversarial MDPs with Linear Function ApproximationYan Dai, Haipeng Luo, Chen-Yu Wei, Julian ZimmertICML 2023 · 15 citations
Builds on20
- Model-Based Reinforcement Learning with Value-Targeted RegressionAlex Ayoub, Zeyu Jia, Csaba Szepesvári, Mengdi Wang et al.ICML 2020 · 324 citations
- Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient AlgorithmsChi Jin, Qinghua Liu, Sobhan MiryoosefiNeurIPS 2021 · 264 citations
- Learning Near Optimal Policies with Low Inherent Bellman ErrorAndrea Zanette, Alessandro Lazaric, Mykel J. Kochenderfer, Emma BrunskillICML 2020 · 238 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
- Provably Efficient Reinforcement Learning for Discounted MDPs with Feature MappingDongruo Zhou, Jiafan He, Quanquan GuICML 2021 · 143 citations
Related papers
- Near Instance-Optimal PAC Reinforcement Learning for Deterministic MDPsAndrea Tirinzoni, Aymen Al Marjani, Emilie KaufmannNeurIPS 2022 · 20 citations
- Task-Optimal Exploration in Linear Dynamical SystemsAndrew J. Wagenmaker, Max Simchowitz, Kevin JamiesonICML 2021 · 24 citations
- Instance-Dependent Fixed-Budget Pure Exploration in Reinforcement LearningYeongjong Kim, Yeoneung Kim, Kwang-Sung JunICLR 2026
- Near-Optimal Deployment Efficiency in Reward-Free Reinforcement Learning with Linear Function ApproximationDan Qiao, Yu-Xiang WangICLR 2023
- Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision ProcessesJiafan He, Heyang Zhao, Dongruo Zhou, Quanquan GuICML 2023 · 68 citations
