stat.MLMay 17, 2026

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

Authors: Artemy RubtsovRahul SinghEric MoulinesAlexey NaumovSergey Samsonov

Organizations: HSE University, Moscow, Russia · Mohamed Bin Zayed University of AI, UAE · EPITA, France · HSE University & Steklov Mathematical Institute of Russian Academy of Sciences, Moscow, Russia

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^{-ω}, ω(1/2,1)ω\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.

Explore similar work

CardsList