English

Fast Rates for the Regret of Offline Reinforcement Learning

Machine Learning 2023-07-13 v2 Artificial Intelligence Optimization and Control Machine Learning

Abstract

We study the regret of reinforcement learning from offline data generated by a fixed behavior policy in an infinite-horizon discounted Markov decision process (MDP). While existing analyses of common approaches, such as fitted QQ-iteration (FQI), suggest a O(1/n)O(1/\sqrt{n}) convergence for regret, empirical behavior exhibits \emph{much} faster convergence. In this paper, we present a finer regret analysis that exactly characterizes this phenomenon by providing fast rates for the regret convergence. First, we show that given any estimate for the optimal quality function QQ^*, the regret of the policy it defines converges at a rate given by the exponentiation of the QQ^*-estimate's pointwise convergence rate, thus speeding it up. The level of exponentiation depends on the level of noise in the \emph{decision-making} problem, rather than the estimation problem. We establish such noise levels for linear and tabular MDPs as examples. Second, we provide new analyses of FQI and Bellman residual minimization to establish the correct pointwise convergence guarantees. As specific cases, our results imply O(1/n)O(1/n) regret rates in linear cases and exp(Ω(n))\exp(-\Omega(n)) regret rates in tabular cases. We extend our findings to general function approximation by extending our results to regret guarantees based on LpL_p-convergence rates for estimating QQ^* rather than pointwise rates, where L2L_2 guarantees for nonparametric QQ^*-estimation can be ensured under mild conditions.

Keywords

Cite

@article{arxiv.2102.00479,
  title  = {Fast Rates for the Regret of Offline Reinforcement Learning},
  author = {Yichun Hu and Nathan Kallus and Masatoshi Uehara},
  journal= {arXiv preprint arXiv:2102.00479},
  year   = {2023}
}