Model-Based Reinforcement Learning with Value-Targeted Regression
Not on the map. Why?
Abstract (arXiv)
This paper studies model-based reinforcement learning (RL) for regret minimization. We focus on finite-horizon episodic RL where the transition model P belongs to a known family of models P, a special case of which is when models in P take the form of linear mixtures: Pθ = sumi=1d θiPi. We propose a model based RL algorithm that is based on optimism principle: In each episode, the set of models that are `consistent' with the data collected is constructed. The criterion of consistency is based on the total squared error of that the model incurs on the task of predicting values as determined by the last value estimate along the transitions. The next value function is then chosen by solving the optimistic planning problem with the constructed set of models. We derive a bound on the regret, which, in the special case of linear mixtures, the regret bound takes the form tildeO(d√H3T), where H, T and d are the horizon, total number of steps and dimension of θ, respectively. In particular, this regret bound is independent of the total number of states or actions, and is close to a lower bound Omega(√HdT). For a general model family P, the regret bound is derived using the notion of the so-called Eluder dimension proposed by Russo & Van Roy (2014).
Key facts. Cited by 233 papers in the map (data through 7 Oct 2026). First posted on arXiv 1 Jun 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 56; 2022: 57; 2023: 59; 2024: 32; 2025: 13; 2026: 15.
233 papers.
Sort:
By area:
- 2020Is Pessimism Provably Efficient for Offline RL?cited by 362
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 160
- 2021Provably Efficient Reinforcement Learning with Linear Function Approximation Under Adaptivity Constraintscited by 124
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Human-in-the-loop: Provably Efficient Preference-based Reinforcement Learning with General Function Approximationcited by 80
- 2025Outcome-based Exploration for LLM Reasoningcited by 79
- 2021Understanding Domain Randomization for Sim-to-real Transfercited by 69
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 63
- 2024Active Preference Optimization for Sample Efficient RLHFcited by 61
- 2022Unpacking Reward Shaping: Understanding the Benefits of Reward Engineering on Sample Complexitycited by 59
38 more
- 2022Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision Processescited by 57
- 2021Cautiously Optimistic Policy Optimization and Exploration with Linear Function Approximationcited by 53
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
- 2021Randomized Exploration for Reinforcement Learning with General Value Function Approximationcited by 49
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
- 2023Reason for Future, Act for Now: A Principled Framework for Autonomous LLM Agents with Provable Sample Efficiencycited by 46
- 2021Improved Variance-Aware Confidence Sets for Linear Bandits and Linear Mixture MDPcited by 45
- 2021Near-optimal Representation Learning for Linear Bandits and Linear RLcited by 45
- 2022Leveraging Offline Data in Online Reinforcement Learningcited by 42
- 2021An Exponential Lower Bound for Linearly-Realizable MDPs with Constant Suboptimality Gapcited by 40
- 2021MADE: Exploration via Maximizing Deviation from Explored Regionscited by 40
- 2023Variance-Dependent Regret Bounds for Linear Bandits and Reinforcement Learning: Adaptivity and Computational Efficiencycited by 39
- 2021A Provably Efficient Algorithm for Linear Markov Decision Process with Low Switching Costcited by 38
- 2021First-Order Regret in Reinforcement Learning with Linear Function Approximation: A Robust Estimation Approachcited by 38
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2021Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiationcited by 35
- 2021Deep reinforcement learning in medical imaging: A literature reviewcited by 34
- 2021Variance-Aware Off-Policy Evaluation with Linear Function Approximationcited by 34
- 2021Proper Value Equivalencecited by 33
- 2022Instance-Dependent Near-Optimal Policy Identification in Linear MDPs via Online Experiment Designcited by 33
- 2021Provable Model-based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvaturecited by 32
- 2022On the Complexity of Adversarial Decision Makingcited by 32
- 2022Value Gradient weighted Model-Based Reinforcement Learningcited by 31
- 2022Model-based RL with Optimistic Posterior Sampling: Structural Conditions and Sample Complexitycited by 31
- 2022A General Framework for Sample-Efficient Function Approximation in Reinforcement Learningcited by 31
- 2021Robust Policy Gradient against Strong Data Corruptioncited by 30
- 2022Nearly Minimax Optimal Reinforcement Learning with Linear Function Approximationcited by 30
- 2021Reward-Free Model-Based Reinforcement Learning with Linear Function Approximationcited by 29
- 2022Provably Efficient Model-Free Constrained RL with Linear Function Approximationcited by 29
- 2023Tight Guarantees for Interactive Decision Making with the Decision-Estimation Coefficientcited by 29
- 2023Provable and Practical: Efficient Exploration in Reinforcement Learning via Langevin Monte Carlocited by 29
- 2021Near-optimal Policy Optimization Algorithms for Learning Adversarial Linear Mixture MDPscited by 28
- 2023A Theoretical Analysis of Optimistic Proximal Policy Optimization in Linear Markov Decision Processescited by 28
- 2021Sample-Efficient Reinforcement Learning Is Feasible for Linearly Realizable MDPs with Limited Revisitingcited by 27
- 2022Contrastive UCB: Provably Efficient Contrastive Self-Supervised Learning in Online Reinforcement Learningcited by 26
- 2021Almost Optimal Algorithms for Two-player Zero-Sum Linear Mixture Markov Gamescited by 25
- 2022Proximal Point Imitation Learningcited by 25
- 2021Learning Representations for Pixel-based Control: What Matters and Why?cited by 24
50 most cited of 233 All 233 (JSON)
- 2026Minimax-Optimality of Posterior Sampling for Reinforcement Learning27 Sep 2026
- 2026Mini-Batch Risk-Averse Deep Q-Learning: A Robot Navigation Case Study7 Sep 2026
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost31 Aug 2026
- 2026Minimax-Optimal Policy Regret in Partially Observable Markov Games1 Jun 2026
- 2026Theoretical Foundations and Effective Algorithms for Policy-Aware Simulator Learning27 May 2026
- 2026Improved Model-based Reinforcement Learning with Smooth Kernels8 May 2026
- 2026Robust Parameter Learning for Uncertain MDPs2 May 2026
- 2026Reinforcement Learning with Markov Risk Measures and Multipattern Risk Approximation1 May 2026
- 2026When Can You Poison Rewards? A Tight Characterization of Reward Poisoning in Linear MDPs11 Apr 2026
- 2026Near-Optimal Primal-Dual Algorithm for Learning Linear Mixture CMDPs with Adversarial Rewards29 Mar 2026
- 2026Regret Bounds for Reinforcement Learning from Multi-Source Imperfect Preferences20 Mar 2026
- 2026Learning Approximate Nash Equilibria in Cooperative Multi-Agent Reinforcement Learning via Mean-Field Subsampling4 Mar 2026
38 more
- 2026Is Pure Exploitation Sufficient in Exogenous MDPs with Linear Function Approximation?28 Jan 2026
- 2026Eluder dimension: localise it!14 Jan 2026
- 2026Revisiting Weighted Strategy for Non-stationary Parametric Bandits and MDPs3 Jan 2026
- 2025Homomorphic Mappings for Value-Preserving State Aggregation in Markov Decision Processes11 Oct 2025
- 2025Rectifying Regression in Reinforcement Learning1 Oct 2025
- 2025Replicable Reinforcement Learning with Linear Function Approximation10 Sep 2025
- 2025Outcome-based Exploration for LLM Reasoning8 Sep 2025
- 2025The Sample Complexity of Online Strategic Decision Making with Information Asymmetry and Knowledge Transportability11 Jun 2025
- 2025Pointwise confidence estimation in the non-linear ℓ2-regularized least squares8 Jun 2025
- 2025Calibrated Value-Aware Model Learning with Probabilistic Environment Models28 May 2025
- 2025Linear Mixture Distributionally Robust Markov Decision Processes23 May 2025
- 2025Towards Optimal Differentially Private Regret Bounds in Linear MDPs12 Apr 2025
- 2025Dynamic Assortment Selection and Pricing with Censored Preference Feedback3 Apr 2025
- 2025Incentivize without Bonus: Provably Efficient Model-based Online Multi-agent RL for Markov Games13 Feb 2025
- 2025Best Policy Learning from Trajectory Preference Feedback31 Jan 2025
- 2025Digital Twin Calibration with Model-Based Reinforcement Learning4 Jan 2025
- 2024Near-Optimal Dynamic Regret for Adversarial Linear Mixture MDPs5 Nov 2024
- 2024Demystifying Linear MDPs and Novel Dynamics Aggregation Framework31 Oct 2024
- 2024RL-STaR: Theoretical Analysis of Reinforcement Learning Frameworks for Self-Taught Reasoner31 Oct 2024
- 2024RA-PbRL: Provably Efficient Risk-Aware Preference-Based Reinforcement Learning31 Oct 2024
- 2024Learning Infinite-Horizon Average-Reward Linear Mixture MDPs of Bounded Span19 Oct 2024
- 2024Exploiting Exogenous Structure for Sample-Efficient Reinforcement Learning22 Sep 2024
- 2024Bellman Unbiasedness: Toward Provably Efficient Distributional Reinforcement Learning with General Value Function Approximation31 Jul 2024
- 2024Random Latent Exploration for Deep Reinforcement Learning18 Jul 2024
- 2024Misspecified Q-Learning with Sparse Linear Function Approximation: Tight Bounds on Approximation Error18 Jul 2024
- 2024Warm-up Free Policy Optimization: Improved Regret in Linear Markov Decision Processes3 Jul 2024
- 2024Operator World Models for Reinforcement Learning28 Jun 2024
- 2024Meta-Gradient Search Control: A Method for Improving the Efficiency of Dyna-style Planning27 Jun 2024
- 2024A New View on Planning in Online Reinforcement Learning3 Jun 2024
- 2024Policy Zooming: Adaptive Discretization-based Infinite-Horizon Average-Reward Reinforcement Learning29 May 2024
- 2024Provably Efficient Reinforcement Learning with Multinomial Logit Function Approximation27 May 2024
- 2024Reinforcement Learning for Infinite-Horizon Average-Reward Linear MDPs via Approximation by Discounted-Reward MDPs23 May 2024
- 2024Sample-efficient Learning of Infinite-horizon Average-reward MDPs with General Function Approximation19 Apr 2024
- 2024Achieving Constant Regret in Linear Markov Decision Processes16 Apr 2024
- 2024Differentially Private Reinforcement Learning with Self-Play11 Apr 2024
- 2024Distributionally Robust Reinforcement Learning with Interactive Data Collection: Fundamental Hardness and Near-Optimal Algorithms4 Apr 2024
- 2024Utilizing Maximum Mean Discrepancy Barycenter for Propagating the Uncertainty of Value Functions in Reinforcement Learning31 Mar 2024
- 2024Prior-dependent analysis of posterior sampling reinforcement learning with function approximation17 Mar 2024
50 newest of 233 All 233 (JSON)
- 2020Is Pessimism Provably Efficient for Offline RL?cited by 362
- 2021Pessimistic Model-based Offline Reinforcement Learning under Partial Coveragecited by 160
- 2022Near-optimal Offline Reinforcement Learning with Linear Representation: Leveraging Variance Information with Pessimismcited by 63
- 2022Leveraging Offline Data in Online Reinforcement Learningcited by 42
- 2022Proximal Point Imitation Learningcited by 25
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximationcited by 22
- 2022Offline Reinforcement Learning with Differential Privacycited by 19
- 2021Optimal Uniform OPE and Model-based Offline Reinforcement Learning in Time-Homogeneous, Reward-Free and Task-Agnostic Settingscited by 14
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 14
- 2024Offline Multi-task Transfer RL with Representational Penalizationcited by 13
- 2021Provably Efficient Representation Selection in Low-rank Markov Decision Processes: From Online to Offline RLcited by 12
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
12 most cited of 18 All 18 (JSON)
- 2024Offline Multi-task Transfer RL with Representational Penalization19 Feb 2024
- 2023Reward-Consistent Dynamics Models are Strongly Generalizable for Offline Reinforcement Learning9 Oct 2023
- 2023Pessimistic Nonlinear Least-Squares Value Iteration for Offline Reinforcement Learning2 Oct 2023
- 2023Bayesian Inverse Transition Learning for Offline Settings9 Aug 2023
- 2023What can online reinforcement learning with function approximation benefit from general coverage conditions?25 Apr 2023
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learning14 Apr 2023
- 2023The Provable Benefits of Unsupervised Data Sharing for Offline Reinforcement Learning27 Feb 2023
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximation23 Nov 2022
- 2022Leveraging Offline Data in Online Reinforcement Learning9 Nov 2022
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficient3 Oct 2022
- 2022Proximal Point Imitation Learning22 Sep 2022
- 2022Offline Reinforcement Learning with Differential Privacy2 Jun 2022
12 newest of 18 All 18 (JSON)
- 2024Active Preference Optimization for Sample Efficient RLHFcited by 61
- 2023Provably Efficient Iterated CVaR Reinforcement Learning with Function Approximation and Human Feedbackcited by 13
- 2024RA-PbRL: Provably Efficient Risk-Aware Preference-Based Reinforcement Learningcited by 8
- 2025Best Policy Learning from Trajectory Preference Feedbackcited by 3
- 2026Regret Bounds for Reinforcement Learning from Multi-Source Imperfect Preferencescited by 1
- 2026Regret Bounds for Reinforcement Learning from Multi-Source Imperfect Preferences20 Mar 2026
- 2025Best Policy Learning from Trajectory Preference Feedback31 Jan 2025
- 2024RA-PbRL: Provably Efficient Risk-Aware Preference-Based Reinforcement Learning31 Oct 2024
- 2024Active Preference Optimization for Sample Efficient RLHF16 Feb 2024
- 2023Provably Efficient Iterated CVaR Reinforcement Learning with Function Approximation and Human Feedback6 Jul 2023
- 2021Bilinear Classes: A Structural Framework for Provable Generalization in RLcited by 176
- 2021Provably Efficient Reinforcement Learning with Linear Function Approximation Under Adaptivity Constraintscited by 124
- 2022When Is Partially Observable Reinforcement Learning Not Scary?cited by 100
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2022Human-in-the-loop: Provably Efficient Preference-based Reinforcement Learning with General Function Approximationcited by 80
- 2021Understanding Domain Randomization for Sim-to-real Transfercited by 69
- 2022Unpacking Reward Shaping: Understanding the Benefits of Reward Engineering on Sample Complexitycited by 59
- 2022Nearly Minimax Optimal Reinforcement Learning for Linear Markov Decision Processescited by 57
- 2021Cautiously Optimistic Policy Optimization and Exploration with Linear Function Approximationcited by 53
- 2022Computationally Efficient Horizon-Free Reinforcement Learning for Linear Mixture MDPscited by 52
- 2021Randomized Exploration for Reinforcement Learning with General Value Function Approximationcited by 49
- 2022Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision Processescited by 49
12 most cited of 202 All 202 (JSON)
- 2026Minimax-Optimality of Posterior Sampling for Reinforcement Learning27 Sep 2026
- 2026Mini-Batch Risk-Averse Deep Q-Learning: A Robot Navigation Case Study7 Sep 2026
- 2026Provably Efficient Federated Reinforcement Learning with Linear Function Approximation and Logarithmic Communication Cost31 Aug 2026
- 2026Minimax-Optimal Policy Regret in Partially Observable Markov Games1 Jun 2026
- 2026Improved Model-based Reinforcement Learning with Smooth Kernels8 May 2026
- 2026Robust Parameter Learning for Uncertain MDPs2 May 2026
- 2026Reinforcement Learning with Markov Risk Measures and Multipattern Risk Approximation1 May 2026
- 2026When Can You Poison Rewards? A Tight Characterization of Reward Poisoning in Linear MDPs11 Apr 2026
- 2026Near-Optimal Primal-Dual Algorithm for Learning Linear Mixture CMDPs with Adversarial Rewards29 Mar 2026
- 2026Learning Approximate Nash Equilibria in Cooperative Multi-Agent Reinforcement Learning via Mean-Field Subsampling4 Mar 2026
- 2026Is Pure Exploitation Sufficient in Exogenous MDPs with Linear Function Approximation?28 Jan 2026
- 2026Eluder dimension: localise it!14 Jan 2026
12 newest of 202 All 202 (JSON)