Control of Two-way Coupled Fluid Systems with Differentiable SolversDownload PDF

Published: 27 Apr 2022, Last Modified: 22 Oct 2023ICLR 2022 GPL PosterReaders: Everyone
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: [![CatalyzeX](/images/catalyzex_icon.svg) 1 code implementation](
1 Reply