and 12 bodies. For balls, the training was on 6 balls, and generalization was to 3 and 9 balls. For strings, the
training was on 15 masses with 1 end pinned, and generalization was to 30 masses with 0 and 2 ends pinned.
The URLs to the full videos of each rollout are in Table 1.
7
Table 1: Rollout video URLs
System True Model
n-body 6 https://youtu.be/otIGNTFJwpU https://youtu.be/CXzubiwu4GM
n-body 3 https://youtu.be/zOTVzziJz24 https://youtu.be/3bFrbjdTc6s
n-body 12 https://youtu.be/JgSRaYmQyNY https://youtu.be/Rl44jdaIIiw
Balls 6 https://youtu.be/9c-UPXSjOHI https://youtu.be/SV741g_qp8M
Balls 3 https://youtu.be/op2d9IHh3Ak https://youtu.be/i0lOn_v8xVg
Balls 9 https://youtu.be/oQoquoVJ8V0 https://youtu.be/41y_saKvi3s
Reasoning about objects, relations, and physics is central to human
intelligence, and a key goal of artificial intelligence. Here we introduce the
interaction network, a model which can reason about how objects in complex
systems interact, supporting dynamical predictions, as well as inferences about
the abstract properties of the system. Our model takes graphs as input,
performs object- and relation-centric reasoning in a way that is analogous to a
simulation, and is implemented using deep neural networks. We evaluate its
ability to reason about several challenging physical domains: n-body problems,
rigid-body collision, and non-rigid dynamics. Our results show it can be
trained to accurately simulate the physical trajectories of dozens of objects
over thousands of time steps, estimate abstract quantities such as energy, and
generalize automatically to systems with different numbers and configurations
of objects and relations. Our interaction network implementation is the first
general-purpose, learnable physics engine, and a powerful general framework for
reasoning about object and relations in a wide variety of complex real-world
domains.
Peter W. Battaglia Razvan Pascanu Matthew Lai Danilo Rezende Koray Kavukcuoglu