system, calculated the forces associated with each inter-entity interaction, and applied them to the individual
entities as accelerations by dividing by the entity’s mass, parameterized as inverse-mass, a=Fm1in both the
engine and model input. This also allows static objects to be represented as having m1= 0. The previous
positions and velocities, and newly computed accelerations, were input to an Euler integrator to update the
current velocities and positions of the entities. By using a custom engine we were able to have perfect control
over the details of the simulation, and use one engine for all physical domains. It produced trajectory rollouts
that were indistinguishable from those of an off-the-shelf simulation engine, but was highly efﬁcient because it
could perform thousands of runs in parallel on a GPU, allowing millions of simulated steps to be generated each
second.
A.1.2 Physical domains
n-body All objects in n-body systems exerted gravitational forces on each other, which were a function of
the objects’ pairwise distances and masses, giving rise to very complex dynamics that are highly sensitive to
initial conditions. Across simulation runs, the objects’ masses varied, while all other non-dynamic variables
were held constant. The gravitational constant was set to ensure that objects could move across several hundred
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