Surrogate neural network-based partial differential equation (PDE) solvers have the potential to solve PDEs in an accelerated manner, but they are largely limited to systems featuring fixed domain sizes, geometric layouts, and boundary conditions. We propose Specialized Neural Accelerator-Powered Domain Decomposition Methods (SNAP-DDM), a DDM-based approach to PDE solving in which subdomain problems containing arbitrary boundary conditions and geometric parameters are accurately solved using an ensemble of specialized neural operators. We tailor SNAP-DDM to 2D electromagnetics and fluidic flow problems and show how innovations in network architecture and loss function engineering can produce specialized surrogate subdomain solvers with near unity accuracy. We utilize these solvers with standard DDM algorithms to accurately solve freeform electromagnetics and fluids problems featuring a wide range of domain sizes.
翻译:基于替代神经网络的偏微分方程(PDE)求解器有望以加速方式求解偏微分方程,但其应用在很大程度上局限于具有固定域尺寸、几何布局和边界条件的系统。我们提出了专用神经加速器驱动的区域分解方法(SNAP-DDM),这是一种基于DDM的PDE求解方法,其中包含任意边界条件和几何参数的子域问题通过一组专用神经算子集成进行精确求解。我们将SNAP-DDM定制于二维电磁学和流体流动问题,并展示了网络架构与损失函数工程方面的创新如何能够产生精度接近单位值的专用替代子域求解器。我们利用这些求解器结合标准DDM算法,精确求解了涉及多种域尺寸的自由形态电磁学和流体问题。