Offline Minimax Soft-Q-learning Under Realizability and Partial Coverage
Masatoshi Uehara, Nathan Kallus, Jason D. Lee, Wen Sun
Abstract
In offline RL, we have no opportunity to explore so we must make assumptions that the data is sufficient to guide picking a good policy, and we want to make these assumptions as harmless as possible. In this work, we propose value-based algorithms for offline RL with PAC guarantees under just partial coverage, specifically, coverage of just a single comparator policy, and realizability of the soft (entropy-regularized) Qfunction of the single policy and a related function defined as a saddle point of certain minimax optimization problem. This offers refined and generally more lax conditions for offline RL. We further show an analogous result for vanilla Q-functions under a soft margin condition. To attain these guarantees, we leverage novel minimax learning algorithms and analyses to accurately estimate either soft or vanilla Q-functions with strong L 2 -convergence guarantees. Our algorithms' loss functions arise from casting the estimation problems as nonlinear convex optimization problems and Lagrangifying. Surprisingly we handle partial coverage even without explicitly enforcing pessimism.
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 b00c3da9-d485-486c-afe4-4d3a93349a1fCited by top-tier papers5
- Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-constraintWei Xiong, Hanze Dong, Chenlu Ye, Ziqi Wang et al.ICML 2024 · 346 citations
- Q#: Provably Optimal Distributional RL for LLM Post-TrainingJin Peng Zhou, Kaiwen Wang, Jonathan D. Chang, Zhaolin Gao et al.NeurIPS 2025 · 18 citations
- Federated Offline Reinforcement Learning: Collaborative Single-Policy Coverage SufficesJiin Woo, Laixi Shi, Gauri Joshi, Yuejie ChiICML 2024 · 9 citations
- Adversarially Trained Weighted Actor-Critic for Safe Offline Reinforcement LearningHonghao Wei, Xiyue Peng, Arnob Ghosh, Xin LiuNeurIPS 2024 · 4 citations
- Worst-Case Offline Reinforcement Learning with Arbitrary Data SupportKohei MiyaguchiNeurIPS 2024
Builds on22
- MOPO: Model-based Offline Policy OptimizationTianhe Yu, Garrett Thomas, Lantao Yu, Stefano Ermon et al.NeurIPS 2020 · 989 citations
- MOReL: Model-Based Offline Reinforcement LearningRahul Kidambi, Aravind Rajeswaran, Praneeth Netrapalli, Thorsten JoachimsNeurIPS 2020 · 870 citations
- Is Pessimism Provably Efficient for Offline RL?Ying Jin, Zhuoran Yang, Zhaoran WangICML 2021 · 419 citations
- Bridging Offline Reinforcement Learning and Imitation Learning: A Tale of PessimismParia Rashidinejad, Banghua Zhu, Cong Ma, Jiantao Jiao et al.NeurIPS 2021 · 373 citations
- Bellman-consistent Pessimism for Offline Reinforcement LearningTengyang Xie, Ching-An Cheng, Nan Jiang, Paul Mineiro et al.NeurIPS 2021 · 339 citations
Related papers
- Optimal Conservative Offline RL with General Function Approximation via Augmented LagrangianParia Rashidinejad, Hanlin Zhu, Kunhe Yang, Stuart Russell et al.ICLR 2023
- Towards Instance-Optimal Offline Reinforcement Learning with PessimismMing Yin, Yu-Xiang WangNeurIPS 2021 · 93 citations
- Optimal Single-Policy Sample Complexity and Transient Coverage for Average-Reward Offline RLMatthew Zurek, Guy Zamir, Yudong ChenNeurIPS 2025 · 2 citations
- Importance Weighted Actor-Critic for Optimal Conservative Offline Reinforcement LearningHanlin Zhu, Paria Rashidinejad, Jiantao JiaoNeurIPS 2023 · 21 citations
- Beyond the Return: Off-policy Function Estimation under User-specified Error-measuring DistributionsAudrey Huang, Nan JiangNeurIPS 2022 · 9 citations
