Control of Two-way Coupled Fluid Systems with Differentiable Solvers

Brener Ramos, Felix Trost, Nils Thuerey

Keywords: differentiable physics, control, physical simulations, nonlinear dynamics, fluids, deep learning

TL;DR: A neural network learns fluid control in an unsupervised way with differentiable physics.

Abstract: We investigate the use of deep neural networks to control complex nonlinear dynamical systems, specifically the movement of a rigid body immersed in a fluid. We solve the Navier Stokes equations with two way coupling, which gives rise to nonlinear perturbations that make the control task very challenging. Neural networks are trained to act as controllers with desired characteristics through a process of learning from a differentiable simulator. Here we introduce a set of physically interpretable loss terms to let the networks learn robust and stable interactions. We demonstrate that controllers trained in a canonical setting with quiescent initial conditions reliably generalize to varied and challenging environments such as previously unseen inflow conditions and forcing. Further, we show that controllers trained with our approach outperform a variety of classical and learned alternatives in terms of evaluation metrics and generalizing capabilities.

Community Implementations: [ 1 code implementation](https://www.catalyzex.com/paper/arxiv:2206.00342/code)