End-to-end differentiable physics for learning and control

• Main Authors: Smith, Kevin A (Author), Allen, Kelsey Rebecca (Author), Tenenbaum, Joshua B (Author)
• Other Authors: Massachusetts Institute of Technology. Department of Brain and Cognitive Sciences (Contributor)
• Format: Article
• Language: English
• Published: Curran Associates Inc, 2020-08-17T15:06:34Z.
• Subjects: Article
• Online Access: Get fulltext

 © 2018 Curran Associates Inc.All rights reserved. We present a differentiable physics engine that can be integrated as a module in deep neural networks for end-to-end learning. As a result, structured physics knowledge can be embedded into larger systems, allowing them, for example, to match observations by performing precise simulations, while achieves high sample efficiency. Specifically, in this paper we demonstrate how to perform backpropagation analytically through a physical simulator defined via a linear complementarity problem. Unlike traditional finite difference methods, such gradients can be computed analytically, which allows for greater flexibility of the engine. Through experiments in diverse domains, we highlight the system's ability to learn physical parameters from data, efficiently match and simulate observed visual behavior, and readily enable control via gradient-based planning methods. Code for the engine and experiments is included with the paper.