Gaussian Approximation for Asynchronous Q-learning
Organizations: HSE University, Moscow, Russia · HSE University & Lomonosov MSU, Moscow, Russia · 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 the asynchronous Q-learning algorithm with a polynomial stepsize . Assuming that the sequence of state-action-next-state triples forms a uniformly geometrically ergodic Markov chain, we establish a rate of order up to over the class of hyper-rectangles, where is the number of samples used by the algorithm and and denote the numbers of states and actions, respectively. To obtain this result, we prove a high-dimensional central limit theorem for sums of martingale differences, which may be of independent interest. Finally, we present bounds for high-order moments for the algorithm's last iterate.
Figures & tables
Appendix figures & tables1 asset
Supplementary material from the paper’s appendix.
Appendix
| Notation | Meaning |
|---|---|
| Markov Decision Process | |
| Markov Decision Process (MDP) | |
| State space, | |
| Action space, | |
| Reward function | |
| Transition kernel of the MDP | |
Hitting-Time Bounds for Asynchronous Q-Learning
Sharp asymptotic theory for Q-learning with LDTZ learning rate and its generalization
LD2Z: ) schedule has shown appreciable empirical performance, but its theoretical and statistical properties remain largely unexplored, especially in the Q-learning setting. We address this gap in the literature by first considering a general class of power-law decay to zero (PD2Z-: ). Proceeding step-by-step, we present a sharp non-asymptotic error bound for Q-learning with PD2Z- schedule, which then is used to derive a central limit theory for a new tail Polyak-Ruppert averaging estimator. Finally, we also provide a novel time-uniform Gaussian approximation (also known as strong invariance principle) for the partial sum process of Q-learning iterates, which facilitates bootstrap-based inference. All our theoretical results are complemented by extensive numerical experiments. Beyond being new theoretical and statistical contributions to the Q-learning literature, our results definitively establish that LD2Z and in general PD2Z- achieve a best-of-both-worlds property: they inherit the rapid decay from initialization (characteristic of constant step-sizes) while retaining the asymptotic convergence guarantees (characteristic of polynomially decaying schedules). This dual advantage explains the empirical success of LD2Z while providing practical guidelines for inference through our results.