robo-gym -- An Open Source Toolkit for Distributed Deep Reinforcement Learning on Real and Simulated Robots

TL;DR

Robo-gym is an open source toolkit designed to facilitate seamless transfer of deep reinforcement learning algorithms from simulation to real robots, supporting distributed training and real-world applications.

Contribution

It introduces a unified framework that simplifies simulation-to-real transfer in robotics and supports distributed deep reinforcement learning training.

Findings

Successful deployment on industrial robots: mobile and arm robots.

Enables distributed training across multiple machines.

Facilitates simulation-to-real transfer with minimal fine-tuning.

Abstract

Applying Deep Reinforcement Learning (DRL) to complex tasks in the field of robotics has proven to be very successful in the recent years. However, most of the publications focus either on applying it to a task in simulation or to a task in a real world setup. Although there are great examples of combining the two worlds with the help of transfer learning, it often requires a lot of additional work and fine-tuning to make the setup work effectively. In order to increase the use of DRL with real robots and reduce the gap between simulation and real world robotics, we propose an open source toolkit: robo-gym. We demonstrate a unified setup for simulation and real environments which enables a seamless transfer from training in simulation to application on the robot. We showcase the capabilities and the effectiveness of the framework with two real world applications featuring industrial robots: a mobile robot and a robot arm. The distributed capabilities of the framework enable several advantages like using distributed algorithms, separating the workload of simulation and training on different physical machines as well as enabling the future opportunity to train in simulation and real world at the same time. Finally we offer an overview and comparison of robo-gym with other frequently used state-of-the-art DRL frameworks.