Timezone: »

 
Poster
Learning to Simulate Complex Physics with Graph Networks
Alvaro Sanchez-Gonzalez · Jonathan Godwin · Tobias Pfaff · Rex (Zhitao) Ying · Jure Leskovec · Peter Battaglia

Tue Jul 14 12:00 PM -- 12:45 PM & Wed Jul 15 01:00 AM -- 01:45 AM (PDT) @
in Poster Session 6 »

Here we present a machine learning framework and model implementation that can learn to simulate a wide variety of challenging physical domains, involving fluids, rigid solids, and deformable materials interacting with one another. Our framework---which we term "Graph Network-based Simulators" (GNS)---represents the state of a physical system with particles, expressed as nodes in a graph, and computes dynamics via learned message-passing. Our results show that our model can generalize from single-timestep predictions with thousands of particles during training, to different initial conditions, thousands of timesteps, and at least an order of magnitude more particles at test time. Our model was robust to hyperparameter choices across various evaluation metrics: the main determinants of long-term performance were the number of message-passing steps, and mitigating the accumulation of error by corrupting the training data with noise. Our GNS framework advances the state-of-the-art in learned physical simulation, and holds promise for solving a wide range of complex forward and inverse problems.

Keywords: Networks and Relational Learning · Structured Prediction · Time Series and Sequence Models · Sequential, Network, and Time-Series Modeling

Author Information

Alvaro Sanchez-Gonzalez (DeepMind)
Jonathan Godwin (DeepMind)
Tobias Pfaff (DeepMind)
Rex (Zhitao) Ying (Stanford University)
Jure Leskovec (Stanford University)
Peter Battaglia (DeepMind)

More from the Same Authors