English

Offline Minimax Soft-Q-learning Under Realizability and Partial Coverage

Machine Learning 2023-11-14 v2 Machine Learning

Abstract

In offline reinforcement learning (RL) we have no opportunity to explore so we must make assumptions that the data is sufficient to guide picking a good policy, taking the form of assuming some coverage, realizability, Bellman completeness, and/or hard margin (gap). 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 soft (entropy-regularized) Q-function 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 to accurately estimate soft or vanilla Q-functions with L2L^2-convergence guarantees. Our algorithms' loss functions arise from casting the estimation problems as nonlinear convex optimization problems and Lagrangifying.

Keywords

Cite

@article{arxiv.2302.02392,
  title  = {Offline Minimax Soft-Q-learning Under Realizability and Partial Coverage},
  author = {Masatoshi Uehara and Nathan Kallus and Jason D. Lee and Wen Sun},
  journal= {arXiv preprint arXiv:2302.02392},
  year   = {2023}
}

Comments

The original title of this paper was "Refined Value-Based Offline RL under Realizability and Partial Coverage," but it was later changed. This paper has been accepted for NeurIPS 2023

R2 v1 2026-06-28T08:32:22.321Z