Acme: A Research Framework for Distributed Reinforcement Learning

TL;DR

Acme is a modular, scalable framework designed to facilitate the development, implementation, and reproducibility of distributed reinforcement learning algorithms across various scales and complexities.

Contribution

It introduces a highly modular, scalable framework for RL that includes reference implementations and supports large-scale distributed algorithms.

Findings

Provides baseline implementations for state-of-the-art RL algorithms.

Demonstrates scalability of algorithms to large, complex environments.

Enhances reproducibility and rapid prototyping in RL research.

Abstract

Deep reinforcement learning (RL) has led to many recent and groundbreaking advances. However, these advances have often come at the cost of both increased scale in the underlying architectures being trained as well as increased complexity of the RL algorithms used to train them. These increases have in turn made it more difficult for researchers to rapidly prototype new ideas or reproduce published RL algorithms. To address these concerns this work describes Acme, a framework for constructing novel RL algorithms that is specifically designed to enable agents that are built using simple, modular components that can be used at various scales of execution. While the primary goal of Acme is to provide a framework for algorithm development, a secondary goal is to provide simple reference implementations of important or state-of-the-art algorithms. These implementations serve both as a validation of our design decisions as well as an important contribution to reproducibility in RL research. In this work we describe the major design decisions made within Acme and give further details as to how its components can be used to implement various algorithms. Our experiments provide baselines for a number of common and state-of-the-art algorithms as well as showing how these algorithms can be scaled up for much larger and more complex environments. This highlights one of the primary advantages of Acme, namely that it can be used to implement large, distributed RL algorithms that can run at massive scales while still maintaining the inherent readability of that implementation. This work presents a second version of the paper which coincides with an increase in modularity, additional emphasis on offline, imitation and learning from demonstrations algorithms, as well as various new agents implemented as part of Acme.