Physics Informed Differentiable Solvers for Learning Parametric Solution Manifolds in Heterogeneous Physical Systems

TL;DR

This paper introduces a physics-informed neural network framework that learns a continuous solution manifold for parameterized PDEs, enabling efficient modeling of heterogeneous systems with a single training run and applications in flow and uncertainty quantification.

Contribution

It presents a novel differentiable solver reformulation of PINNs that learns parametric solution manifolds for PDEs, incorporating autoencoders for complex heterogeneity representation.

Findings

Accurate, mass-conserving flow solutions achieved.

Single training run suffices for multiple parameter instances.

Supports efficient uncertainty quantification.

Abstract

Learning the full family of solutions to parameterized partial differential equations (PDEs) is a central challenge to our ability to model the behavior of heterogeneous systems, with a variety of fundamental and application-oriented implications in fields such as hydrogeology where system properties exhibit significant (and often uncertain) spatial heterogeneity. We address this by reformulating a Physics-Informed Neural Network (PINN) as a differentiable solver that learns the continuous solution manifold for steady-state Darcy flow. Our framework requires only a single training run, circumventing the need for costly re-training for each new parameter instance. Its versatility is demonstrated through two representations of spatially heterogeneous hydraulic conductivity fields: a direct analytical form and a novel data-driven formulation resting on an autoencoder to create a low-dimensional latent encoding. A key innovation is the integration of the differentiable decoder into the physics-informed loss function, enabling on-the-fly reconstruction of complex conductivity fields via automatic differentiation. The approach yields accurate, mass-conserving flow solutions and supports efficient uncertainty quantification, providing a general methodology for physics-constrained data-driven modeling of heterogeneous systems.