Efficient Local Planning with Linear Function Approximation
Not on the map. Why?
Abstract (arXiv)
We study query and computationally efficient planning algorithms with linear function approximation and a simulator. We assume that the agent only has local access to the simulator, meaning that the agent can only query the simulator at states that have been visited before. This setting is more practical than many prior works on reinforcement learning with a generative model. We propose two algorithms, named confident Monte Carlo least square policy iteration (Confident MC-LSPI) and confident Monte Carlo Politex (Confident MC-Politex) for this setting. Under the assumption that the Q-functions of all policies are linear in known features of the state-action pairs, we show that our algorithms have polynomial query and computational costs in the dimension of the features, the effective planning horizon, and the targeted sub-optimality, while these costs are independent of the size of the state space. One technical contribution of our work is the introduction of a novel proof technique that makes use of a virtual policy iteration algorithm. We use this method to leverage existing results on ℓ_∞-bounded approximate policy iteration to show that our algorithm can learn the optimal policy for the given initial state even only with local access to the simulator. We believe that this technique can be extended to broader settings beyond this work.
Key facts. Cited by 21 papers in the map (data through 7 Oct 2026). Builds on 26 of its 53 references that are in the map. First posted on arXiv 12 Aug 2021.
Builds on
26 of 53 references.
References · 26
- 2016Contextual Decision Processes with Low Bellman Rank are PAC-Learnablecited by 290
- 2020Model-Based Reinforcement Learning with Value-Targeted Regressioncited by 233
- 2021Bellman Eluder Dimension: New Rich Classes of RL Problems, and Sample-Efficient Algorithmscited by 222
- 2019Go-Explore: a New Approach for Hard-Exploration Problemscited by 207
- 2019Provably Efficient Exploration in Policy Optimizationcited by 203
- 2019Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret Boundcited by 199
- 2020Learning Near Optimal Policies with Low Inherent Bellman Errorcited by 180
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2019Is a Good Representation Sufficient for Sample Efficient Reinforcement Learning?cited by 140
- 2019Model-Based Reinforcement Learning with a Generative Model is Minimax Optimalcited by 124
- 2019Learning with Good Feature Representations in Bandits and in RL with a Generative Modelcited by 124
- 2021Provably Efficient Reinforcement Learning with Linear Function Approximation Under Adaptivity Constraintscited by 124
14 more
- 2019Optimism in Reinforcement Learning with Generalized Linear Function Approximationcited by 111
- 2020Breaking the Sample Size Barrier in Model-Based Reinforcement Learning with a Generative Modelcited by 108
- 2020Provably Efficient Reinforcement Learning for Discounted MDPs with Feature Mappingcited by 102
- 2020PC-PG: Policy Cover Directed Exploration for Provable Policy Gradient Learningcited by 91
- 2020Exponential Lower Bounds for Planning in MDPs With Linearly-Realizable Optimal Action-Value Functionscited by 71
- 2021Cautiously Optimistic Policy Optimization and Exploration with Linear Function Approximationcited by 53
- 2020Learning Infinite-horizon Average-reward MDPs with Linear Function Approximationcited by 48
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2021Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisitingcited by 27
- 2020Agnostic Q-learning with Function Approximation in Deterministic Systems: Tight Bounds on Approximation Error and Sample Complexitycited by 19
- 2021Improved Regret Bound and Experience Replay in Regularized Policy Iterationcited by 18
- 2020Efficient Planning in Large MDPs with Weak Linear Function Approximationcited by 17
- 2021On Query-efficient Planning in MDPs under Linear Realizability of the Optimal State-value Functioncited by 17
- 2019Comments on the Du-Kakade-Wang-Yang Lower Boundscited by 16
Built on by
Citing papers per year: 2021: 1; 2022: 5; 2023: 7; 2024: 3; 2025: 2; 2026: 3.
21 papers.
Sort:
By area:
- 2024Dataset Reset Policy Optimization for RLHFcited by 41
- 2023Distributionally Robust Model-based Reinforcement Learning with Large State Spacescited by 17
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLcited by 17
- 2026Reuse your FLOPs: Scaling RL on Hard Problems by Conditioning on Very Off-Policy Prefixescited by 16
- 2022Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Modelcited by 15
- 2022Provably Efficient Causal Model-Based Reinforcement Learning for Systematic Generalizationcited by 12
- 2023Sample Efficient Deep Reinforcement Learning via Local Planningcited by 9
- 2023Hardness of Independent Learning and Sparse Equilibrium Computation in Markov Gamescited by 9
- 2023Can Agents Run Relay Race with Strangers? Generalization of RL to Out-of-Distribution Trajectoriescited by 9
- 2022Efficient Global Planning in Large MDPs via Stochastic Primal-Dual Optimizationcited by 8
- 2022Confident Approximate Policy Iteration for Efficient Local Planning in qπ-realizable MDPscited by 8
- 2022Near-optimal Policy Identification in Active Reinforcement Learningcited by 7
9 more
- 2021Improved Algorithms for Misspecified Linear Markov Decision Processescited by 6
- 2023Efficient Planning in Combinatorial Action Spaces with Applications to Cooperative Multi-Agent Reinforcement Learningcited by 6
- 2023Exponential Hardness of Reinforcement Learning with Linear Function Approximationcited by 4
- 2023Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practicecited by 4
- 2024Confident Natural Policy Gradient for Local Planning in qπ-realizable Constrained MDPscited by 4
- 2026Near-Optimal Sample Complexity for Online Constrained MDPscited by 1
- 2025In-Context Compositional Q-Learning for Offline Reinforcement Learningcited by 0
- 2025Computational Hardness of Reinforcement Learning with Partial qπ-Realizabilitycited by 0
- 2026Frozen Policy Iteration: Computationally Efficient RL under Linear Qπ Realizability for Deterministic Dynamicscited by 0
- 2026Frozen Policy Iteration: Computationally Efficient RL under Linear Qπ Realizability for Deterministic Dynamics28 Feb 2026
- 2026Near-Optimal Sample Complexity for Online Constrained MDPs16 Feb 2026
- 2026Reuse your FLOPs: Scaling RL on Hard Problems by Conditioning on Very Off-Policy Prefixes26 Jan 2026
- 2025Computational Hardness of Reinforcement Learning with Partial qπ-Realizability24 Oct 2025
- 2025In-Context Compositional Q-Learning for Offline Reinforcement Learning28 Sep 2025
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RL26 Oct 2024
- 2024Confident Natural Policy Gradient for Local Planning in qπ-realizable Constrained MDPs26 Jun 2024
- 2024Dataset Reset Policy Optimization for RLHF12 Apr 2024
- 2023Distributionally Robust Model-based Reinforcement Learning with Large State Spaces5 Sep 2023
- 2023Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practice22 May 2023
- 2023Can Agents Run Relay Race with Strangers? Generalization of RL to Out-of-Distribution Trajectories26 Apr 2023
- 2023Hardness of Independent Learning and Sparse Equilibrium Computation in Markov Games22 Mar 2023
9 more
- 2023Exponential Hardness of Reinforcement Learning with Linear Function Approximation25 Feb 2023
- 2023Efficient Planning in Combinatorial Action Spaces with Applications to Cooperative Multi-Agent Reinforcement Learning8 Feb 2023
- 2023Sample Efficient Deep Reinforcement Learning via Local Planning29 Jan 2023
- 2022Near-optimal Policy Identification in Active Reinforcement Learning19 Dec 2022
- 2022Confident Approximate Policy Iteration for Efficient Local Planning in qπ-realizable MDPs27 Oct 2022
- 2022Efficient Global Planning in Large MDPs via Stochastic Primal-Dual Optimization21 Oct 2022
- 2022Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Model22 Aug 2022
- 2022Provably Efficient Causal Model-Based Reinforcement Learning for Systematic Generalization14 Feb 2022
- 2021Improved Algorithms for Misspecified Linear Markov Decision Processes12 Sep 2021
- 2024Dataset Reset Policy Optimization for RLHFcited by 41
- 2024Dataset Reset Policy Optimization for RLHF12 Apr 2024
- 2023Distributionally Robust Model-based Reinforcement Learning with Large State Spacescited by 17
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RLcited by 17
- 2022Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Modelcited by 15
- 2022Provably Efficient Causal Model-Based Reinforcement Learning for Systematic Generalizationcited by 12
- 2023Sample Efficient Deep Reinforcement Learning via Local Planningcited by 9
- 2023Hardness of Independent Learning and Sparse Equilibrium Computation in Markov Gamescited by 9
- 2023Can Agents Run Relay Race with Strangers? Generalization of RL to Out-of-Distribution Trajectoriescited by 9
- 2022Efficient Global Planning in Large MDPs via Stochastic Primal-Dual Optimizationcited by 8
- 2022Confident Approximate Policy Iteration for Efficient Local Planning in qπ-realizable MDPscited by 8
- 2022Near-optimal Policy Identification in Active Reinforcement Learningcited by 7
- 2021Improved Algorithms for Misspecified Linear Markov Decision Processescited by 6
- 2023Efficient Planning in Combinatorial Action Spaces with Applications to Cooperative Multi-Agent Reinforcement Learningcited by 6
12 most cited of 18 All 18 (JSON)
- 2026Frozen Policy Iteration: Computationally Efficient RL under Linear Qπ Realizability for Deterministic Dynamics28 Feb 2026
- 2026Near-Optimal Sample Complexity for Online Constrained MDPs16 Feb 2026
- 2025Computational Hardness of Reinforcement Learning with Partial qπ-Realizability24 Oct 2025
- 2024Overcoming the Sim-to-Real Gap: Leveraging Simulation to Learn to Explore for Real-World RL26 Oct 2024
- 2024Confident Natural Policy Gradient for Local Planning in qπ-realizable Constrained MDPs26 Jun 2024
- 2023Distributionally Robust Model-based Reinforcement Learning with Large State Spaces5 Sep 2023
- 2023Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practice22 May 2023
- 2023Can Agents Run Relay Race with Strangers? Generalization of RL to Out-of-Distribution Trajectories26 Apr 2023
- 2023Hardness of Independent Learning and Sparse Equilibrium Computation in Markov Games22 Mar 2023
- 2023Exponential Hardness of Reinforcement Learning with Linear Function Approximation25 Feb 2023
- 2023Efficient Planning in Combinatorial Action Spaces with Applications to Cooperative Multi-Agent Reinforcement Learning8 Feb 2023
- 2023Sample Efficient Deep Reinforcement Learning via Local Planning29 Jan 2023
12 newest of 18 All 18 (JSON)