Settling the Sample Complexity of Model-Based Offline Reinforcement Learning
On the map: Offline RL and imitation learning from data → Offline RL algorithms
Abstract (arXiv)
This paper is concerned with offline reinforcement learning (RL), which learns using pre-collected data without further exploration. Effective offline RL would be able to accommodate distribution shift and limited data coverage. However, prior algorithms or analyses either suffer from suboptimal sample complexities or incur high burn-in cost to reach sample optimality, thus posing an impediment to efficient offline RL in sample-starved applications. We demonstrate that the model-based (or "plug-in") approach achieves minimax-optimal sample complexity without burn-in cost for tabular Markov decision processes (MDPs). Concretely, consider a finite-horizon (resp. γ-discounted infinite-horizon) MDP with S states and horizon H (resp. effective horizon frac11-γ), and suppose the distribution shift of data is reflected by some single-policy clipped concentrability coefficient Cstarclipped. We prove that model-based offline RL yields varepsilon-accuracy with a sample complexity of [ begincases fracH4SCclippedstarvarepsilon2 & (finite-horizon MDPs) fracSCclippedstar(1-γ)3varepsilon2 & (infinite-horizon MDPs) endcases ] up to log factor, which is minimax optimal for the entire varepsilon-range. The proposed algorithms are "pessimistic" variants of value iteration with Bernstein-style penalties, and do not require sophisticated variance reduction. Our analysis framework is established upon delicate leave-one-out decoupling arguments in conjunction with careful self-bounding techniques tailored to MDPs.
Key facts. Cited by 83 papers in the map (data through 14 Oct 2026). Placed in Offline RL algorithms, one of 13 directions in Offline RL and imitation learning from data. Builds on 2 of its 8 references that are in the map. First posted on arXiv 11 Apr 2022.
Builds on
2 of 8 references.
Closest ancestors · 1
Same direction, newest first
Other references · 1
Built on by
Citing papers per year: 2022: 15; 2023: 23; 2024: 13; 2025: 19; 2026: 13.
83 papers.
Sort:
By area:
- 2022Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexitycited by 87
- 2022Nearly Minimax Optimal Offline Reinforcement Learning with Linear Function Approximation: Single-Agent MDP and Markov Gamecited by 55
- 2023Provable Offline Preference-Based Reinforcement Learningcited by 55
- 2023The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Modelcited by 52
- 2023Double Pessimism is Provably Efficient for Distributionally Robust Offline Reinforcement Learning: Generic Algorithm and Robust Partial Coveragecited by 51
- 2023Settling the Sample Complexity of Online Reinforcement Learningcited by 45
- 2022Provably Efficient Offline Multi-agent Reinforcement Learning via Strategy-wise Bonuscited by 30
- 2023RLIF: Interactive Imitation Learning as Reinforcement Learningcited by 30
- 2022Optimal Conservative Offline RL with General Function Approximation via Augmented Lagrangiancited by 29
- 2023Sample Complexity of Variance-reduced Distributionally Robust Q-learningcited by 29
- 2023Towards Robust Offline Reinforcement Learning under Diverse Data Corruptioncited by 24
- 2026Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statisticcited by 21
38 more
- 2022Pessimism for Offline Linear Contextual Bandits using ℓp Confidence Setscited by 18
- 2023Offline Learning in Markov Games with General Function Approximationcited by 18
- 2023Soft Robust MDPs and Risk-Sensitive MDPs: Equivalence, Policy Gradient, and Sample Complexitycited by 17
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learningcited by 16
- 2023High-probability sample complexities for policy evaluation with linear function approximationcited by 16
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Modelscited by 16
- 2022Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Modelcited by 15
- 2022Offline Reinforcement Learning via Linear-Programming with Error-Bound Induced Constraintscited by 15
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 15
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processescited by 15
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processescited by 14
- 2022On Gap-dependent Bounds for Offline Reinforcement Learningcited by 13
- 2022Provably Efficient Offline Reinforcement Learning with Trajectory-Wise Rewardcited by 13
- 2023Provable Benefits of Policy Learning from Human Preferences in Contextual Bandit Problemscited by 13
- 2022Sample Complexity of Nonparametric Off-Policy Evaluation on Low-Dimensional Manifolds using Deep Networkscited by 12
- 2022Offline Reinforcement Learning with Differentiable Function Approximation is Provably Efficientcited by 12
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coveragecited by 12
- 2024Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Sufficescited by 11
- 2025Greedy Sampling Is Provably Efficient for RLHFcited by 11
- 2023Fast and Regret Optimal Best Arm Identification: Fundamental Limits and Low-Complexity Algorithmscited by 10
- 2025Towards a Sharp Analysis of Offline Policy Learning for f-Divergence-Regularized Contextual Banditscited by 10
- 2026Beyond Pessimism: Offline Learning in KL-regularized Gamescited by 10
- 2022Strategic Decision-Making in the Presence of Information Asymmetry: Provably Efficient RL with Algorithmic Instrumentscited by 9
- 2023Pessimistic Nonlinear Least-Squares Value Iteration for Offline Reinforcement Learningcited by 9
- 2024Learning from Sparse Offline Datasets via Conservative Density Estimationcited by 9
- 2025A Unified Theoretical Analysis of Private and Robust Offline Alignment: from RLHF to DPOcited by 8
- 2023A Survey on Offline Model-Based Reinforcement Learningcited by 7
- 2024Latent Safety-Constrained Policy Approach for Safe Offline Reinforcement Learningcited by 7
- 2023Optimal Sample Complexity of Stable Discounted Markov Decision Processescited by 6
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learningcited by 6
- 2024Hybrid Reinforcement Learning Breaks Sample Size Barriers in Linear MDPscited by 6
- 2025ORVIT: Near-Optimal Online Distributionally Robust Reinforcement Learningcited by 6
- 2026Offline Two-Player Zero-Sum Markov Games with KL Regularizationcited by 6
- 2022A Near-Optimal Primal-Dual Method for Off-Policy Learning in CMDPcited by 5
- 2022Offline Estimation of Controlled Markov Chains: Minimaxity and Sample Complexitycited by 5
- 2023Provably Efficient Offline Goal-Conditioned Reinforcement Learning with General Function Approximation and Single-Policy Concentrabilitycited by 5
- 2025Offline Learning for Combinatorial Multi-armed Banditscited by 5
- 2025Recursive Entropic Risk Optimization in Discounted MDPs: Sample Complexity Bounds with a Generative Modelcited by 5
50 most cited of 83 All 83 (JSON)
- 2026Near-Optimal Sample Complexity for Recursive Entropic Risk Reinforcement Learning with a Generative Model3 Oct 2026
- 2026Robust Q-learning for mean-field control under Wasserstein uncertainty in common noise18 Jun 2026
- 2026Counterfactually Safe Reinforcement Learning24 May 2026
- 2026On the Sample Complexity of Discounted Reinforcement Learning with Optimized Certainty Equivalents20 May 2026
- 2026Offline Two-Player Zero-Sum Markov Games with KL Regularization13 May 2026
- 2026Fast Rates for Offline Contextual Bandits with Forward-KL Regularization under Single-Policy Concentrability9 May 2026
- 2026Improved Model-based Reinforcement Learning with Smooth Kernels8 May 2026
- 2026Adaptive Estimation and Optimal Control in Offline Contextual MDPs without Stationarity5 May 2026
- 2026On the Optimal Sample Complexity of Offline Multi-Armed Bandits with KL Regularization4 May 2026
- 2026Pessimism-Free Offline Learning in General-Sum Games via KL Regularization30 Apr 2026
- 2026Beyond Pessimism: Offline Learning in KL-regularized Games8 Apr 2026
- 2026Demystifying Group Relative Policy Optimization: Its Policy Gradient is a U-Statistic1 Mar 2026
38 more
- 2026Improved Bounds for Reward-Agnostic and Reward-Free Exploration18 Feb 2026
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025Greedy Sampling Is Provably Efficient for RLHF28 Oct 2025
- 2025Learning Upper Lower Value Envelopes to Shape Online RL: A Principled Approach22 Oct 2025
- 2025Generalized Fitted Q-Iteration with Clustered Data4 Oct 2025
- 2025ORVIT: Near-Optimal Online Distributionally Robust Reinforcement Learning5 Aug 2025
- 2025Central Limit Theorems for Transition Probabilities of Controlled Markov Chains2 Aug 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Optimal Single-Policy Sample Complexity and Transient Coverage for Average-Reward Offline RL26 Jun 2025
- 2025Recursive Entropic Risk Optimization in Discounted MDPs: Sample Complexity Bounds with a Generative Model30 May 2025
- 2025Fully Offline Reinforcement Learning28 May 2025
- 2025STITCH-OPE: Trajectory Stitching with Guided Diffusion for Off-Policy Evaluation27 May 2025
- 2025A Unified Theoretical Analysis of Private and Robust Offline Alignment: from RLHF to DPO21 May 2025
- 2025Policy Testing in Markov Decision Processes21 May 2025
- 2025Augmenting Online RL with Offline Data is All You Need: A Unified Hybrid RL Algorithm Design and Analysis19 May 2025
- 2025Towards Optimal Offline Reinforcement Learning15 Mar 2025
- 2025Statistical Inference in Reinforcement Learning: A Selective Survey22 Feb 2025
- 2025Towards a Sharp Analysis of Offline Policy Learning for f-Divergence-Regularized Contextual Bandits9 Feb 2025
- 2025Dual Alignment Maximin Optimization for Offline Model-based RL2 Feb 2025
- 2025Offline Learning for Combinatorial Multi-armed Bandits31 Jan 2025
- 2024Latent Safety-Constrained Policy Approach for Safe Offline Reinforcement Learning11 Dec 2024
- 2024Tractable Offline Learning of Regular Decision Processes4 Sep 2024
- 2024Hybrid Reinforcement Learning Breaks Sample Size Barriers in Linear MDPs8 Aug 2024
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning10 Jul 2024
- 2024Primitive Agentic First-Order Optimization7 Jun 2024
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Models6 Jun 2024
- 2024A Fast Convergence Theory for Offline Decision Making3 Jun 2024
- 2024DyDiff: Long-Horizon Rollout via Dynamics Diffusion for Offline Reinforcement Learning29 May 2024
- 2024Offline RL via Feature-Occupancy Gradient Ascent22 May 2024
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processes19 Mar 2024
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancement24 Feb 2024
- 2024Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage Suffices8 Feb 2024
- 2024Learning from Sparse Offline Datasets via Conservative Density Estimation16 Jan 2024
- 2023RLIF: Interactive Imitation Learning as Reinforcement Learning21 Nov 2023
- 2023Bridging Distributionally Robust Learning and Offline RL: An Approach to Mitigate Distribution Shift and Partial Data Coverage27 Oct 2023
- 2023Towards Robust Offline Reinforcement Learning under Diverse Data Corruption19 Oct 2023
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processes13 Oct 2023
- 2023Pessimistic Nonlinear Least-Squares Value Iteration for Offline Reinforcement Learning2 Oct 2023
50 newest of 83 All 83 (JSON)
- 2022Distributionally Robust Model-Based Offline Reinforcement Learning with Near-Optimal Sample Complexitycited by 87
- 2022Nearly Minimax Optimal Offline Reinforcement Learning with Linear Function Approximation: Single-Agent MDP and Markov Gamecited by 55
- 2023Double Pessimism is Provably Efficient for Distributionally Robust Offline Reinforcement Learning: Generic Algorithm and Robust Partial Coveragecited by 51
- 2022Provably Efficient Offline Multi-agent Reinforcement Learning via Strategy-wise Bonuscited by 30
- 2023RLIF: Interactive Imitation Learning as Reinforcement Learningcited by 30
- 2022Optimal Conservative Offline RL with General Function Approximation via Augmented Lagrangiancited by 29
- 2023Towards Robust Offline Reinforcement Learning under Diverse Data Corruptioncited by 24
- 2023Offline Learning in Markov Games with General Function Approximationcited by 18
- 2023Reward-agnostic Fine-tuning: Provable Statistical Benefits of Hybrid Reinforcement Learningcited by 16
- 2022Offline Reinforcement Learning via Linear-Programming with Error-Bound Induced Constraintscited by 15
- 2023Minimax-Optimal Reward-Agnostic Exploration in Reinforcement Learningcited by 15
- 2024Sample Complexity of Offline Distributionally Robust Linear Markov Decision Processescited by 14
12 most cited of 38 All 38 (JSON)
- 2026Pessimism-Free Offline Learning in General-Sum Games via KL Regularization30 Apr 2026
- 2025Long-Horizon Model-Based Offline Reinforcement Learning Without Explicit Conservatism4 Dec 2025
- 2025Optimal Single-Policy Sample Complexity and Transient Coverage for Average-Reward Offline RL26 Jun 2025
- 2025Fully Offline Reinforcement Learning28 May 2025
- 2025Towards Optimal Offline Reinforcement Learning15 Mar 2025
- 2025Towards a Sharp Analysis of Offline Policy Learning for f-Divergence-Regularized Contextual Bandits9 Feb 2025
- 2025Dual Alignment Maximin Optimization for Offline Model-based RL2 Feb 2025
- 2024Latent Safety-Constrained Policy Approach for Safe Offline Reinforcement Learning11 Dec 2024
- 2024Tractable Offline Learning of Regular Decision Processes4 Sep 2024
- 2024Hybrid Reinforcement Learning Breaks Sample Size Barriers in Linear MDPs8 Aug 2024
- 2024Pessimism Meets Risk: Risk-Sensitive Offline Reinforcement Learning10 Jul 2024
- 2024A Fast Convergence Theory for Offline Decision Making3 Jun 2024
12 newest of 38 All 38 (JSON)
- 2025ORVIT: Near-Optimal Online Distributionally Robust Reinforcement Learningcited by 6
- 2025Recursive Entropic Risk Optimization in Discounted MDPs: Sample Complexity Bounds with a Generative Modelcited by 5
- 2026Fast Rates for Offline Contextual Bandits with Forward-KL Regularization under Single-Policy Concentrabilitycited by 3
- 2025Statistical Inference in Reinforcement Learning: A Selective Surveycited by 2
- 2025Statistical and Algorithmic Foundations of Reinforcement Learningcited by 2
- 2026On the Sample Complexity of Discounted Reinforcement Learning with Optimized Certainty Equivalentscited by 1
- 2025Generalized Fitted Q-Iteration with Clustered Datacited by 0
- 2025Learning Upper Lower Value Envelopes to Shape Online RL: A Principled Approachcited by 0
- 2026Improved Model-based Reinforcement Learning with Smooth Kernelscited by 0
- 2026Counterfactually Safe Reinforcement Learningcited by 0
- 2026Near-Optimal Sample Complexity for Recursive Entropic Risk Reinforcement Learning with a Generative Modelcited by 0
- 2026Near-Optimal Sample Complexity for Recursive Entropic Risk Reinforcement Learning with a Generative Model3 Oct 2026
- 2026Counterfactually Safe Reinforcement Learning24 May 2026
- 2026On the Sample Complexity of Discounted Reinforcement Learning with Optimized Certainty Equivalents20 May 2026
- 2026Fast Rates for Offline Contextual Bandits with Forward-KL Regularization under Single-Policy Concentrability9 May 2026
- 2026Improved Model-based Reinforcement Learning with Smooth Kernels8 May 2026
- 2025Learning Upper Lower Value Envelopes to Shape Online RL: A Principled Approach22 Oct 2025
- 2025Generalized Fitted Q-Iteration with Clustered Data4 Oct 2025
- 2025ORVIT: Near-Optimal Online Distributionally Robust Reinforcement Learning5 Aug 2025
- 2025Statistical and Algorithmic Foundations of Reinforcement Learning19 Jul 2025
- 2025Recursive Entropic Risk Optimization in Discounted MDPs: Sample Complexity Bounds with a Generative Model30 May 2025
- 2025Statistical Inference in Reinforcement Learning: A Selective Survey22 Feb 2025
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Modelscited by 16
- 2023Provable Benefits of Policy Learning from Human Preferences in Contextual Bandit Problemscited by 13
- 2025Greedy Sampling Is Provably Efficient for RLHFcited by 11
- 2025A Unified Theoretical Analysis of Private and Robust Offline Alignment: from RLHF to DPOcited by 8
- 2025Greedy Sampling Is Provably Efficient for RLHF28 Oct 2025
- 2025A Unified Theoretical Analysis of Private and Robust Offline Alignment: from RLHF to DPO21 May 2025
- 2024Self-Play with Adversarial Critic: Provable and Scalable Offline Alignment for Language Models6 Jun 2024
- 2023Provable Benefits of Policy Learning from Human Preferences in Contextual Bandit Problems24 Jul 2023
- 2024Primitive Agentic First-Order Optimizationcited by 0
- 2024Primitive Agentic First-Order Optimization7 Jun 2024
- 2023Provable Offline Preference-Based Reinforcement Learningcited by 55
- 2023The Curious Price of Distributional Robustness in Reinforcement Learning with a Generative Modelcited by 52
- 2023Settling the Sample Complexity of Online Reinforcement Learningcited by 45
- 2023Sample Complexity of Variance-reduced Distributionally Robust Q-learningcited by 29
- 2022Pessimism for Offline Linear Contextual Bandits using ℓp Confidence Setscited by 18
- 2023Soft Robust MDPs and Risk-Sensitive MDPs: Equivalence, Policy Gradient, and Sample Complexitycited by 17
- 2023High-probability sample complexities for policy evaluation with linear function approximationcited by 16
- 2022Minimax-Optimal Multi-Agent RL in Markov Games With a Generative Modelcited by 15
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processescited by 15
- 2022Sample Complexity of Nonparametric Off-Policy Evaluation on Low-Dimensional Manifolds using Deep Networkscited by 12
- 2023Fast and Regret Optimal Best Arm Identification: Fundamental Limits and Low-Complexity Algorithmscited by 10
- 2026Beyond Pessimism: Offline Learning in KL-regularized Gamescited by 10
12 most cited of 26 All 26 (JSON)
- 2026Robust Q-learning for mean-field control under Wasserstein uncertainty in common noise18 Jun 2026
- 2026Offline Two-Player Zero-Sum Markov Games with KL Regularization13 May 2026
- 2026Adaptive Estimation and Optimal Control in Offline Contextual MDPs without Stationarity5 May 2026
- 2026On the Optimal Sample Complexity of Offline Multi-Armed Bandits with KL Regularization4 May 2026
- 2026Beyond Pessimism: Offline Learning in KL-regularized Games8 Apr 2026
- 2026Improved Bounds for Reward-Agnostic and Reward-Free Exploration18 Feb 2026
- 2025Central Limit Theorems for Transition Probabilities of Controlled Markov Chains2 Aug 2025
- 2025Policy Testing in Markov Decision Processes21 May 2025
- 2025Augmenting Online RL with Offline Data is All You Need: A Unified Hybrid RL Algorithm Design and Analysis19 May 2025
- 2025Offline Learning for Combinatorial Multi-armed Bandits31 Jan 2025
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancement24 Feb 2024
- 2023Optimal Sample Complexity for Average Reward Markov Decision Processes13 Oct 2023
12 newest of 26 All 26 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024On Sample-Efficient Offline Reinforcement Learning: Data Diversity, Posterior Sampling, and Beyond
- 2022Offline Reinforcement Learning with Realizability and Single-policy Concentrability
- 2021Importance of Empirical Sample Complexity Analysis for Offline Reinforcement Learning
- 2021Nearly Horizon-Free Offline Reinforcement Learning
- 2023Offline Reinforcement Learning with Additional Covering Distributions
- 2020What are the Statistical Limits of Offline RL with Linear Function Approximation?
- 2026On the Complexity of Offline Reinforcement Learning with Q^star-Approximation and Partial Coverage
- 2023Offline Policy Evaluation for Reinforcement Learning with Adaptively Collected Data