of restitution. The data distribution was much more challenging: for more than 99% of the time steps, a ball was
not in contact with any others, and its next-step velocity equaled its current velocity. For the remaining steps,
however, collisions caused next-step velocities that was a complex function of its state and the state of the object
it collides with. The training scene contained 6 balls inside a box with 4 walls, and test scenes contained either 3,
6, or 9 balls. The balls’ radii were sampled uniformly from [0:1;0:3]m, and masses from [0:75;1:25]kg. The
walls were static rectangular boxes, positioned so that the horizontal and vertical lengths of area they enclosed
varied independently between [1;3]m. The balls’ initial positions were randomly sampled so that all balls ﬁt
within the box and did not interpenetrate any other object, with initial x- andy-velocity components sampled
uniformly from [5;5]m/s. The restitutions, an attribute of their collision relations, was sampled uniformly
from[0:4;1].
Rigid body collision systems are highly nonlinear (chaotic) dynamical systems, as well. Collision forces were
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