Statistically Efficient Off-Policy Policy Gradients
Nathan Kallus, Masatoshi Uehara
Abstract
Policy gradient methods in reinforcement learning update policy parameters by taking steps in the direction of an estimated gradient of policy value. In this paper, we consider the statistically efficient estimation of policy gradients from off-policy data, where the estimation is particularly non-trivial. We derive the asymptotic lower bound on the feasible mean-squared error in both Markov and non-Markov decision processes and show that existing estimators fail to achieve it in general settings. We propose a meta-algorithm that achieves the lower bound without any parametric assumptions and exhibits a unique 3-way double robustness property. We discuss how to estimate nuisances that the algorithm relies on. Finally, we establish guarantees on the rate at which we approach a stationary point when we take steps in the direction of our new estimated policy gradient.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 3d6c4ac5-87d4-4dca-a08f-d848f0f1e912Cited by top-tier papers14
- ExGRPO: Learning to Reason from ExperienceRunzhe Zhan, Yafu Li, Zhi Wang, Xiaoye Qu et al.ICLR 2026 · 51 citations
- Doubly Robust Distributionally Robust Off-Policy Evaluation and LearningNathan Kallus, Xiaojie Mao, Kaiwen Wang, Zhengyuan ZhouICML 2022 · 39 citations
- Doubly Robust Off-Policy Actor-Critic: Convergence and OptimalityTengyu Xu, Zhuoran Yang, Zhaoran Wang, Yingbin LiangICML 2021 · 31 citations
- Doubly Robust Off-Policy Value and Gradient Estimation for Deterministic PoliciesNathan Kallus, Masatoshi UeharaNeurIPS 2020 · 16 citations
- Doubly Robust Alignment for Large Language ModelsErhan Xu, Kai Ye, Hongyi Zhou, Luhan Zhu et al.NeurIPS 2025 · 14 citations
Builds on1
Related papers
- Optimal Estimation of Policy Gradient via Double Fitted IterationChengzhuo Ni, Ruiqi Zhang, Xiang Ji, Xuezhou Zhang et al.ICML 2022 · 1 citation
- Occupancy-based Policy Gradient: Estimation, Convergence, and OptimalityAudrey Huang, Nan JiangNeurIPS 2024 · 5 citations
- A Policy Gradient Method for Confounded POMDPsMao Hong, Zhengling Qi, Yanxun XuICLR 2024 · 5 citations
- Robust On-Policy Sampling for Data-Efficient Policy Evaluation in Reinforcement LearningRujie Zhong, Duohan Zhang, Lukas Schäfer, Stefano V. Albrecht et al.NeurIPS 2022 · 19 citations
- Sample Efficient Reinforcement Learning with REINFORCEJunzi Zhang, Jongho Kim, Brendan O'Donoghue, Stephen P. BoydAAAI 2021 · 162 citations
