Reward-Free Model-Based Reinforcement Learning with Linear Function Approximation
Not on the map. Why?
Abstract (arXiv)
We study the model-based reward-free reinforcement learning with linear function approximation for episodic Markov decision processes (MDPs). In this setting, the agent works in two phases. In the exploration phase, the agent interacts with the environment and collects samples without the reward. In the planning phase, the agent is given a specific reward function and uses samples collected from the exploration phase to learn a good policy. We propose a new provably efficient algorithm, called UCRL-RFE under the Linear Mixture MDP assumption, where the transition probability kernel of the MDP can be parameterized by a linear function over certain feature mappings defined on the triplet of state, action, and next state. We show that to obtain an ε-optimal policy for arbitrary reward function, UCRL-RFE needs to sample at most tildeO(H5d2ε-2) episodes during the exploration phase. Here, H is the length of the episode, d is the dimension of the feature mapping. We also propose a variant of UCRL-RFE using Bernstein-type bonus and show that it needs to sample at most tildeO(H4d(H + d)ε-2) to achieve an ε-optimal policy. By constructing a special class of linear Mixture MDPs, we also prove that for any reward-free algorithm, it needs to sample at least tilde Omega(H2dε-2) episodes to obtain an ε-optimal policy. Our upper bound matches the lower bound in terms of the dependence on ε and the dependence on d if H ≥ d.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). Builds on 19 of its 31 references that are in the map. First posted on arXiv 12 Oct 2021.
Builds on
19 of 31 references.
References · 19
- 2019Provably Efficient Reinforcement Learning with Linear Function Approximationcited by 514
- 2018Is Q-learning Provably Efficient?cited by 485
- 2020Model-Based Reinforcement Learning with Value-Targeted Regressioncited by 233
- 2019Reinforcement Learning in Feature Space: Matrix Bandit, Kernels, and Regret Boundcited by 199
- 2020Nearly Minimax Optimal Reinforcement Learning for Linear Mixture Markov Decision Processescited by 199
- 2019Tighter Problem-Dependent Regret Bounds in Reinforcement Learning without Domain Knowledge using Value Function Boundscited by 194
- 2020Learning Near Optimal Policies with Low Inherent Bellman Errorcited by 180
- 2020Reward-Free Exploration for Reinforcement Learningcited by 177
- 2019Optimism in Reinforcement Learning with Generalized Linear Function Approximationcited by 111
- 2020A Sharp Analysis of Model-based Reinforcement Learning with Self-Playcited by 104
- 2020Provably Efficient Reinforcement Learning for Discounted MDPs with Feature Mappingcited by 102
- 2019Sample Complexity of Reinforcement Learning using Linearly Combined Model Ensemblescited by 100
7 more
- 2019Frequentist Regret Bounds for Randomized Least-Squares Value Iterationcited by 100
- 2020Fast active learning for pure exploration in reinforcement learningcited by 92
- 2020On Reward-Free Reinforcement Learning with Linear Function Approximationcited by 89
- 2020Adaptive Reward-Free Explorationcited by 77
- 2020Provably Efficient Reward-Agnostic Navigation with Linear Value Iterationcited by 56
- 2020Nearly Minimax Optimal Reward-free Reinforcement Learningcited by 26
- 2021Nearly Minimax Optimal Regret for Learning Infinite-horizon Average-reward MDPs with Linear Function Approximationcited by 18
Built on by
Citing papers per year: 2021: 1; 2022: 8; 2023: 8; 2024: 5; 2025: 3; 2026: 4.
29 papers.
Sort:
By area:
- 2022Human-in-the-loop: Provably Efficient Preference-based Reinforcement Learning with General Function Approximationcited by 80
- 2022BYOL-Explore: Exploration by Bootstrapped Predictioncited by 71
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
- 2022On the Statistical Efficiency of Reward-Free Exploration in Non-Linear RLcited by 27
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitionscited by 16
- 2023Provably Feedback-Efficient Reinforcement Learning via Active Reward Learningcited by 15
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learningcited by 15
- 2023Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPscited by 14
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 14
- 2024Offline Multitask Representation Learning for Reinforcement Learningcited by 14
- 2022Near-Optimal Deployment Efficiency in Reward-Free Reinforcement Learning with Linear Function Approximationcited by 13
17 more
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
- 2022Safe Exploration Incurs Nearly No Additional Sample Complexity for Reward-free RLcited by 9
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?cited by 8
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPscited by 7
- 2023On the Model-Misspecification in Reinforcement Learningcited by 7
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximationcited by 5
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learningcited by 5
- 2025Linear Mixture Distributionally Robust Markov Decision Processescited by 5
- 2021Adaptive Multi-Goal Explorationcited by 4
- 2024Efficient Reinforcement Learning in Probabilistic Reward Machinescited by 4
- 2024Near-Optimal Reinforcement Learning with Self-Play under Adaptivity Constraintscited by 3
- 2025Statistical and Algorithmic Foundations of Reinforcement Learningcited by 2
- 2026A Reward-Free Viewpoint on Multi-Objective Reinforcement Learningcited by 2
- 2025Bayesian Optimization for Dynamic Pricing and Learningcited by 0
- 2026The Horizon Threshold in Cooperative Multi-Agent Reward-Free Explorationcited by 0
- 2026Provably Efficient Offline-to-Online Value Adaptation with General Function Approximationcited by 0
- 2026Generalizing Preference-based Reinforcement Learning: a Rationality Model for Incomparabilitycited by 0
- 2026Generalizing Preference-based Reinforcement Learning: a Rationality Model for Incomparability13 Jul 2026
- 2026A Reward-Free Viewpoint on Multi-Objective Reinforcement Learning27 Apr 2026
- 2026Provably Efficient Offline-to-Online Value Adaptation with General Function Approximation15 Apr 2026
- 2026The Horizon Threshold in Cooperative Multi-Agent Reward-Free Exploration1 Feb 2026
- 2025Bayesian Optimization for Dynamic Pricing and Learning14 Oct 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Linear Mixture Distributionally Robust Markov Decision Processes23 May 2025
- 2024Efficient Reinforcement Learning in Probabilistic Reward Machines19 Aug 2024
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning10 Jul 2024
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximation24 Jun 2024
- 2024Offline Multitask Representation Learning for Reinforcement Learning18 Mar 2024
- 2024Near-Optimal Reinforcement Learning with Self-Play under Adaptivity Constraints2 Feb 2024
17 more
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?9 Oct 2023
- 2023On the Model-Misspecification in Reinforcement Learning19 Jun 2023
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitions11 Jun 2023
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning17 May 2023
- 2023Provably Feedback-Efficient Reinforcement Learning via Active Reward Learning18 Apr 2023
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learning14 Apr 2023
- 2023Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPs20 Mar 2023
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPs17 Mar 2023
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient3 Oct 2022
- 2022Near-Optimal Deployment Efficiency in Reward-Free Reinforcement Learning with Linear Function Approximation3 Oct 2022
- 2022Safe Exploration Incurs Nearly No Additional Sample Complexity for Reward-free RL28 Jun 2022
- 2022On the Statistical Efficiency of Reward-Free Exploration in Non-Linear RL21 Jun 2022
- 2022BYOL-Explore: Exploration by Bootstrapped Prediction16 Jun 2022
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPs23 May 2022
- 2022Human-in-the-loop: Provably Efficient Preference-based Reinforcement Learning with General Function Approximation23 May 2022
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processes26 Jan 2022
- 2021Adaptive Multi-Goal Exploration23 Nov 2021
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitionscited by 16
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learningcited by 15
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 14
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learningcited by 5
- 2026Provably Efficient Offline-to-Online Value Adaptation with General Function Approximationcited by 0
- 2026Provably Efficient Offline-to-Online Value Adaptation with General Function Approximation15 Apr 2026
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning10 Jul 2024
- 2023Provably Efficient Adversarial Imitation Learning with Unknown Transitions11 Jun 2023
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learning17 May 2023
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learning14 Apr 2023
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient3 Oct 2022
- 2022Human-in-the-loop: Provably Efficient Preference-based Reinforcement Learning with General Function Approximationcited by 80
- 2022BYOL-Explore: Exploration by Bootstrapped Predictioncited by 71
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
- 2022On the Statistical Efficiency of Reward-Free Exploration in Non-Linear RLcited by 27
- 2023Provably Feedback-Efficient Reinforcement Learning via Active Reward Learningcited by 15
- 2023Improved Sample Complexity for Reward-free Reinforcement Learning under Low-rank MDPscited by 14
- 2024Offline Multitask Representation Learning for Reinforcement Learningcited by 14
- 2022Near-Optimal Deployment Efficiency in Reward-Free Reinforcement Learning with Linear Function Approximationcited by 13
- 2022Safe Exploration Incurs Nearly No Additional Sample Complexity for Reward-free RLcited by 9
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?cited by 8
- 2023Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPscited by 7
12 most cited of 23 All 23 (JSON)
- 2026Generalizing Preference-based Reinforcement Learning: a Rationality Model for Incomparability13 Jul 2026
- 2026A Reward-Free Viewpoint on Multi-Objective Reinforcement Learning27 Apr 2026
- 2026The Horizon Threshold in Cooperative Multi-Agent Reward-Free Exploration1 Feb 2026
- 2025Bayesian Optimization for Dynamic Pricing and Learning14 Oct 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Linear Mixture Distributionally Robust Markov Decision Processes23 May 2025
- 2024Efficient Reinforcement Learning in Probabilistic Reward Machines19 Aug 2024
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximation24 Jun 2024
- 2024Offline Multitask Representation Learning for Reinforcement Learning18 Mar 2024
- 2024Near-Optimal Reinforcement Learning with Self-Play under Adaptivity Constraints2 Feb 2024
- 2023When is Agnostic Reinforcement Learning Statistically Tractable?9 Oct 2023
- 2023On the Model-Misspecification in Reinforcement Learning19 Jun 2023
12 newest of 23 All 23 (JSON)