让神经PDE模型在新几何上仍保持物理规律与精度
Structure-Preserving Learning Improves Geometry Generalization in Neural PDEs
- 用可学习的有限元空间结合几何网格,显式建模形状影响
- 在未见过的几何上实现比传统方法更优的预测精度
- 适合需要高保真物理模拟的工程与科学场景
我们致力于构建用于科学与工程的物理基础模型,实现实时求解偏微分方程(PDEs),并在适应未见几何时保持结构与精度。为此,提出通用几何神经惠特尼形式(Geo-NeW):一种数据驱动的有限元方法。联合学习定义在底层几何上的微分算子与兼容的降维有限元空间。模型通过有限元外微分计算精确保持物理守恒律。几何以离散化网格形式输入,通过Transformer编码并作为学习有限元空间的基础。该设计将几何与边界条件显式关联到解,提供强大归纳偏置,显著提升对未见域的泛化能力。我们提出新的本构模型参数化方式,确保解的存在性与唯一性。该方法在多个稳态PDE基准上达到领先性能,并在分布外几何上显著优于传统基线。
原文摘要 · Abstract (English)
We aim to develop physics foundation models for science and engineering that provide real-time solutions to Partial Differential Equations (PDEs) which preserve structure and accuracy under adaptation to unseen geometries. To this end, we introduce General-Geometry Neural Whitney Forms (Geo-NeW): a data-driven finite element method. We jointly learn a differential operator and compatible reduced finite element spaces defined on the underlying geometry. The resulting model is solved to generate predictions, while exactly preserving physical conservation laws through Finite Element Exterior Calculus. Geometry enters the model as a discretized mesh both through a transformer-based encoding and as the basis for the learned finite element spaces. This explicitly connects the underlying geometry and imposed boundary conditions to the solution, providing a powerful inductive bias for learning neural PDEs, which we demonstrate improves generalization to unseen domains. We provide a novel parameterization of the constitutive model ensuring the existence and uniqueness of the solution. Our approach demonstrates state-of-the-art performance on several steady-state PDE benchmarks, and provides a significant improvement over conventional baselines on out-of-distribution geometries.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。