Optimal Horizon-Free Reward-Free Exploration for Linear Mixture MDPs
Not on the map. Why?
Abstract (arXiv)
We study reward-free reinforcement learning (RL) with linear function approximation, where the agent works in two phases: (1) in the exploration phase, the agent interacts with the environment but cannot access the reward; and (2) in the planning phase, the agent is given a reward function and is expected to find a near-optimal policy based on samples collected in the exploration phase. The sample complexities of existing reward-free algorithms have a polynomial dependence on the planning horizon, which makes them intractable for long planning horizon RL problems. In this paper, we propose a new reward-free algorithm for learning linear mixture Markov decision processes (MDPs), where the transition probability can be parameterized as a linear combination of known feature mappings. At the core of our algorithm is uncertainty-weighted value-targeted regression with exploration-driven pseudo-reward and a high-order moment estimator for the aleatoric and epistemic uncertainties. When the total reward is bounded by 1, we show that our algorithm only needs to explore tilde O( d2varepsilon-2) episodes to find an varepsilon-optimal policy, where d is the dimension of the feature mapping. The sample complexity of our algorithm only has a polylogarithmic dependence on the planning horizon and therefore is "horizon-free". In addition, we provide an Omega(d2varepsilon-2) sample complexity lower bound, which matches the sample complexity of our algorithm up to logarithmic factors, suggesting that our algorithm is optimal.
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 41 of its 46 references that are in the map. First posted on arXiv 17 Mar 2023.
Builds on
41 of 46 references.
References · 41
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
- 2017Constrained Policy Optimizationcited by 928
- 2019Provably Efficient Reinforcement Learning with Linear Function Approximationcited by 514
- 2018Reward Constrained Policy Optimizationcited by 397
- 2016Contextual Decision Processes with Low Bellman Rank are PAC-Learnablecited by 290
- 2020Model-Based Reinforcement Learning with Value-Targeted Regressioncited by 233
- 2019Provably Efficient Exploration in Policy Optimizationcited by 203
- 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
- 2020Learning Near Optimal Policies with Low Inherent Bellman Errorcited by 180
- 2020Reward-Free Exploration for Reinforcement Learningcited by 177
- 2019Is a Good Representation Sufficient for Sample Efficient Reinforcement Learning?cited by 140
29 more
- 2019Optimism in Reinforcement Learning with Generalized Linear Function Approximationcited by 111
- 2020Provably Efficient Reinforcement Learning for Discounted MDPs with Feature Mappingcited by 102
- 2019Sample Complexity of Reinforcement Learning using Linearly Combined Model Ensemblescited by 100
- 2019Frequentist Regret Bounds for Randomized Least-Squares Value Iterationcited by 100
- 2020Is Reinforcement Learning More Difficult Than Bandits? A Near-optimal Algorithm Escaping the Curse of Horizoncited by 99
- 2020Fast active learning for pure exploration in reinforcement learningcited by 92
- 2020On Reward-Free Reinforcement Learning with Linear Function Approximationcited by 89
- 2020Logarithmic Regret for Reinforcement Learning with Linear Function Approximationcited by 79
- 2020Adaptive Reward-Free Explorationcited by 77
- 2018On Oracle-Efficient PAC RL with Rich Observationscited by 71
- 2020Exponential Lower Bounds for Planning in MDPs With Linearly-Realizable Optimal Action-Value Functionscited by 71
- 2019Reinforcement Learning with Convex Constraintscited by 57
- 2022Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision Processescited by 57
- 2020Provably Efficient Reward-Agnostic Navigation with Linear Value Iterationcited by 56
- 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
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2022VOQL: Towards Optimal Regret in Model-free RL with Nonlinear Function Approximationcited by 44
- 2021Nearly Horizon-Free Offline Reinforcement Learningcited by 36
- 2020Is Long Horizon Reinforcement Learning More Difficult Than Short Horizon Reinforcement Learning?cited by 34
- 2021Stochastic Shortest Path: Minimax, Parameter-Free and Towards Horizon-Free Regretcited by 34
- 2021Reward-Free Model-Based Reinforcement Learning with Linear Function Approximationcited by 29
- 2020Model-Free Reinforcement Learning: from Clipped Pseudo-Regret to Sample Complexitycited by 28
- 2022Horizon-Free Reinforcement Learning in Polynomial Time: the Power of Stationary Policiescited by 27
- 2020Nearly Minimax Optimal Reward-free Reinforcement Learningcited by 26
- 2022Sample Efficient Deep Reinforcement Learning via Uncertainty Estimationcited by 26
- 2022Towards Deployment-Efficient Reinforcement Learning: Lower Bound and Optimalitycited by 24
- 2021Settling the Horizon-Dependence of Sample Complexity in Reinforcement Learningcited by 17
- 2021Near-Optimal Reward-Free Exploration for Linear Mixture MDPs with Plug-in Solvercited by 13
Built on by
Citing papers per year: 2023: 1; 2024: 3; 2025: 2; 2026: 1.
7 papers.
- 2024Traversing Pareto Optimal Policies: Provably Efficient Multi-Objective Reinforcement Learningcited by 13
- 2024Model-based RL as a Minimalist Approach to Horizon-Free and Second-Order Boundscited by 11
- 2023On the Model-Misspecification in Reinforcement Learningcited by 7
- 2025Replicable Reinforcement Learning with Linear Function Approximationcited by 6
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximationcited by 5
- 2025Rate optimal learning of equilibria from datacited by 2
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Costcited by 0
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost31 Aug 2026
- 2025Rate optimal learning of equilibria from data10 Oct 2025
- 2025Replicable Reinforcement Learning with Linear Function Approximation10 Sep 2025
- 2024Model-based RL as a Minimalist Approach to Horizon-Free and Second-Order Bounds16 Aug 2024
- 2024Traversing Pareto Optimal Policies: Provably Efficient Multi-Objective Reinforcement Learning24 Jul 2024
- 2024Uncertainty-Aware Reward-Free Exploration with General Function Approximation24 Jun 2024
- 2023On the Model-Misspecification in Reinforcement Learning19 Jun 2023