Neural Field Dynamics Model for
Granular Object Piles Manipulation


Shangjie Xue, Shuo Cheng, Pujith Kachana, Danfei Xu

Georgia Institute of Technology

[paper] [code (coming)]

Abstract

We present a learning-based dynamics model for granular material manipulation. Inspired by the Eulerian approach commonly used in fluid dynamics, our method adopts a fully convolutional neural network that operates on a density field-based representation of object piles and pushers, allowing it to exploit the spatial locality of inter-object interactions as well as the translation equivariance through convolution operations. Furthermore, our differentiable action rendering module makes the model fully differentiable and can be directly integrated with a gradient-based trajectory optimization algorithm. We evaluate our model with a wide array of piles manipulation tasks both in simulation and real-world experiments and demonstrate that it significantly exceeds existing latent or particle-based methods in both accuracy and computation efficiency, and exhibits zero-shot generalization capabilities across various environments and tasks.

Introduction

output.mp4

Trajectory Optimization Visualization

Qualitative Results

2_cp.mp4
3_oa.mp4
4_do.mp4
5_dp.mp4
6_s.mp4
9_spd.mp4
7_so.mp4
8_sdo.mp4

Real-world Demos

rw_blocks.mp4
rw_blocks_oa.mp4
rw_beans.mp4
rw_beans_oa.mp4
rw_big_few.mp4
rw_big_more.mp4