Learning Physical Constraints with Neural Projections
Shuqi Yang, Xingzhe He, Bo Zhu
Abstract
We propose a new family of neural networks to predict the behaviors of physical systems by learning their underpinning constraints. A neural projection operator lies at the heart of our approach, composed of a lightweight network with an embedded recursive architecture that interactively enforces learned underpinning constraints and predicts the various governed behaviors of different physical systems. Our neural projection operator is motivated by the position-based dynamics model that has been used widely in game and visual effects industries to unify the various fast physics simulators. Our method can automatically and effectively uncover a broad range of constraints from observation point data, such as length, angle, bending, collision, boundary effects, and their arbitrary combinations, without any connectivity priors. We provide a multi-group point representation in conjunction with a configurable network connection mechanism to incorporate prior inputs for processing complex physical systems. We demonstrated the efficacy of our approach by learning a set of challenging physical systems all in a unified and simple fashion including: rigid bodies with complex geometries, ropes with varying length and bending, articulated soft and rigid bodies, and multi-object collisions with complex boundaries.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers10
- Equivariant Graph Mechanics Networks with ConstraintsWenbing Huang, Jiaqi Han, Yu Rong, Tingyang Xu et al.ICLR 2022 · 107 citations
- Learning to Accelerate Partial Differential Equations via Latent Global EvolutionTailin Wu, Takashi Maruyama, Jure LeskovecNeurIPS 2022 · 49 citations
- Learning Articulated Rigid Body Dynamics with Lagrangian Graph Neural NetworkRavinder Bhattoo, Sayan Ranu, N. M. Anoop KrishnanNeurIPS 2022 · 39 citations
- Constraint-based graph network simulatorYulia Rubanova, Alvaro Sanchez-Gonzalez, Tobias Pfaff, Peter W. BattagliaICML 2022 · 34 citations
- PlasticityNet: Learning to Simulate Metal, Sand, and Snow for Optimization Time IntegrationXuan Li, Yadi Cao, Minchen Li, Yin Yang et al.NeurIPS 2022 · 18 citations
Builds on5
- CLEVRER: Collision Events for Video Representation and ReasoningKexin Yi, Chuang Gan, Yunzhu Li, Pushmeet Kohli et al.ICLR 2020 · 584 citations
- Symplectic Recurrent Neural NetworksZhengdao Chen, Jianyu Zhang, Martín Arjovsky, Léon BottouICLR 2020 · 261 citations
- Hamiltonian Generative NetworksPeter Toth, Danilo J. Rezende, Andrew Jaegle, Sébastien Racanière et al.ICLR 2020 · 242 citations
- Learning to Control PDEs with Differentiable PhysicsPhilipp Holl, Nils Thuerey, Vladlen KoltunICLR 2020 · 221 citations
- Learning Compositional Koopman Operators for Model-Based ControlYunzhu Li, Hao He, Jiajun Wu, Dina Katabi et al.ICLR 2020 · 135 citations
Related papers
- Unsupervised Learning of Lagrangian Dynamics from Images for Prediction and ControlYaofeng Desmond Zhong, Naomi Ehrich LeonardNeurIPS 2020 · 49 citations
- Learning Physical Dynamics with Subequivariant Graph Neural NetworksJiaqi Han, Wenbing Huang, Hengbo Ma, Jiachen Li et al.NeurIPS 2022 · 72 citations
- Neural Modular Physics for Elastic SimulationYifei Li, Haixu Wu, Zeyi Xu, Tuur Stuyck et al.ICML 2026 · 1 citation
- Learning 3D Particle-based Simulators from RGB-D VideosWilliam F. Whitney, Tatiana Lopez-Guevara, Tobias Pfaff, Yulia Rubanova et al.ICLR 2024 · 16 citations
- Physics-as-Inverse-Graphics: Unsupervised Physical Parameter Estimation from VideoMiguel Jaques, Michael Burke, Timothy M. HospedalesICLR 2020 · 58 citations
