Neural PDE Solvers for Irregular Domains

Biswajit Khara,Ethan Herron,Aditya Balu,Dhruv Gamdha,Chih-Hsuan Yang,Kumar Saurabh,Anushrut Jignasu,Zhanhong Jiang,Soumik Sarkar,Chinmay Hegde,Baskar Ganapathysubramanian,Adarsh Krishnamurthy
DOI: https://doi.org/10.1016/j.cad.2024.103709
IF: 3.652
2024-04-16
Computer-Aided Design
Abstract:Highlights • Immersed boundary network predicts field PDE solution over arbitrary domains. • PDE loss to learn watertight boundary conditions imposed by complex geometries. • Single trained IBN can produce solutions to a PDE across different arbitrary shapes. • Analysis of convergence and generalization error bounds of the PDE-based loss. • Illustrate the approach on Poisson's and Navier–Stokes PDE for different geometries. Neural network-based approaches for solving partial differential equations (PDEs) have recently received special attention. However, most neural PDE solvers only apply to rectilinear domains and do not systematically address the imposition of boundary conditions over irregular domain boundaries. In this paper, we present a neural framework to solve partial differential equations over domains with irregularly shaped (non-rectilinear) geometric boundaries. Our network takes in the shape of the domain as an input (represented as a binary mask) and can generalize to novel (unseen) irregular domains; the key technical ingredient to realizing this model is a novel approach for identifying the interior and exterior of the computational grid in a differentiable manner. We also perform a careful error analysis which reveals theoretical insights into several sources of error incurred in the model-building process. Finally, we showcase various applications in 2D and 3D, along with favorable comparisons with ground truth solutions.
computer science, software engineering
What problem does this paper attempt to address?