arXiv · 1803.02811
Accelerated Methods for Deep Reinforcement Learning
Abstract
Deep reinforcement learning (RL) has achieved many recent successes, yet experiment turn-around time remains a key bottleneck in research and in practice. We investigate how to optimize existing deep RL algorithms for modern computers, specifically for a combination of CPUs and GPUs. We confirm that both policy gradient and Q-value learning algorithms can be adapted to learn using many parallel simulator instances. We further find it possible to train using batch sizes considerably larger than are standard, without negatively affecting sample complexity or final performance. We leverage these facts to build a unified framework for parallelization that dramatically hastens experiments in both classes of algorithm. All neural network computations use GPUs, accelerating both data collection and training. Our results include using an entire DGX-1 to learn successful strategies in Atari games in mere minutes, using both synchronous and asynchronous algorithms.
Explore related subjects
Keep this discovery
Adam Stooke, Pieter Abbeel. 2018-03-07. Accelerated Methods for Deep Reinforcement Learning. https://arxiv.org/abs/1803.02811
Cite the original work for its findings. Save a collection to share your selection of sources.