Research

PlasticineLab: A Soft-Body Manipulation Benchmark with Differentiable Physics.

ICLR

Authors

  • Zhiao Huang
  • Yuanming Hu
  • Tao Du
  • Siyuan Zhou
  • Hao Su
  • Joshua B. Tenenbaum
  • Chuang Gan

Published on

04/07/2021

Simulated virtual environments serve as one of the main driving forces behind developing and evaluating skill learning algorithms. However, existing environments typically only simulate rigid body physics. Additionally, the simulation process usually does not provide gradients that might be useful for planning and control optimizations. We introduce a new differentiable physics benchmark called PasticineLab, which includes a diverse collection of soft body manipulation tasks. In each task, the agent uses manipulators to deform the plasticine into the desired configuration. The underlying physics engine supports differentiable elastic and plastic deformation using the DiffTaichi system, posing many under-explored challenges to robotic agents. We evaluate several existing reinforcement learning (RL) methods and gradient-based methods on this benchmark. Experimental results suggest that 1) RL-based approaches struggle to solve most of the tasks efficiently; 2) gradient-based approaches, by optimizing open-loop control sequences with the built-in differentiable physics engine, can rapidly find a solution within tens of iterations, but still fall short on multi-stage tasks that require long-term planning. We expect that PlasticineLab will encourage the development of novel algorithms that combine differentiable physics and RL for more complex physics-based skill learning tasks.

This paper has been published at ICML 2021

Please cite our work using the BibTeX below.

@misc{huang2021plasticinelab,
      title={PlasticineLab: A Soft-Body Manipulation Benchmark with Differentiable Physics}, 
      author={Zhiao Huang and Yuanming Hu and Tao Du and Siyuan Zhou and Hao Su and Joshua B. Tenenbaum and Chuang Gan},
      year={2021},
      eprint={2104.03311},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}
Close Modal