SERL: A Software Suite for Sample-Efficient Robotic Reinforcement Learning

Jianlan Luo,Zheyuan Hu,Charles Xu,You Liang Tan,Jacob Berg,Archit Sharma,Stefan Schaal,Chelsea Finn,Abhishek Gupta,Sergey Levine
2024-02-13
Abstract:In recent years, significant progress has been made in the field of robotic reinforcement learning (RL), enabling methods that handle complex image observations, train in the real world, and incorporate auxiliary data, such as demonstrations and prior experience. However, despite these advances, robotic RL remains hard to use. It is acknowledged among practitioners that the particular implementation details of these algorithms are often just as important (if not more so) for performance as the choice of algorithm. We posit that a significant challenge to widespread adoption of robotic RL, as well as further development of robotic RL methods, is the comparative inaccessibility of such methods. To address this challenge, we developed a carefully implemented library containing a sample efficient off-policy deep RL method, together with methods for computing rewards and resetting the environment, a high-quality controller for a widely-adopted robot, and a number of challenging example tasks. We provide this library as a resource for the community, describe its design choices, and present experimental results. Perhaps surprisingly, we find that our implementation can achieve very efficient learning, acquiring policies for PCB board assembly, cable routing, and object relocation between 25 to 50 minutes of training per policy on average, improving over state-of-the-art results reported for similar tasks in the literature. These policies achieve perfect or near-perfect success rates, extreme robustness even under perturbations, and exhibit emergent recovery and correction behaviors. We hope that these promising results and our high-quality open-source implementation will provide a tool for the robotics community to facilitate further developments in robotic RL. Our code, documentation, and videos can be found at
Robotics,Artificial Intelligence
What problem does this paper attempt to address?
The problem that this paper attempts to solve is **to improve the usability and efficiency of robotic reinforcement learning (RL) in the real world**. Although significant progress has been made in the field of robotic reinforcement learning in recent years, enabling robots to handle complex image observations, train in the real world, and utilize auxiliary data (such as demonstrations and prior experiences), there are still many challenges in the practical application of these technologies. The main challenges include: 1. **The importance of implementation details**: The specific implementation details of an algorithm are often crucial to performance, even more important than which algorithm is chosen. 2. **Definition of the reward function**: In the case of using image observations, the definition of the reward function is very difficult and needs to rely on a complex perception system. 3. **Environment reset**: For tasks that require multiple attempts, how to automatically reset the robot and its environment to the initial state is a mechanical operation problem. 4. **Sample efficiency**: When learning in the real world, the training cost per minute and per hour is very high, so improving sample efficiency is very important. 5. **Controller design**: The design of the low - level controller needs to be particularly careful, especially in contact - rich tasks, to ensure both accuracy and safety. To address these challenges, the author has developed a software suite named **Sample - Efficient Robotic Reinforcement Learning (SERL)**. The main goal of SERL is to provide an out - of - the - box solution, enabling non - experts to easily use reinforcement learning to train physical robots, thereby promoting the wide application and development of robotic reinforcement learning. Specifically, SERL contains the following components: 1. **Efficient reinforcement learning implementation**: An efficient off - policy deep reinforcement learning method that supports image observations and demonstrations. 2. **Multiple implementations of the reward function**: Including multiple methods such as classifiers and adversarial training for inferring rewards from image observations. 3. **Automatic reset**: Automatically reset the environment between tasks by learning a "forward - backward" controller. 4. **Universal adapter**: A software package that can be connected to any robotic manipulator. 5. **Impedance controller**: Particularly suitable for contact - rich manipulation tasks, ensuring safety and accuracy when in contact with objects. Through these components, SERL can complete the training of tasks such as PCB board assembly, cable routing, and object re - positioning within 25 to 50 minutes, achieving near - perfect success rates and extremely high robustness. These results indicate that through a carefully designed software package, existing reinforcement learning techniques can achieve efficient real - world applications.