Well-posed by Design: Learning Constitutive Laws from Velocity Data using Convex Neural Network Potentials

Abstract

Learning constitutive laws of complex fluids from velocity data (i.e. indirect observations) is a PDE-constrained inverse problem in which expressive neural parameterizations risk breaking the well-posedness of the forward physics model. We address this tension by learning, rather than the constitutive law itself, the dissipation potential: a scalar function whose convexity, frame-indifference, and dissipativity propagate into the underlying continuum mechanics and guarantee the structural properties needed for a well-posed and generalizable forward PDE. We introduce ICNNE, an input-convex neural architecture that exactly enforces convexity, frame-indifference, and evenness in the second strain-rate invariant via symmetrization, alleviating the numerical instabilities that occur at small strain rates in standard formulations. By weakly enforcing zero gradients in the origin of the potential via penalization, ICNNE also satisfies the dissipativity property. The loss objective and its gradient are computed using a combination of finite element and neural network methods, coupling the Firedrake and PyTorch libraries. We evaluate on four problems spanning compressible and incompressible regimes: compressible Navier-Stokes, Herschel-Bulkley yield-stress flow, Hibler's viscous-plastic sea-ice model, and discrete-element-method data with no known constitutive law. We show that the learned potentials (i) recover ground-truth physics where it is known, (ii) transfer accurately to geometries unseen during training, and (iii) succeed in regimes where unstructured methods diverge. These results suggest that embedding mathematical structure into the parameterization, rather than into the loss, is a robust path to learning physics from indirect data.