EWRL Workshop 2022 Workshop Paper
On Convergence of Neural asynchronous Q-iteration
- Elena Smirnova
Deep Q-network algorithm is a successful and scalable algorithm on a variety of application domains. The algorithm incrementally trains a Q-network using one-step gradient descent over a mini-batch of randomly sampled transitions in a replay buffer. In this work, we formalize the Deep Q-network algorithm using an abstract algorithm, Neural asynchronous Q-iteration. We show convergence at a rate O(γ̃N ), where N is the number of iterations, γ̃ is a contraction coefficient that depends on a neural network and a rate of a mini-batch update. In particular, we show that Neural Tangent Kernel of a multi-layer ReLU neural network is a non-expansive operator and asynchronous Bellman operator is a smooth Bellman operator on average. Compared to previous work, our analysis is carried in the infinite-dimensional value function space w. r. t. kernel-based value functions. In this space, the algorithm takes a simple form, called Kernel Value Iteration.