Zap Q-Learning With Nonlinear Function Approximation
Shuhang Chen, Adithya M Devraj, Fan Lu, Ana Busic, Sean Meyn
Abstract
Zap Q-learning is a recent class of reinforcement learning algorithms, motivated primarily as a means to accelerate convergence. Stability theory has been absent outside of two restrictive classes: the tabular setting, and optimal stopping. This paper introduces a new framework for analysis of a more general class of recursive algorithms known as stochastic approximation. Based on this general theory, it is shown that Zap Q-learning is consistent under a non-degeneracy assumption, even when the function approximation architecture is nonlinear. Zap Q-learning with neural network function approximation emerges as a special case, and is tested on examples from OpenAI Gym. Based on multiple experiments with a range of neural network sizes, it is found that the new algorithms converge quickly and are robust to choice of function approximation architecture.
BibTeX
@inproceedings{NEURIPS2020_c42f891c,
author = {Chen, Shuhang and Devraj, Adithya M and Lu, Fan and Busic, Ana and Meyn, Sean},
booktitle = {Advances in Neural Information Processing Systems},
editor = {H. Larochelle and M. Ranzato and R. Hadsell and M.F. Balcan and H. Lin},
pages = {16879--16890},
publisher = {Curran Associates, Inc.},
title = {Zap Q-Learning With Nonlinear Function Approximation},
url = {https://proceedings.neurips.cc/paper_files/paper/2020/file/c42f891cebbc81aa59f8f183243ac2b9-Paper.pdf},
volume = {33},
year = {2020}
}