Efficient Differentiable Simulation of Articulated Bodies
Qiao, Yi-Ling, Liang, Junbang, Koltun, Vladlen, and Lin, Ming C

Publication: Proceedings of the 38th International Conference on Machine Learning

Abstract: We present a method for efficient differentiable simulation of articulated bodies. This enables integration of articulated body dynamics into deep learning frameworks, and gradient-based optimization of neural networks that operate on articulated bodies. We derive the gradients of the contact solver using spatial algebra and the adjoint method. Our approach is an order of magnitude faster than autodiff tools. By only saving the initial states throughout the simulation process, our method reduces memory requirements by two orders of magnitude. We demonstrate the utility of efficient differentiable dynamics for articulated bodies in a variety of applications. We show that reinforcement learning with articulated systems can be accelerated using gradients provided by our method. In applications to control and inverse problems, gradient-based optimization enabled by our work accelerates convergence by more than an order of magnitude.

Bibtex:

@inproceedings{qiao-2021-efficient-bodies,
  title = {Efficient Differentiable Simulation of Articulated Bodies},
  author = {Qiao, Yi-Ling and Liang, Junbang and Koltun, Vladlen and Lin, Ming C},
  year = {2021},
  month = {18--24 Jul},
  booktitle = {Proceedings of the 38th International Conference on Machine Learning},
  publisher = {PMLR},
  series = {Proceedings of Machine Learning Research},
  volume = {139},
  pages = {8661--8671},
  editor = {Meila, Marina and Zhang, Tong}
}