Robust Rigid Body Assembly via Contact-Implicit Optimal Control with Exact Second-Order Derivatives
Christian Dietz, Sebastian Albrecht, Gianluca Frison, Moritz Diehl, Armin Nurkanović
TL;DR
This work tackles robust, sample-efficient planning of contact-rich rigid-body assembly by introducing a differentiable physics framework that delivers up to second-order derivatives to the solver. It combines an optimization-based, interior-point smoothed collision-detection approach (via a growth-distance SDF) with a smooth, differentiable multi-contact physics model and a robust, multi-scenario contact-implicit optimal control formulation using Cartesian impedance tracking. The main contributions include the modified SDF as a growth distance with an efficient derivative pipeline, a multi-scenario OCP that generalizes to sim-to-real mismatches, and extensive real-world and simulation results showing high success rates and the advantage of exact Hessians over common approximations. The approach achieves reliable assembly on real hardware and offers a flexible framework for incorporating additional constraints or frictional effects, advancing the practical use of derivative-based optimization for assembly tasks.
Abstract
Efficient planning of assembly motions is a long standing challenge in the field of robotics that has been primarily tackled with reinforcement learning and sampling-based methods by using extensive physics simulations. This paper proposes a sample-efficient robust optimal control approach for the determination of assembly motions, which requires significantly less physics simulation steps during planning through the efficient use of derivative information. To this end, a differentiable physics simulation is constructed that provides second-order analytic derivatives to the numerical solver and allows one to traverse seamlessly from informative derivatives to accurate contact simulation. The solution of the physics simulation problem is made differentiable by using smoothing inspired by interior-point methods applied to both the collision detection as well as the contact resolution problem. We propose a modified variant of an optimization-based formulation of collision detection formulated as a linear program and present an efficient implementation for the nominal evaluation and corresponding first- and second-order derivatives. Moreover, a multi-scenario-based trajectory optimization problem that ensures robustness with respect to sim-to-real mismatches is derived. The capability of the considered formulation is illustrated by results where over 99\% successful executions are achieved in real-world experiments. Thereby, we carefully investigate the effect of smooth approximations of the contact dynamics and robust modeling on the success rates. Furthermore, the method's capability is tested on different peg-in-hole problems in simulation to show the benefit of using exact Hessians over commonly used Hessian approximations.
