ManiSkill: A Unified Benchmark for Generalizable Manipulation Skills
FeaturedJiayuan Gu, Sean Xiang, Stone Tao, Xuanlin Li, Chongkai Gao, Yihe Tang, Xiaodi Yuan, Yunfan Jiang, Pu Hua, Tongzhou Mu, Fanbo Xiang, Yuzhe Qin, Xiangfeng Hu, Xiaowen Ou, Yen-Yu Chang, Xuanlin Li, Hao Su · UC San Diego, BIGAI · 2023
Framework
PyTorch + SAPIEN
License
MIT
Stars
2,914
Summary
ManiSkill is a GPU-parallelized robotics simulation benchmark with 20+ manipulation tasks, PartNet-Mobility assets, and unified observation/action spaces, supporting RL, IL, and VLA training at millions of steps per hour.
Abstract Summary
Key Points
- GPU-parallelized simulation: 1M+ steps/sec, 1000+ parallel envs.
- 20+ tasks: articulated objects, tools, deformable materials.
- Unified observation/action spaces across all tasks.
- 100K+ motion-planned demonstrations for offline IL.
- Standard benchmark for VLA models (OpenVLA, π0, Octo).
Building Environments for Generalizable Manipulation Skills
Building high-quality environments demands cross-disciplinary knowledge and expertise, including physical simulation, rendering, robotics, machine learning, software engineering, etc . Our workflow highlights a verification-driven iterative development process, which is illustrated in Appendix A.1 .
Real-time Soft Body Simulation and Rendering
In this section, we describe our new physical simulator for soft bodies and their dynamic interactions with the existing rigid-body simulation. Our key contributions include: 1) a highly efficient GPU MPM simulator; 2) an effective 2-way dynamic coupling method to support interactions between our soft-body simulator and any rigid-body simulator, in our case, SAPIEN.
These features enable us to create the first real-time MPM-based soft-body manipulation environment with 2-way coupling.
Parallelizing Physical Simulation and Rendering
ManiSkill2 aims to be a general and user-friendly framework, with low barriers for customization and minimal limitations. Therefore, we choose Python as our scripting language to model environments, and the open-source, highly flexible SAPIEN ( Xiang et al., 2020 ) as our physical simulator.
The Python language determines that the only way to effectively parallelize execution is through multi-process, which is integrated in common visual RL training libraries ( Raffin et al., 2021 ; Weng et al., 2021 ) .
Applications
In this section, we show how ManiSkill2 supports widespread applications, including sense-plan-act frameworks and imitation / reinforcement learning. In this section, for tasks that have asset variations, we report results on training objects. Results on held-out objects are presented in Appendix F .
Besides, we demonstrate that policies trained in ManiSkill2 have the potential to be directly deployed in the real world. Sense-plan-act (SPA) is a classical framework in robotics.
Sources and Demonstrations
Figures are reproduced from Gu et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.
Related Papers
Symmetries Here and There, Combined Everywhere: Cross-space Symmetry Compositions in Robotics
Loizos Hadjiloizou, Rodrigo Pérez-Dattari, Noémie Jaquier
Formalizes cross-space symmetry compositions in robotics and derives equivariant network architectures that exploit multiple simultaneous symmetries for improved generalization.
Eureka: Human-Level Reward Design via Coding Large Language Models
Yecheng Jason Ma, William Liang, Guanzhi Wang et al.
Eureka uses GPT-4 to write reward functions for RL environments, achieving human-level reward design on 29 tasks and enabling zero-shot sim-to-real transfer on Shadow Hand dexterous manipulation.
LeRobot: A Library for Real-World Robot Learning
Remi Cadene, Simon Alibert, Alexander Soare et al.
Hugging Face's LeRobot is an open-source PyTorch framework providing pretrained models, datasets, and training scripts for imitation and reinforcement learning on real robots, lowering the entry barrier to robot learning.
Gymnasium: A Standard Interface for Reinforcement Learning Environments
Farama Foundation, Jordan Terry, Mark Towers et al.
Gymnasium is the maintained successor to OpenAI Gym, providing a standardized API for RL environments with 100+ built-in tasks, vectorized parallel execution, and native support for physics engines like MuJoCo, PyBullet, and IsaacGym.