分域求解偏微分方程,提升精度与通信效率
Non-overlapping, Schwarz-type Domain Decomposition Method for Physics and Equality Constrained Artificial Neural Networks
- 将区域分解为非重叠子域,用界面损失函数约束物理与等式条件
- 支持高波数、复值解,64子域下仍保持良好泛化能力
- 适合大规模物理信息神经网络,降低子域间通信开销
我们提出一种非重叠、Schwarz型区域分解方法,采用广义界面条件,用于偏微分方程(PDE)的正向与反向物理信息机器学习。该方法在每个子域中使用物理与等式约束人工神经网络(PECANN)。不同于原始PECANN仅依赖初边值条件约束PDE,本方法结合边界条件与控制方程,为每个子域构建唯一界面损失函数。该设计提升了子域界面参数的学习效果,同时通过延迟相邻子域间信息交换,降低通信开销。针对各子域中的约束优化问题,采用带条件自适应更新策略的增广拉格朗日法,将原问题转化为无约束对偶优化。该方法可有效求解泊松方程与亥姆霍兹方程,即使在高波数和复值解情况下亦表现良好。数值实验表明,最多达64个子域时,方法仍具稳定泛化能力。
原文摘要 · Abstract (English)
We present a non-overlapping, Schwarz-type domain decomposition method with a generalized interface condition, designed for physics-informed machine learning of partial differential equations (PDEs) in both forward and inverse contexts. Our approach employs physics and equality-constrained artificial neural networks (PECANN) within each subdomain. Unlike the original PECANN method, which relies solely on initial and boundary conditions to constrain PDEs, our method uses both boundary conditions and the governing PDE to constrain a unique interface loss function for each subdomain. This modification improves the learning of subdomain-specific interface parameters while reducing communication overhead by delaying information exchange between neighboring subdomains. To address the constrained optimization in each subdomain, we apply an augmented Lagrangian method with a conditionally adaptive update strategy, transforming the problem into an unconstrained dual optimization. A distinct advantage of our domain decomposition method is its ability to learn solutions to both Poisson's and Helmholtz equations, even in cases with high-wavenumber and complex-valued solutions. Through numerical experiments with up to 64 subdomains, we demonstrate that our method consistently generalizes well as the number of subdomains increases.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。