EqCollide: Equivariant and Collision-Aware Deformable Objects Neural Simulator
About
Simulating collisions of deformable objects is a fundamental yet challenging task due to the complexity of modeling solid mechanics and multi-body interactions. Existing data-driven methods often suffer from lack of equivariance to physical symmetries, inadequate handling of collisions, and limited scalability. Here we introduce EqCollide, the first end-to-end equivariant neural fields simulator for deformable objects and their collisions. We propose an equivariant encoder to map object geometry and velocity into latent control points. A subsequent equivariant Graph Neural Network-based Neural Ordinary Differential Equation models the interactions among control points via collision-aware message passing. To reconstruct velocity fields, we query a neural field conditioned on control point features, enabling continuous and resolution-independent motion predictions. Experimental results on 2D and 3D scenarios show that EqCollide achieves accurate, stable, and scalable simulations across diverse object configurations. It achieves $24.34\%$ to $57.62\%$ lower rollout MSE, even compared with the best-performing baseline model. Furthermore, EqCollide could generalize to more colliding objects and extended temporal horizons, and stay robust to input transformed with group action. Code is available at: https://github.com/AI4Science-WestlakeU/EqCollide
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Displacement Prediction | 2D DeformableObjectsCollision unseen object combinations (test) | MSE (1-step)0.003 | 5 | |
| Displacement Prediction | 2D DeformableObjectsCollision unseen object shapes (test) | MSE (1 step)0.002 | 5 | |
| Deformable object collision prediction | Cube 3D (test) | Penetration Index (I_pen)0.00e+0 | 4 | |
| Deformable object collision prediction | Alphabet 3D (test) | Penetration Error (x10^-6)0.00e+0 | 4 | |
| Deformable object collision prediction | Cow 3D (test) | I_pen (×10^-6)14.3 | 4 | |
| Deformable objects neural simulation | Three-object collision unseen (test) | MSE (1-step)0.002 | 4 | |
| Displacement Prediction | Spot cow 3D deformable body collision (test) | MSE (1-step)0.0034 | 4 | |
| Displacement Prediction | Cubic 3D deformable body collision horizontal plane | MSE (1-step)3.00e-4 | 4 | |
| Displacement Prediction | Cubic 3D deformable body collision midair | MSE (1-step)0.0075 | 4 |