Mosaic Flows: A Transferable Deep Learning Framework for Solving PDEs on Unseen Domains

Abstract

Physics-informed neural networks (PINNs) are increasingly employed to replace/augment traditional numerical methods in solving partial differential equations (PDEs). While state-of-the-art PINNs have many attractive features, they approximate a specific realization of a PDE system and hence are problem-specific. That is, the model needs to be re-trained each time the boundary conditions (BCs) and domain shape/size change. This limitation prohibits the application of PINNs to realistic or large-scale engineering problems especially since the costs and efforts associated with their training are considerable. We introduce a transferable framework for solving boundary value problems (BVPs) via deep neural networks which can be trained once and used forever for various unseen domains and BCs. We first introduce genomic flow network(GFNet), a neural network that can infer the solution of a BVP across arbitrary BCson a small square domain called genome. Then, we proposed mosaic flow(MF) predictor, a novel iterative algorithm that assembles the GFNet's inferences for BVPs on large domains with unseen sizes/shapes and BCs while preserving the spatial regularity of the solution. We demonstrate that our framework can estimate the solution of Laplace and Navier-Stokes equations in domains of unseen shapes and BCs that are, respectively, 1200 and 12 times larger than the training domains. Since our framework eliminates the need to re-train models for unseen domains and BCs, it demonstrates up to 3 orders-of-magnitude speedups compared to the state-of-the-art.

23 pages, 10 figures

References in corpus (11)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Fourier Neural Operator for Parametric Partial Differential Equations
• Prediction of Aerodynamic Flow Fields Using Convolutional Neural Networks
• Deep Generative Modeling for Mechanistic-based Learning and Design of Metamaterial Systems
• Variational Physics-Informed Neural Networks For Solving Partial Differential Equations
• Understanding and mitigating gradient pathologies in physics-informed neural networks
• When and why PINNs fail to train: A neural tangent kernel perspective
• Latent Map Gaussian Processes for Mixed Variable Metamodeling
• Adaptive Neural Network-Based Approximation to Accelerate Eulerian Fluid Simulation
• Exact imposition of boundary conditions with distance functions in physics-informed deep neural networks
• Convergence rate of DeepONets for learning operators arising from advection-diffusion equations

Cited by in corpus (7)

• Deep transfer operator learning for partial differential equations under conditional shift
• In-Context Operator Learning with Data Prompts for Differential Equation Problems
• Physics-Informed Neural Networks for Parametric Compressible Euler Equations
• Local neural operator for solving transient partial differential equations on varied domains
• Multi-Fidelity Machine Learning Applied to Steady Fluid Flows
• One-Shot Transfer Learning of Physics-Informed Neural Networks
• Accelerated Gradient-based Design Optimization Via Differentiable Physics-Informed Neural Operator: A Composites Autoclave Processing Case Study