English

A Statistical Analysis of Polyak-Ruppert Averaged Q-learning

Machine Learning 2023-02-21 v4 Machine Learning

Abstract

We study Q-learning with Polyak-Ruppert averaging in a discounted Markov decision process in synchronous and tabular settings. Under a Lipschitz condition, we establish a functional central limit theorem for the averaged iteration QˉT\bar{\boldsymbol{Q}}_T and show that its standardized partial-sum process converges weakly to a rescaled Brownian motion. The functional central limit theorem implies a fully online inference method for reinforcement learning. Furthermore, we show that QˉT\bar{\boldsymbol{Q}}_T is the regular asymptotically linear (RAL) estimator for the optimal Q-value function Q\boldsymbol{Q}^* that has the most efficient influence function. We present a nonasymptotic analysis for the \ell_{\infty} error, EQˉTQ\mathbb{E}\|\bar{\boldsymbol{Q}}_T-\boldsymbol{Q}^*\|_{\infty}, showing that it matches the instance-dependent lower bound for polynomial step sizes. Similar results are provided for entropy-regularized Q-learning without the Lipschitz condition.

Keywords

Cite

@article{arxiv.2112.14582,
  title  = {A Statistical Analysis of Polyak-Ruppert Averaged Q-learning},
  author = {Xiang Li and Wenhao Yang and Jiadong Liang and Zhihua Zhang and Michael I. Jordan},
  journal= {arXiv preprint arXiv:2112.14582},
  year   = {2023}
}

Comments

Accepted by AISTATS 2023

R2 v1 2026-06-24T08:34:45.182Z