Off-Policy Policy Gradient with State Distribution Correction
Not on the map. Why?
Abstract (arXiv)
We study the problem of off-policy policy optimization in Markov decision processes, and develop a novel off-policy policy gradient method. Prior off-policy policy gradient approaches have generally ignored the mismatch between the distribution of states visited under the behavior policy used to collect data, and what would be the distribution of states under the learned policy. Here we build on recent progress for estimating the ratio of the state distributions under behavior and evaluation policies for policy evaluation, and present an off-policy policy gradient optimization technique that can account for this mismatch in distributions. We present an illustrative example of why this is important and a theoretical convergence guarantee for our approach. Empirically, we compare our method in simulations to several strong baselines which do not correct for this mismatch, significantly improving in the quality of the policy discovered.
Key facts. Cited by 43 papers in the map (data through 7 Oct 2026). First posted on arXiv 17 Apr 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 16; 2022: 12; 2023: 8; 2024: 3; 2025: 2; 2026: 2.
43 papers.
Sort:
By area:
- 2022When Should We Prefer Offline Reinforcement Learning Over Behavioral Cloning?cited by 86
- 2021Towards Instance-Optimal Offline Reinforcement Learning with Pessimismcited by 78
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Offline Reinforcement Learning with Reverse Model-based Imaginationcited by 47
- 2021Nearly Horizon-Free Offline Reinforcement Learningcited by 36
- 2021Towards Hyperparameter-free Policy Selection for Offline Reinforcement Learningcited by 32
- 2023Adversarial Model for Offline Reinforcement Learningcited by 31
- 2021Mean-Field Multi-Agent Reinforcement Learning: A Decentralized Network Approachcited by 27
- 2021A Generalized Projected Bellman Error for Off-policy Value Estimation in Reinforcement Learningcited by 26
- 2021Finite-Sample Analysis of Off-Policy Natural Actor-Critic with Linear Function Approximationcited by 22
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximationcited by 22
- 2021On the Convergence and Optimality of Policy Gradient for Markov Coherent Riskcited by 19
31 more
- 2023A General Offline Reinforcement Learning Framework for Interactive Recommendationcited by 18
- 2021Global Optimality and Finite Sample Analysis of Softmax Off-Policy Actor Critic under State Distribution Mismatchcited by 17
- 2023B-Coder: Value-Based Deep Reinforcement Learning for Program Synthesiscited by 16
- 2022Model-Based Offline Meta-Reinforcement Learning with Regularizationcited by 14
- 2022Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Datacited by 12
- 2023Reinforcement Learning Tutor Better Supported Lower Performers in a Math Taskcited by 11
- 2021Characterizing the Gap Between Actor-Critic and Policy Gradientcited by 9
- 2022Continual Auxiliary Task Learningcited by 9
- 2022Offline Policy Optimization with Eligible Actionscited by 7
- 2023A Policy Gradient Method for Confounded POMDPscited by 6
- 2024On Sample-Efficient Offline Reinforcement Learning: Data Diversity, Posterior Sampling, and Beyondcited by 6
- 2021Learning Expected Emphatic Traces for Deep RLcited by 5
- 2021Dr Jekyll and Mr Hyde: the Strange Case of Off-Policy Policy Updatescited by 5
- 2021Off-Policy Actor-Critic with Emphatic Weightingscited by 5
- 2021On the Convergence Rate of Off-Policy Policy Optimization Methods with Density-Ratio Correctioncited by 4
- 2022A Temporal-Difference Approach to Policy Gradient Estimationcited by 4
- 2023Budgeting Counterfactual for Offline RLcited by 4
- 2021The Curse of Passive Data Collection in Batch Reinforcement Learningcited by 2
- 2022Behavior Estimation from Multi-Source Data for Offline Reinforcement Learningcited by 2
- 2025On The Statistical Complexity of Offline Decision-Makingcited by 2
- 2021Neural Network Compatible Off-Policy Natural Actor-Critic Algorithmcited by 1
- 2022Optimal Estimation of Off-Policy Policy Gradient via Double Fitted Iterationcited by 1
- 2022Variance Reduction based Partial Trajectory Reuse to Accelerate Policy Gradient Optimizationcited by 1
- 2022Coordinate Ascent for Off-Policy RL with Global Convergence Guaranteescited by 1
- 2024Causal prompting model-based offline reinforcement learningcited by 1
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasetscited by 1
- 2025Online Optimization for Offline Safe Reinforcement Learningcited by 1
- 2026Experience Sharing in Mutual Reinforcement Learning for Heterogeneous Language Modelscited by 1
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learningcited by 1
- 2023Risk-Sensitive and Robust Model-Based Reinforcement Learning and Planningcited by 0
- 2023Reward Dropout Improves Control: Bi-objective Perspective on Reinforced LMcited by 0
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning2 Aug 2026
- 2026Experience Sharing in Mutual Reinforcement Learning for Heterogeneous Language Models8 May 2026
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
- 2025On The Statistical Complexity of Offline Decision-Making10 Jan 2025
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024Causal prompting model-based offline reinforcement learning3 Jun 2024
- 2024On Sample-Efficient Offline Reinforcement Learning: Data Diversity, Posterior Sampling, and Beyond6 Jan 2024
- 2023Reward Dropout Improves Control: Bi-objective Perspective on Reinforced LM6 Oct 2023
- 2023B-Coder: Value-Based Deep Reinforcement Learning for Program Synthesis4 Oct 2023
- 2023A General Offline Reinforcement Learning Framework for Interactive Recommendation1 Oct 2023
- 2023Budgeting Counterfactual for Offline RL12 Jul 2023
- 2023A Policy Gradient Method for Confounded POMDPs26 May 2023
31 more
- 2023Reinforcement Learning Tutor Better Supported Lower Performers in a Math Task11 Apr 2023
- 2023Risk-Sensitive and Robust Model-Based Reinforcement Learning and Planning2 Apr 2023
- 2023Adversarial Model for Offline Reinforcement Learning21 Feb 2023
- 2022Coordinate Ascent for Off-Policy RL with Global Convergence Guarantees10 Dec 2022
- 2022Behavior Estimation from Multi-Source Data for Offline Reinforcement Learning29 Nov 2022
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximation23 Nov 2022
- 2022Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Data16 Oct 2022
- 2022Offline Policy Optimization with Eligible Actions1 Jul 2022
- 2022Variance Reduction based Partial Trajectory Reuse to Accelerate Policy Gradient Optimization6 May 2022
- 2022When Should We Prefer Offline Reinforcement Learning Over Behavioral Cloning?12 Apr 2022
- 2022Continual Auxiliary Task Learning22 Feb 2022
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddings13 Feb 2022
- 2022Model-Based Offline Meta-Reinforcement Learning with Regularization7 Feb 2022
- 2022A Temporal-Difference Approach to Policy Gradient Estimation4 Feb 2022
- 2022Optimal Estimation of Off-Policy Policy Gradient via Double Fitted Iteration31 Jan 2022
- 2021Off-Policy Actor-Critic with Emphatic Weightings16 Nov 2021
- 2021Global Optimality and Finite Sample Analysis of Softmax Off-Policy Actor Critic under State Distribution Mismatch4 Nov 2021
- 2021Towards Hyperparameter-free Policy Selection for Offline Reinforcement Learning26 Oct 2021
- 2021Neural Network Compatible Off-Policy Natural Actor-Critic Algorithm19 Oct 2021
- 2021Towards Instance-Optimal Offline Reinforcement Learning with Pessimism17 Oct 2021
- 2021Offline Reinforcement Learning with Reverse Model-based Imagination1 Oct 2021
- 2021Dr Jekyll and Mr Hyde: the Strange Case of Off-Policy Policy Updates29 Sep 2021
- 2021Mean-Field Multi-Agent Reinforcement Learning: A Decentralized Network Approach5 Aug 2021
- 2021Learning Expected Emphatic Traces for Deep RL12 Jul 2021
- 2021The Curse of Passive Data Collection in Batch Reinforcement Learning18 Jun 2021
- 2021Characterizing the Gap Between Actor-Critic and Policy Gradient13 Jun 2021
- 2021On the Convergence Rate of Off-Policy Policy Optimization Methods with Density-Ratio Correction2 Jun 2021
- 2021Finite-Sample Analysis of Off-Policy Natural Actor-Critic with Linear Function Approximation26 May 2021
- 2021A Generalized Projected Bellman Error for Off-policy Value Estimation in Reinforcement Learning28 Apr 2021
- 2021Nearly Horizon-Free Offline Reinforcement Learning25 Mar 2021
- 2021On the Convergence and Optimality of Policy Gradient for Markov Coherent Risk4 Mar 2021
- 2022When Should We Prefer Offline Reinforcement Learning Over Behavioral Cloning?cited by 86
- 2021Towards Instance-Optimal Offline Reinforcement Learning with Pessimismcited by 78
- 2021Offline Reinforcement Learning with Reverse Model-based Imaginationcited by 47
- 2021Nearly Horizon-Free Offline Reinforcement Learningcited by 36
- 2021Towards Hyperparameter-free Policy Selection for Offline Reinforcement Learningcited by 32
- 2023Adversarial Model for Offline Reinforcement Learningcited by 31
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximationcited by 22
- 2023A General Offline Reinforcement Learning Framework for Interactive Recommendationcited by 18
- 2022Model-Based Offline Meta-Reinforcement Learning with Regularizationcited by 14
- 2022Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Datacited by 12
- 2022Offline Policy Optimization with Eligible Actionscited by 7
- 2024On Sample-Efficient Offline Reinforcement Learning: Data Diversity, Posterior Sampling, and Beyondcited by 6
12 most cited of 17 All 17 (JSON)
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
- 2024SeMOPO: Learning High-quality Model and Policy from Low-quality Offline Visual Datasets13 Jun 2024
- 2024Causal prompting model-based offline reinforcement learning3 Jun 2024
- 2024On Sample-Efficient Offline Reinforcement Learning: Data Diversity, Posterior Sampling, and Beyond6 Jan 2024
- 2023A General Offline Reinforcement Learning Framework for Interactive Recommendation1 Oct 2023
- 2023Budgeting Counterfactual for Offline RL12 Jul 2023
- 2023Adversarial Model for Offline Reinforcement Learning21 Feb 2023
- 2022Behavior Estimation from Multi-Source Data for Offline Reinforcement Learning29 Nov 2022
- 2022On Instance-Dependent Bounds for Offline Reinforcement Learning with Linear Function Approximation23 Nov 2022
- 2022Data-Efficient Pipeline for Offline Reinforcement Learning with Limited Data16 Oct 2022
- 2022Offline Policy Optimization with Eligible Actions1 Jul 2022
- 2022When Should We Prefer Offline Reinforcement Learning Over Behavioral Cloning?12 Apr 2022
12 newest of 17 All 17 (JSON)
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddingscited by 53
- 2021Mean-Field Multi-Agent Reinforcement Learning: A Decentralized Network Approachcited by 27
- 2021A Generalized Projected Bellman Error for Off-policy Value Estimation in Reinforcement Learningcited by 26
- 2021Finite-Sample Analysis of Off-Policy Natural Actor-Critic with Linear Function Approximationcited by 22
- 2021On the Convergence and Optimality of Policy Gradient for Markov Coherent Riskcited by 19
- 2021Global Optimality and Finite Sample Analysis of Softmax Off-Policy Actor Critic under State Distribution Mismatchcited by 17
- 2023B-Coder: Value-Based Deep Reinforcement Learning for Program Synthesiscited by 16
- 2023Reinforcement Learning Tutor Better Supported Lower Performers in a Math Taskcited by 11
- 2021Characterizing the Gap Between Actor-Critic and Policy Gradientcited by 9
- 2022Continual Auxiliary Task Learningcited by 9
- 2023A Policy Gradient Method for Confounded POMDPscited by 6
- 2021Learning Expected Emphatic Traces for Deep RLcited by 5
12 most cited of 26 All 26 (JSON)
- 2026Reusing Rollouts under Policy Lag: Prefix-Normalized Policy Optimization for LLM Reinforcement Learning2 Aug 2026
- 2026Experience Sharing in Mutual Reinforcement Learning for Heterogeneous Language Models8 May 2026
- 2025On The Statistical Complexity of Offline Decision-Making10 Jan 2025
- 2023Reward Dropout Improves Control: Bi-objective Perspective on Reinforced LM6 Oct 2023
- 2023B-Coder: Value-Based Deep Reinforcement Learning for Program Synthesis4 Oct 2023
- 2023A Policy Gradient Method for Confounded POMDPs26 May 2023
- 2023Reinforcement Learning Tutor Better Supported Lower Performers in a Math Task11 Apr 2023
- 2023Risk-Sensitive and Robust Model-Based Reinforcement Learning and Planning2 Apr 2023
- 2022Coordinate Ascent for Off-Policy RL with Global Convergence Guarantees10 Dec 2022
- 2022Variance Reduction based Partial Trajectory Reuse to Accelerate Policy Gradient Optimization6 May 2022
- 2022Continual Auxiliary Task Learning22 Feb 2022
- 2022Off-Policy Evaluation for Large Action Spaces via Embeddings13 Feb 2022
12 newest of 26 All 26 (JSON)