English

On Gaussian approximation for entropy-regularized Q-learning with function approximation

Machine Learning 2026-05-19 v1 Machine Learning

Abstract

In this paper, we derive rates of convergence in the high-dimensional central limit theorem for Polyak--Ruppert averaged iterates generated by entropy-regularized asynchronous Q-learning with linear function approximation and a polynomial stepsize kωk^{-\omega}, ω(1/2,1)\omega \in (1/2,1). Assuming that the sequence of observed triples (sk,ak,sk+1)k0(s_k,a_k,s_{k+1})_{k \geq 0} forms a uniformly geometrically ergodic Markov chain, and under suitable regularity conditions for the projected soft Bellman equation, we establish a Gaussian approximation bound in the convex distance with rate of order n1/4n^{-1/4}, up to polylogarithmic factors in nn, where nn is the number of samples used by the algorithm. To obtain this result, we combine a linearization of the soft Bellman recursion with a Gaussian approximation for the leading martingale term. Finally, we derive high-order moment bounds for the algorithm's last iterate, which might be of independent interest.

Keywords

Cite

@article{arxiv.2605.17678,
  title  = {On Gaussian approximation for entropy-regularized Q-learning with function approximation},
  author = {Artemy Rubtsov and Rahul Singh and Eric Moulines and Alexey Naumov and Sergey Samsonov},
  journal= {arXiv preprint arXiv:2605.17678},
  year   = {2026}
}