Title: ACCELERATING REINFORCEMENT LEARNING THROUGH GPU ATARI EMULATION

Abstract: We introduce CuLE (CUDA Learning Environment), a CUDA port of the Atari Learning Environment (ALE) which is used for the development of deep reinforcement algorithms. CuLE overcomes many limitations of existing CPU-based emulators and scales naturally to multiple GPUs. It leverages GPU parallelization to run thousands of games simultaneously and it renders frames directly on the GPU, to avoid the bottleneck arising from the limited CPU-GPU communication bandwidth. CuLE generates up to 155M frames per hour on a single GPU, a finding previously achieved only through a cluster of CPUs. Beyond highlighting the differences between CPU and GPU emulators in the context of reinforcement learning, we show how to leverage the high throughput of CuLE by effective batching of the training data, and show accelerated convergence for A2C+V-trace. CuLE is available at [URL revealed upon acceptance].

1. INTRODUCTION:

Figure 1 : In a typical DRL system, environments run on CPUs and DNNs on GPUs. The limited CPU-GPU communication bandwidth and small set of environments prevent full GPU usage.

Initially triggered by the success of DQN Mnih et al. (2015) , research in Deep Reinforcement Learning (DRL) has grown in popularity in the last years Lillicrap et al. (2015) ; Mnih et al. (2016; 2015) , leading to intelligent agents that solve non-trivial tasks in complex environments. But DRL also soon proved to be a challenging computational problem, especially if one wants to achieve peak performance on modern architectures.

Fig. 1 is a schematic representation of a modern DRL algorithm implementation: CPU environments execute a set of actions {a t-1 } at time t -1, and produce observable states {s t } and rewards {r t }. These data are migrated to a Deep Neural Network (DNN) on the GPU to eventually select the next action set, {a t }, which is copied back to the CPU. This sequence of operations defines the inference path, whose main aim is to generate training data. A training buffer on the GPU stores the states generated on the inference path; this is periodically used to update the DNN's weights θ, according to the training rule of the DRL algorithm (training path). A computationally efficient DRL system should balance the data generation and training processes, while minimizing the communication overhead along the inference path and consuming, along the training path, as many data per second as possible Babaeizadeh et al. (2016; 2017) . The solution to this problem is however non-trivial and many DRL implementations do not leverage the full computational potential of modern systems Stooke & Abbeel (2018b) .

We focus our attention on the inference path and move from the traditional CPU implementation of the Atari Learning Environment (ALE), a set of Atari 2600 games that emerged as an excellent DRL benchmark Bellemare et al. (2013) ; Machado et al. (2017) . We show that significant performance bottlenecks primarily stem from CPU environment emulation: the CPU cannot run a large set of environments simultaneously, the CPU-GPU communication bandwidth is limited, and the GPU is consequently underutilized. To both investigate and mitigate these limitations we introduce CuLE (CUDA Learning Environment), a DRL library containing a CUDA enabled Atari 2600 emulator, that renders frames directly in the GPU memory, avoids off-chip communication and achieves high GPU utilization by processing thousands of environments in parallel-something so far achievable only through large and costly distributed systems. Compared to the traditional CPU-based approach, GPU emulation improves the utilization of the computational resources: CuLE on a single GPU generates more Frames Per Second1 (FPS) on the inference path (between 39K and 125K, depending on the game, see Table 1 ) compared to its CPU counterpart (between 12.5K and 19.8K). Beyond offering CuLE ([URL revealed upon acceptance]) as a tool for research in the DRL field, our contribution can be summarized as follow:

(1) We identify common computational bottlenecks in several DRL implementations that prevent effective utilization of high throughput compute units and effective scaling to distributed systems.

(2) We introduce an effective batching strategy for large environment sets, that allows leveraging the high throughput generated by CuLE to quickly reach convergence with A2C+V-trace Espeholt et al. (2018) , and show effective scaling on multiple GPUs. This leads to the consumption of 26-68K FPS along the training path on a single GPU, and up to 187K FPS using four GPUs, comparable (Table 1 ) to those achieved by large clusters Stooke & Abbeel (2018a) ; Espeholt et al. (2018) .

(3) We analyze advantages and limitations of GPU emulation with CuLE in DRL, including the effect of thread divergence and of the lower (compared to CPU) number of instructions per second per thread, and hope that our insights may be of value for the development of efficient DRL systems. The wall clock convergence time of a DRL algorithm is determined by two main factors: its sample efficiency, and the computational efficiency of its implementation. Here we analyze the sample and computational efficiency of different DRL algorithms, in connection with their implementation.

2. RELATED WORK:

We first divide DRL algorithms into policy gradient and Q-value methods, as in Stooke & Abbeel (2018b) . Q-learning optimizes the error on the estimated action values as a proxy for policy optimization, whereas policy gradient methods directly learn the relation between a state, s t , and the optimal action, a t ; since at each update they follow, by definition, the gradient with respect to the policy itself, they improve the policy more efficiently. Policy methods are also considered more general, e.g. they can handle continuous actions easily. Also the on-or off-policy nature of an algorithm profoundly affects both its sample and computational efficiency. Off-policy methods allow re-using experiences multiple times, which directly improves the sample efficiency; additionally, old data stored in the GPU memory (replay buffer in Fig. 1 ) can be used to continuously update the DNN on the GPU, leading to high GPU utilization without saturating the inference path. The replay buffer has a positive effect on the stability of learning as well Mnih et al. (2015) . On-policy algorithms saturate the inference path more easily, as frames have to be generated on-the-fly using the current policy and moved from the CPU emulators to the GPU for processing with the DNN. On-policy updates are generally effective but they are also more prone to fall into local minima because of noise, especially if the number of environment is small -this is the reason why on-policy algorithms largely benefit (in term of stability) from a significant increase of the number of environments.

Among the policy gradient methods, A3C Mnih et al. ( 2016) is an asynchronous, on-policy, actorcritic algorithm where 16 CPU agents use copies of the same DNN to interact with CPU environments; N -step bootstrapping is used to reduce the variance of the critic, V (s t ; θ): the agents send updates to the DNN after every N = 5 actions, and the new set of global θ weights are broadcast to all agents after each update. A3C can solve an Atari game in approximately 4 days; its hybrid CPU-GPU implementation (GA3C Babaeizadeh et al. (2016; 2017) ) stores a single DNN on the GPU, collects CPU data in a system of queues to send them to the GPU in large batches, and achieves convergence in 1 day. Recent (and faster) policy gradient implementations, like PAAC Clemente et al. (2017) , PPO Schulman et al. (2017) , or A2C OpenAI (2017), store training data directly in a GPU training buffer, as depicted in Fig. 1 , which is well representative of the flow of computation of modern DRL algorithms.

Policy gradient algorithms are often on-policy: their efficient update strategy is counterbalanced by the bottlenecks in the inference path and competition for the use of the GPU along the inference and training path at the same time. Acceleration by scaling to a distributed system is possible but inefficient in this case: in IMPALA Espeholt et al. (2018) a cluster with hundreds of CPU cores is needed to accelerate A2C, while training is desynchronized to hide latency. As a consequence, the algorithm becomes off-policy, and V-trace was introduced to deal with off-policy data (see details in the Appendix). Acceleration on a DGX-1 has also been demonstrated for A2C and PPO, using large batch sizes to increase the GPU occupancy, and asynchronous distributed models that hide latency, but require periodic updates to remain synchronized Stooke & Abbeel (2018b) and overall achieves sublinear scaling with the number of GPUs. In practice, the efficiency generally scales sub-linearly on distributed systems, and though Atari games can be solved in a few hours or even minutes, the cost of such systems makes them prohibitive for many researchers.

Q-value methods, like DQN Mnih et al. (2015) 2017)) are generally off-policy and can be accelerated more easily, as the inference and training path can be decoupled, with the GPU continuously fed with training data from the replay buffer in Fig. 1 . Mapping the two paths on different devices is natural and effective, and indeed DQN methods have been significantly accelerated on distributed systems (see Nair et al. (2015) and Jaderberg et al. (2016) in Table 1 ). The replay buffer increases the sample efficiency, but the overall convergence time has to pay the cost of using a Q-value method. Furthermore, implementation on large scale systems still incurs a communication overhead and thus generally achieves sub-linear scaling on large distributed systems.

The best compromise between sample efficiency, computational efficiency and effectiveness of the DNN update seems to be achieved by using an on-policy while correcting for stale data, gradient policy methods like A2C+V-trace, which is therefore the algorithm analyzed in detail here, but still suffers from computational bottlenecks in its CPU distributed implementation Espeholt et al. (2018) .

Other Data Generation Engines Frameworks to generate training experiences already exist, like the well-known OpenAI Gym, built on top of the C++ Stella emulator Brockman et al. (2016) for Atari games. Other frameworks that target specific problems, such as navigation Mirowski et al. (2016) , aim at creating large benchmark for DRL Tassa et al. (2018) , or accelerate simulation on the CPU with optimized C++ multi-threaded implementations, while also providing large, GPU-friendly batches for inference and training Tian et al. (2017) . To the best of our knowledge, however, none of these attempts directly address the problem of optimizing the inference path, i.e., trying to increase the number of environments while avoiding the CPU and bandwidth bottlenecks, by providing a direct implementation of the environments on the GPU.

3 CUDA LEARNING ENVIRONMENT (CULE) Despite the number and variety of games developed for the Atari 2600, it is relatively simple to emulate the functioning of its hardware much faster than real-time.

The Atari console has a 1.19Mhz CPU (to execute the game instructions) and a 128 bytes RAM (containing the game state); the code of each game is stored in a cartridge ROM (typically 2-4kB). The instruction set of each game contains both game and rendering instructions; the first ones are executed by the CPU, while the second ones are sent to the Television Interface Adaptor (TIA), a secondary processor embedded in the Atari hardware that executes the rendering instructions to update a limited set of registers and render the 160 × 210 game screen using a 128colour palette.

In CuLE, we emulate the functioning of many Atari consoles in parallel using the CUDA programming model, where a sequential host program executes parallel programs, known as kernels, on a GPU. In a trivial mapping of the Atari emulator to CUDA, a single thread emulates both the Atari CPU and TIA to execute the ROM code, update the Atari CPU and TIA registers as well as the game state in the 128 bytes RAM, and eventually render the pixels in the output frame. However, the contrasting nature of the game code execution and renderings tasks, the first dominated by reading from the RAM/ROM and writing tens of bytes to RAM, while the second writes hundreds of pixels to the framebuffer, poses a serious issue in terms of performance, such as thread divergence and an imbalanced number of registers required by the first and second tasks. To mitigate these issues, CuLE uses two CUDA kernels: the first one first loads data from the GPU global memory, where we store the state of each emulated Atari processor, and the 128 bytes RAM data containing the current state of the game; it also reads ROM instructions from the constant GPU memory, executes them to update the Atari CPU and game states, and stores the updated Atari CPU and game states back into GPU global memory. It is important to notice that this first kernel does not execute the TIA instructions read from the ROM, but copies them into the TIA instruction buffer in GPU global memory, which we implemented to decouple the execution of the Atari CPU and TIA instructions in CuLE. The second CuLE kernel emulates the functioning of the TIA processor: it first reads the instructions stored in the TIA instruction buffer, execute them to update the TIA registers, and renders the 160 × 210 output framebuffer in global GPU memory. Despite this implementation requires going through the TIA instruction twice, it has several advantages over the single-kernel trivial implementation. First of all, the requirements in terms of registers per thread and the chance of having divergent code are different for the Atari CPU and TIA kernels, and the use of different kernels achieves a better GPU usage. A second advantage that we exploit is that not all frames are rendered in ALE: the input of the RL algorithm is the pixelwise maximum between the last two frames in a sequence of four, so we can avoid calling the TIA kernel when rendering of the screen is not needed. A last advantage, not exploited in our implementation yet, is that the TIA kernel may be scheduled one the GPU with more than one thread per game, as rendering of diverse rows on the screen is indeed a parallel operation -we leave this optimization for future developments of CuLE.

To better fit our execution model, our game reset strategy is also different from the one in the existing CPU emulators, where 64 startup frames are executed at the end of each episode. Furthermore, wrapper interfaces for RL, such as ALE, randomly execute an additional number of frames (up to 30) to introduce randomness into the initial state. This results into up to 94 frames to reset a game, which may cause massive divergence between thousands of emulators executing in SIMD fashion on a GPU. To address this issue, we generate and store a cache of random initial states (30 by default) when a set of environments is initialized in CuLE. At the end of an episode, each emulator randomly selects one of the cached states as a seed and copies it into the terminal emulator state.

Some of the choices made for the implementation of CuLE are informed by ease of debugging, like associating one state update kernel to one environment, or need for flexibility, like emulating the Atari console instead of directly writing CUDA code for each Atari game. A 1-to-1 mapping between threads and emulators is not the most computationally efficient way to run Atari games on a GPU, but it makes the implementation relatively straightforward and has the additional advantage that the same emulator code can be executed on the CPU for debugging and benchmarking (in the following, we will refer to this implementation as CuLE CPU ). Despite of this, the computational advantage provided by CuLE over traditional CPU emulation remains significant, as shows in the next Section.

4. EXPERIMENTS:

Atari emulation We measure the FPS under different conditions: we get an upper bound on the maximum achievable FPS in the emulation only case, when we emulate the environments and use a random policy to select actions. In the inference only case, we measure the FPS along the inference path: a policy DNN selects the actions, CPU-GPU data transfer occur for CPU emulators, while both emulation and DNN inference run on the GPU when CuLE is used. This is the maximum throughput achievable by off-policy algorithms, when data generation and consumption are decoupled and run on different devices. In the training case, the entire DRL system is at work: emulation, inference, and training may all run on the same GPU. This is representative of the case of on-policy algorithms, but the FPS are also affected by the computational cost of the specific DRL update algorithm; in our experiments we use a vanilla A2C OpenAI (2017), with N-step bootstrapping, and N = 5 as the baseline (for details of A2C and off-policy correction with V-trace, see the Appendix). In the emulation only case, CPU emulation is more efficient for a number of environments up to 128, when the GPU computational power is not leveraged because of the low occupancy. For a larger number of environments, CuLE significantly overcomes OpenAI Gym, for which FPS are mostly stable for 64 environments or more, indicating that the CPU is saturated: the ratio between the median FPS generated by CuLE with 4096 environment (64K) and the peak FPS for OpenAI Gym (18K) is 3.56×. In the inference only case there are two additional overheads: CPU-GPU communication (to transfer observations), and DNN inference on the GPU. Consequently, CPU emulators achieve a lower FPS in inference only when compared to emulation only; the effects of the overheads is more evident for a small number of environments, while the FPS slightly increase with the number of environments without reaching the emulation only FPS. CuLE's FPS are also lower for inference only, because of the latency introduced by DNN inference, but the FPS grow with the number of environments, suggesting that the computational capability of the GPU is still far from being saturated. 2 for emulation only on four Atari games, 512 environments, for CuLE; each panel also shows the number of resetting environments. FPS is higher at the beginning, when all environments are in similar states and thread divergence within warps is minimized; after some steps, correlation is lost, FPS decreases and stabilizes. Minor oscillations in FPS are possibly associated to more or less computational demanding phases in the emulation of the environments (e.g., when a goal is scored in Pong).

Factors affecting the FPS Figs. 3(a)-3(b) shows that the throughput varies dramatically across games: 4096 CuLE CPU environments run at 27K FPS on Riverraid, but only 14K FPS for Boxing: a 1.93× difference, explained by the different complexity of the ROM code of each game. The ratio between the maximum and minimum FPS is amplified in the case of GPU emulation: Riverraid runs in emulation only at 155K FPS when emulated by CuLE and 4096 environments, while UpNDown runs at 41K FPS -a 3.78× ratio.

To better highlight the impact of thread divergence on throughput, we measure the FPS for CuLE, emulation only, 512 environments, and four games (Fig. 4 ). All the environments share the same initial state, but random action selection leads them to diverge after some steps. Each environment resets at the end of an episode. The FPS is maximum at the very beginning, when all the environments are in similar states and the chance to execute the same instruction in all the threads is high. When they move towards different states, code divergence negatively impacts the FPS, until it reaches an asymptotic value. This effect is present in all games and particularly evident for MsPacman in Fig. 4 ; it is not present in CPU emulation (see Appendix). Although divergence can reduce FPS by 30% in the worst case, this has to be compared with case of complete divergence within each thread and for each instruction, which would yield 1/32 3% of the peak performances. Minor oscillations of the FPS are also visible especially for games with a repetitive pattern (e.g. Pong), where different environments can be more or less correlated with a typical oscillation frequency. 2 for Assault, as a function of the number of environments, and different load conditions for A2C with N-step bootstrapping, N = 5).

Performances during training Fig. 5 compares the FPS generated by different emulation engines on a specific game (Assault)2 , for different load conditions, including the training case, and number of environments. As expected, when the entire training path is at work, the FPS decreases even further. However, for CPU emulators, the difference between FPS in the inference only and training cases decreases when the number of environments increases, as the system is bounded by the CPU computational capability and CPU-GPU communication bandwidth. In the case of the CPU scaling to multiple GPUs would be ineffective for on-policy algorithms, such GA3C Babaeizadeh et al. (2016; 2017) , or sub-optimal, in the case of distributed systems Espeholt et al. (2018); Stooke & Abbeel (2018b) . On the other hand, the difference between inference only and training FPS increases with the number of environments for CuLE, because of the additional training overhead on the GPU. The potential speed-up provided by CuLE for vanilla A2C and Assault in Fig. 5 is 2.53× for 1,024 environments, but the system is bounded by the GPU computational power; as a consequence, better batching strategies that reduce the training computational overhead as well as scaling to multiple GPUs are effective to further increase the speed-up ratio, as demonstrated later in this Section.

When data generation and training can be decoupled, like for off-policy algorithms, training can be easily moved to a different GPU and the inference path can be used at maximum speed. The potential speed-up provided by CuLE for off-policy algorithms is then given by the ratio between the inference only median FPS for CuLE (56K) and CuLE CPU (18K), which is 3.11× for 4,096 environments. Furthermore, since the FPS remains flat for CPU emulation, the advantage of CuLE amplifies (for both on-and off-policy methods) with the number of environments.

Frames per second per environment Fig. 3(c)-3(d) show the FPS / environment for different emulation engines on System I, as a function of the number of environments. For 128 environments or fewer, CPU emulators generate frames at a higher rate (compared to CuLE), because CPUs are optimized for low latency, and execute a high number of instructions per second per thread. However, the FPS / environment decrease with the number of environments, that have to share the same CPU cores. Instead, the GPU architecture maximizes the throughput and has a lower number of instructions per second per thread. As a consequence, the FPS / environment is smaller (compared to CPU emulation) for a small number of environments, but they are almost constant up to 512 environments, and starts decreasing only after this point. In practice, CuLE environments provide an efficient means of training with a diverse set of data and collect large statistics about the rewards experienced by numerous agents, and consequently lowering the variance of the value estimate. On the other hand, samples are collected less efficiently in the temporal domain, which may worsen the bias on the estimate of the value function by preventing the use of large N in N-step bootstrapping. The last paragraph of this Section shows how to leverage the high throughput generated by CuLE, considering these peculiarities.

Memory limitations Emulating a massively large number of environments can be problematic considering the relatively small amount of GPU DRAM. Our PyTorch Paszke et al. (2017) implementation of A2C requires each environment to store 4 84x84 frames, plus some additional variables for the emulator state. For 16K environments this translates into 1GB of memory, but the primary issue is the combined memory pressure to store the DNN with 4M parameters and the meta-data during training, including the past states: training with 16K environments easily exhausts the DRAM on a single GPU (while training on multiple GPUs increases the amount of available RAM). Since we did not implement any data compression scheme as in Horgan et al. (2018) , we constrain our training configuration to fewer than 5K environments, but peak performance in terms of FPS would be achieved for a higher number of environments -this is left as a possible future improvement.

A2C We analyze in detail the case of A2C with CuLE on a single GPU. As a baseline, we consider vanilla A2C, using 120 OpenAI Gym CPU environments that send training data to the GPU to update the DNN (Fig. 6(a) ) every N = 5 steps. This configuration takes, on average, 21.2 minutes (and 5.5M training frames) to reach a score of 18 for Pong and 16.6 minutes (4.0M training frames) for a score of 1,500 on Ms-Pacman (Fig. 7 , red line; first column of Table 3 ). CuLE with 1,200 environments generates approximately 2.5× more FPS compared to OpenAI Gym, but this alone is not sufficient to improve the convergence speed (blue line, Fig. 7 ). CuLE generates larger batches but, because FPS / environment is lower when compared to CPU emulation, fewer Updates Per Second (UPS) are performed for training the DNN (Table 3 ), which is detrimental for learning. 3 ), due to the fact that the GPU has to dedicate more time to training. Furthermore, as only the most recent data in a batch are generated with the current policy, we use V-trace Espeholt et al. (2018) for off-policy correction. The net result is an increase of the overall training time when 120 OpenAI Gym CPU environments are used, as this configuration pays for the increased training and communication overhead, while the smaller batch size increases the variance in the estimate of the value function and leads to noisy DNN updates (second column in Table 3 , orange lines in Fig. 7 ). Since CuLE does not suffer from the same computational bottlenecks, and at the same time benefits from the variance reduction associated with the large number (1,200) of environments, using the same batching strategy with CuLE reduces the time to reach a score of 18 for Pong and 1,500 for Pacman respectively to 5.9 and 6.9 minutes. The number of frames required to reach the same score is sometimes higher for CuLE (Table 3 ), which can lead to less sample efficient implementation when compared to the baseline, but the higher FPS largely compensates for this. Extending the batch size in the temporal dimension (N-steps bootstrapping, N = 20) increases the GPU computational load and reduces both the FPS and UPS, but it also reduces the bias in the estimate of the value function, making each DNN update more effective, and leads to an overall decrease of the 2 , and different batching strategies (see also Table 3 ).

Training frames are double for the multi-GPU case (black line).

Training performed on CuLE or OpenAI Gym; testing performed on OpenAI Gym environments (see the last paragraph of Section 4).

Scaling to multiple GPUs:

The black line in Fig. 7 GPUs on a single system. We use the PyTorch multiprocessing facilities to launch 1 process for each GPU and update gradients in a distributed manner using the NVIDIA NCCL multi-GPU communication backend. This Table shows efficient scaling of the training FPS with the number of GPUs. As in the case of A2C+Vtrace, an efficient batching strategy is then required to fully leverage the high throughput generated by CuLE and can significantly improve these figures. The same Table also reports the minimum and maximum GPU utilization measured while running each DRL algorithm. In its vanilla implementation DQN is characterized by long GPU idle times occurring during CPU emulation, leading to a low GPU utilization. The GPU utilization increases when emulation is moved to the GPU, while GPU peak utilization is reached during the DNN training step. GPU underutilization can be observed also for policy-gradient algorithms (A2C, PPO) and CPU emulation; peak utilization is higher for PPO, because of its larger computational complexity.

Nearly full GPU utilization is achieved only by CuLE for a large number of environments.

Correctness of the implementation To guarantee that CuLE is compliant with OpenAI Gym, we seed OpenAI Gym and CuLE environments with the same game state and successfully verify that the sequence of game states and frames returned by each environment are equivalent. A more A last note has to be done on CuLE's implementation, that is informed by ease of debugging, need for flexibility, and compatibility with standard DRL benchmarks. These choices put a limit on the achievable speed up factor (for instance by using emulation of the Atari 2600 console instead of direct CUDA implementations of Atari games), but the analysis and insights provided in our paper furnish indications for the design of efficient simulators for DRL.

A APPENDIX:

A.1 REINFORCEMENT LEARNING, A2C AND V-TRACE

Reinforcement learning:

In RL, an agent observes a state s t at time t and follows a policy π = π(s t ) to select an action a t ; the agent also receives a scalar reward r t from the environment. The goal of RL is to optimize π such that the sum of the expected rewards is maximized.

In model-free policy gradient methods π(a t |s t ; θ) is the output of a policy DNN with weights θ, and represents the probability of selecting action a t in the state s t . Updates to the DNN are generally aligned in the direction of the gradient of E[R t ], where R t = ∞ i=0 γ i r t+i is the discounted reward from time t, with discount factor γ ∈ (0, 1] (see also REINFORCE Williams ( 1992)) The vanilla implementation updates θ along ∇ θ log π(a t |s t ; θ)R t , which is an unbiased estimator of

∇ θ E[R t ].

The training procedure can be improved by reducing the variance of the estimator by subtracting a learned baseline b t (s t ) and using the gradient ∇ θ log π(a t |s t ; θ)[R t -b t (s t )]. One common baseline is the value function V π (s t ) = E[R t |s t ], which is the expected return for the policy π starting from s t . The policy π and the baseline b t can be viewed as actor and critic in an actor-critic architecture Sutton & Barto (1998) .

A2C A2C OpenAI ( 2017) is the synchronous version of A3C Mnih et al. (2016) , a successful actor-critic algorithm, where a single DNN outputs a softmax layer for the policy π (a t |s t ; θ), and a linear layer for V (s t ; θ). In A2C, multiple agents perform simultaneous steps on a set of parallel environments, while the DNN is updated every t max actions using the experiences collected by all the agents in the last t max steps. This means that the variance of the critic V (s t ; θ) is reduced (at the price of an increase in the bias) by N -step bootstrapping, with N = t max . The cost function for the policy is then: log π (a t |s t ; θ) Rt -V (s t ; θ t ) + βH [π (s t ; θ)] ,

(1) where θ t are the DNN weights θ at time t, Rt = k-1 i=0 γ i r t+i + γ k V (s t+k ; θ t ) is the bootstrapped discounted reward from t to t + k and k is upper-bounded by t max , and H [π (s t ; θ)] is an entropy term that favors exploration, weighted by the hyper-parameter β. The cost function for the estimated value function is:

EQUATION

which uses, again, the bootstrapped estimate Rt . Gradients ∇θ are collected from both of the cost functions; standard optimizers, such as Adam or RMSProp, can be used for optimization.

V-trace In the case where there is a large number of environments, such as in CuLE or IM-PALA Espeholt et al. (2018) , the synchronous nature of A2C become detrimental for the learning speed, as one should wait for all the environments to complete t max steps before computing a single DNN update. Faster convergence is achieved (both in our paper and in Espeholt et al. (2018) ) by desynchronizing data generation and DNN updates, which in practice means sampling a subset of experiences generated by the agents, and updating the policy using an approximate gradient, which makes the algorithm slightly off-policy.

To correct for the off-policy nature of the data, that may lead to inefficiency or, even worse, instabilities, in the training process, V-trace is introduced in Espeholt et al. (2018) . In summary, the aim of off-policy correction is to give less weight to experiences that have been generated with policy µ, called the behaviour policy, when it differs from the target policy, π; for a more principled explanation we remand the curios reader to Espeholt et al. (2018) .

For a set of experiences collected from time t = t 0 to time t = t 0 + N following some policy µ, the N -steps V-trace target for V (s t0 ; θ) is defined as:

:

Raw frames are reported here and in the rest of the paper, unless otherwise specified. These are the frames that are actually emulated, but only

25% of them are rendered and used for training. Training frames are obtained dividing the raw frames by 4-see alsoEspeholt et al. (2018).

Other games for which we observe a similar behavior are reported in the Appendix, for sake of space.

Figures and Tables:

Figure 2: Our CUDA-based Atari emulator uses an Atari CPU kernel to emulate the functioning of the Atari CPU and advance the game state, and a second TIA kernel to emulate the TIA and render frames directly in GPU memory. For episode resetting we generate and store a cache of random initial states. Massive parallelization on GPU threads allows the parallel emulation of thousands of Atari games.

Figure 3: FPS and FPS / environment on System I in Table 2, for OpenAI Gym OpenAI (2017), CuLE CPU , and CuLE, as a function of the number of environments, under different load conditions: emulation only, and inference only. The boxplots indicate the minimum, 25 th , 50 th , 75 th percentiles and maximum FPS, for the entire set of 57 Atari games.

Figs. 3(a)-3(b) show the FPS generated by OpenAI Gym, CuLE CPU , and CuLE, on the entire set of Atari games, as a function of the number of environments. In the emulation only case, CPU emulation is more efficient for a number of environments up to 128, when the GPU computational power is not leveraged because of the low occupancy. For a larger number of environments, CuLE significantly overcomes OpenAI Gym, for which FPS are mostly stable for 64 environments or more, indicating that the CPU is saturated: the ratio between the median FPS generated by CuLE with 4096 environment (64K) and the peak FPS for OpenAI Gym (18K) is 3.56×. In the inference only case there are two additional overheads: CPU-GPU communication (to transfer observations), and DNN inference on the GPU. Consequently, CPU emulators achieve a lower FPS in inference only when compared to emulation only; the effects of the overheads is more evident for a small number of environments, while the FPS slightly increase with the number of environments without reaching the emulation only FPS. CuLE's FPS are also lower for inference only, because of the latency introduced by DNN inference, but the FPS grow with the number of environments, suggesting that the computational capability of the GPU is still far from being saturated.

Figure 4: FPS as a function of the environment step, measured on System I in Table2for emulation only on four Atari games, 512 environments, for CuLE; each panel also shows the number of resetting environments. FPS is higher at the beginning, when all environments are in similar states and thread divergence within warps is minimized; after some steps, correlation is lost, FPS decreases and stabilizes. Minor oscillations in FPS are possibly associated to more or less computational demanding phases in the emulation of the environments (e.g., when a goal is scored in Pong).

Figure 5: FPS generated by different emulation engines on System I in Table2for Assault, as a function of the number of environments, and different load conditions for A2C with N-step bootstrapping, N = 5).

Figure 6: Different batching strategies are defined by the number of batches, N-Steps and Steps Per Update (SPU) parameters. The on-policy single batch case (panel a) is a special case of the more general, off-policy multi batch approach (panel b). The batching strategy affects both the computational and convergence aspects of the DRL algorithm, as shown in Fig. 7 and in Table 3. A2C+V-trace and batching strategy To better leverage CuLE, and similar in spirit to the approach in IMPALA Espeholt et al. (2018), we employ a different batching strategy, illustrated in Fig. 6(b): environment steps occur in parallel on the GPU, but training data are read in batches to update the DNN every Steps Per Update (SPU) steps. This batching strategy significantly increases the DNN's UPS at the cost of a slight decrease in FPS (second columns of OpenAI Gym and CuLE in Table 3), due to the fact that the GPU has to dedicate more time to training. Furthermore, as only the most recent data in a batch are generated with the current policy, we use V-trace Espeholt et al. (2018) for off-policy correction. The net result is an increase of the overall training time when 120 OpenAI Gym CPU environments are used, as this configuration pays for the increased training and communication overhead, while the smaller batch size increases the variance in the estimate of the value function and leads to noisy DNN updates (second column in Table3, orange lines in Fig.7). Since CuLE does not suffer from the same computational bottlenecks, and at the same time benefits from the variance reduction associated with the large number (1,200) of environments, using the same batching strategy with CuLE reduces the time to reach a score of 18 for Pong and 1,500 for Pacman respectively to 5.9 and 6.9 minutes. The number of frames required to reach the same score is sometimes higher for CuLE (Table3), which can lead to less sample efficient implementation when compared to the baseline, but the higher FPS largely compensates for this. Extending the batch size in the temporal dimension (N-steps bootstrapping, N = 20) increases the GPU computational load and reduces both the FPS and UPS, but it also reduces the bias in the estimate of the value function, making each DNN update more effective, and leads to an overall decrease of the

Figure 7: Average testing score and standard deviation on four Atari games as a function of the training time, for A2C+V-trace, System III in Table2, and different batching strategies (see also Table3). Training frames are double for the multi-GPU case (black line). Training performed on CuLE or OpenAI Gym; testing performed on OpenAI Gym environments (see the last paragraph of Section 4).

Average training times, raw frames to reach convergence, FPS, and computational resources of existing accelerated DRL schemes, compared to CuLE. Data fromHorgan et al. (2018); FPS are taken from the corresponding papers, if available, and measured on the entire Atari suite for CuLE.

Table content: <table><tr><td>Algorithm</td><td>Time</td><td>Frames</td><td>FPS</td><td>Resources</td><td>Notes</td></tr><tr><td>Ape-X DQN Horgan et al. (2018)</td><td>5 days</td><td>22,800M</td><td>50K</td><td>376 cores, 1 GPU</td><td>-</td></tr><tr><td>Rainbow Hessel et al. (2017)</td><td>10 days</td><td>200M</td><td>-</td><td>1 GPU</td><td>-</td></tr><tr><td>Distributional (C51) Bellemare et al. (2017)</td><td>10 days</td><td>200M</td><td>-</td><td>1 GPU</td><td>-</td></tr><tr><td>A3C Mnih et al. (2016)</td><td>4 days</td><td>-</td><td>2K</td><td>16 cores</td><td>-</td></tr><tr><td>GA3C Babaeizadeh et al. (2016; 2017)</td><td>1 day</td><td>-</td><td>8K</td><td>16 cores, 1 GPU</td><td>-</td></tr><tr><td>Prioritized Dueling Wang et al. (2015)</td><td>9.5 days</td><td>200M</td><td>-</td><td>1 GPU</td><td>-</td></tr><tr><td>DQN Mnih et al. (2015)</td><td>9.5 days</td><td>200M</td><td>-</td><td>1 GPU</td><td>-</td></tr><tr><td>Gorila DQN Nair et al. (2015)</td><td>4 days</td><td>-</td><td>-</td><td>&gt; 100 cores</td><td>-</td></tr><tr><td>Unreal Jaderberg et al. (2016)</td><td>-</td><td>250M</td><td>-</td><td>16 cores</td><td>-</td></tr><tr><td>Stooke (A2C / DQN) Stooke &amp; Abbeel (2018b)</td><td>hours</td><td>200M</td><td>35K</td><td>40 CPUs, 8 GPUs (DGX-1)</td><td>-</td></tr><tr><td>IMPALA (A2C + V-Trace) Espeholt et al. (2018)</td><td>mins/hours</td><td>200M</td><td>250K</td><td>100-200 cores, 1 GPU</td><td>-</td></tr><tr><td>CuLE (emulation only)</td><td>N/A</td><td>N/A</td><td>41K-155K</td><td>System I (1 GPU)</td><td>4096 ALEs</td></tr><tr><td>CuLE (inference only, A2C, single batch)</td><td>N/A</td><td>N/A</td><td>39K-125K</td><td>System I (1 GPU)</td><td>4096 ALEs</td></tr><tr><td>CuLE (training, A2C + V-trace, multiple batches)</td><td>1 hour</td><td>200M</td><td>26K-68K</td><td>System I (1 GPU)</td><td>1200 ALEs</td></tr><tr><td>CuLE (training, A2C + V-trace, multiple batches)*</td><td>mins</td><td>200M</td><td>142-187K</td><td>System III (4 GPUs)</td><td>1200×4 ALEs</td></tr></table>

Systems used for experiments.

Table content: <table><tr><td>System</td><td>Intel CPU</td><td>NVIDIA GPU</td></tr><tr><td>I</td><td>12-core Core i7-5930K @3.50GHz</td><td>Titan V</td></tr><tr><td>II</td><td>6-core Core i7-8086K @5GHz</td><td>Tesla V100</td></tr><tr><td>III</td><td colspan="2">20-core Core E5-2698 v4 @2.20GHz × 2 Tesla V100 × 8, NVLink</td></tr></table>

Training FPS, DNN's Update Per Second (UPS), time to reach a given score, and corresponding number of training frames for four Atari games, A2C+V-trace, and different configurations of the emulation engines, measured on System I in Table2(System III for the multi-GPU case). The best metric in each row is in bold.

Table content: <table><tr><td>Engine</td><td/><td colspan="2">OpenAI Gym</td><td/><td colspan="3">CuLE, 1 GPU</td><td>CuLE, 4 GPUs</td><td>Game</td></tr><tr><td>Envs</td><td>120</td><td>120</td><td>120</td><td>1200</td><td>1200</td><td>1200</td><td>1200</td><td>1200×4</td><td/></tr><tr><td>Batches N-steps</td><td>1 5</td><td>5 5</td><td>20 20</td><td>20 20</td><td>1 5</td><td>5 5</td><td>20 20</td><td>20×4 20</td><td>-</td></tr><tr><td>SPU</td><td>5</td><td>1</td><td>1</td><td>1</td><td>5</td><td>1</td><td>1</td><td>1</td><td/></tr><tr><td>Training KFPS UPS Time [mins] Training Mframes (for average score: 800)</td><td>4.2 7.0 20.2 5.0</td><td>3.4 28.3 --</td><td>3.0 24.7 42.6 7.5</td><td>4.9 4.1 44.2 13.0</td><td>10.6 1.8 18.8 12.0</td><td>11.5 9.6 9.4 6.5</td><td>11.0 9.1 9.9 6.5</td><td>42.7 8.9 7.9 18.0</td><td>Assault</td></tr><tr><td>Training KFPS UPS Time [mins] Training Mframes (for average score: 1,000)</td><td>4.3 7.1 8.1 2.0</td><td>3.3 27.9 35.2 7.0</td><td>3.0 24.8 14.4 2.5</td><td>4.9 4.1 27.1 8.0</td><td>11.9 2.0 --</td><td>12.5 10.4 14.0 10.5</td><td>12.1 10.0 3.4 2.5</td><td>46.6 9.7 2.5 7.0</td><td>Asterix</td></tr><tr><td>Training KFPS UPS Time [mins] Training Mframes (for average score: 1,500)</td><td>4.0 6.7 16.6 4.0</td><td>3.3 27.1 20.5 4.0</td><td>2.8 23.7 14.7 2.5</td><td>4.8 4.0 12.4 3.5</td><td>9.0 1.5 --</td><td>9.6 8.0 6.9 4.0</td><td>9.2 7.7 11.8 6.5</td><td>35.5 7.4 2.4 3.0</td><td>MsPacman</td></tr><tr><td>Training KFPS</td><td>4.3</td><td>3.4</td><td>3.0</td><td>4.8</td><td>10.5</td><td>11.2</td><td>10.6</td><td>41.7K</td><td/></tr><tr><td>UPS Time [mins]</td><td>7.2 21.2</td><td>28.1 12.2</td><td>24.9 8.4</td><td>4.0 8.7</td><td>1.8 -</td><td>9.3 5.9</td><td>8.9 3.1</td><td>8.7 2.4</td><td>Pong</td></tr><tr><td>Training Mframes (for average score: 18)</td><td>5.5</td><td>2.5</td><td>1.5</td><td>2.5</td><td>-</td><td>4.0</td><td>2.0</td><td>6.0</td><td/></tr></table>

Hours to complete 50M training frames (200M raw frames) on System III in Table 2, by GPU count, for two different DRL algorithms, 2,048 environments per GPU.

Table content: <table><tr><td>Generalization for different systems</td><td/><td/></tr><tr><td>Table 5 reports the FPS for the implemen-</td><td/><td/></tr><tr><td>tations of vanilla DQN, A2C, and PPO, on</td><td/><td/></tr><tr><td>System I and II in Table 2. The speed-up</td><td/><td/></tr><tr><td>in terms of FPS provided by CuLE is con-</td><td/><td/></tr><tr><td>sistent across different systems, different</td><td/><td/></tr><tr><td>algorithms, and larger in percentage for a large number of environments. Different DRL algorithms achieve different FPS de-pending on the complexity and frequency</td><td>Algo PPO Rainbow DQN</td><td>1 GPU 2 GPUs 4 GPUs 8 GPUs 9.38 4.45 2.12 1.1 10.5 5.1 3.2 2.1</td></tr><tr><td>of the training step on the GPU.</td><td/><td/></tr></table>

Average FPS and min/max GPU utilization during training for Pong with different algorithms and using different emulation engines on different systems (see Table2); CuLE consistently leads to higher FPS and GPU utilization. question to answer is if the CuLE reset procedure, based on a set of random initial states, can reduce the variability in the environments and consequently be detrimental for the learning and generalization capability of the trained agents. Our code ([URL revealed upon acceptance]) allows the user to test the training agent on CuLE, CuLE CPU , and OpenAI Gym environments: in our tests the results are comparable for the three backends. Results reported in Fig.7and in the Appendix further suggest that CuLE trained agents do not suffer any performance loss when tested on OpenAI Gym environments.5 DISCUSSION AND CONCLUSIONThe common allocation of the tasks in a DRL system dictates that the environment should run on CPUs, whereas GPUs should be dedicated to DNN operations. With most of the existing frameworksBrockman et al. (2016);Mirowski et al. (2016);Tassa et al. (2018);Tian et al. (2017) following this paradigm, the limited CPU-GPU communication bandwidth and CPU capability to emulate a large number of environments represent two limiting factors to effectively accelerate DRL algorithms, even when mapped to expensive distributed systems. By rendering frames directly on the GPU, CuLE overcomes these limitations and generate as many FPS as those generated by large, expensive CPU systems. CuLE promises to be an effective tool to develop and test DRL algorithms by significantly reducing the experiment turnaround time.As already shown by others in the case of DRL on distributed system, our experiments show that proper batching coupled with a slight off-policy gradient policy algorithm can significantly accelerate the wall clock convergence time; CuLE has the additional advantage of allowing effective scaling of this implementation to a system with multiple GPUs. CuLE effectively allows increasing the number of parallel environments but, because of the low number of instructions per second per thread on the GPU, training data can be narrow in the time direction. This can be problematic for problems with sparse temporal rewards, but rather than considering this as a pure limitation of CuLE, we believe that this peculiarity opens the door to new interesting research questions, like active sampling of important statesHessel et al. (2017);Wang et al. (2015) that can then be effectively analyzed on a large number of parallel environments with CuLE. CuLE also hits a new obstacle, which is the limited amount of DRAM available on the GPU; studying new compression schemes, like the one proposed inHessel et al. (2017), as well as training methods with smaller memory footprints may help extend the utility of CuLE to even larger environment counts, and design better GPU-based simulator for RL in the future. Since these are only two of the possible research directions for which CuLE is an effective investigation instrument, CuLE comes with a python interface that allows easy experimentation and is freely available to any researcher at [URL revealed upon acceptance].

Table content: <table><tr><td>Algorithm</td><td>Emulation engine</td><td/><td colspan="2">FPS [GPU utilization %]</td><td/></tr><tr><td/><td/><td>System I [256 envs]</td><td>System I [1024 envs]</td><td>System II [256 envs]</td><td>System II [1024 envs]</td></tr><tr><td/><td>OpenAI</td><td>6.4K [15-42%]</td><td>8.4K [0-69%]</td><td>10.8K [26-32%]</td><td>21.2K [28-75%]</td></tr><tr><td>DQN</td><td>CuLECPU</td><td>7.2K [16-43%]</td><td>8.6K [0-72%]</td><td>6.8K [17-25%]</td><td>20.8K [8-21%]</td></tr><tr><td/><td>CuLE</td><td>14.4K [16-99%]</td><td>25.6K [17-99%]</td><td>11.2K [48-62%]</td><td>33.2K [57-77%]</td></tr><tr><td/><td>OpenAI</td><td>12.8K [2-15%]</td><td>15.2K [0-43%]</td><td>24.4K [5-23%]</td><td>30.4K [3-45%]</td></tr><tr><td>A2C</td><td>CuLECPU</td><td>10.4K [2-15%]</td><td>14.2K [0-43%]</td><td>12.8K [1-18%]</td><td>25.6K [3-47%]</td></tr><tr><td/><td>CuLE</td><td>19.6K [97-98%]</td><td>51K [98-100%]</td><td>23.2K [97-98%]</td><td>48.0K [98-99%]</td></tr><tr><td/><td>OpenAI</td><td>12K [3-99%]</td><td>10.6K [0-96%]</td><td>16.0K [4-33%]</td><td>19.2K [4-62%]</td></tr><tr><td>PPO</td><td>CuLECPU</td><td>10K [2-99%]</td><td>10.2K [0-96%]</td><td>9.2K [2-28%]</td><td>18.4K [3-61%]</td></tr><tr><td/><td>CuLE</td><td>14K [95-99%]</td><td>36K [95-100%]</td><td>14.4K [43-98%]</td><td>28.0K [45-99%]</td></tr></table>

Peer review (iclr2020):
This paper introduces a CUDA port of the Atari Learning Environment. The paper goes into detail examining the benefits that come from a GPU-only implementation, including much better per-GPU utilization as well as no need to run a distributed system of CPUs. They show this hardware scaling can be taken advantage of across a variety of state of the art reinforcement learning algorithms and indeed create new batching strategies to utilize their framework and the GPU better. The paper is well written and goes into some detail describing the implementation of CuLE as well as various design decisions taken, as with splitting the emulation process across several kernels. Finally, the paper is very explicit about a number of optimizations that are not being exploited by the new framework and serve as markers for future work. A question that arises and which is not addressed in the experiments is how the authors verified their port is faithful to the original version; there is no mention of correctness in the paper.