摘要
求解偏微分方程 (PDE) 是对复杂动力系统进行建模的基石。最近的进展表明,数据驱动的基于神经网络的模型在预测时空动力学方面具有巨大优势(例如,与经典数值方法相比,速度大幅提升)。然而,现有的多数神经网络模型依赖于丰富的训练数据,具有有限的推断和泛化能力,并且在复杂条件下(例如,不规则网格或几何形状、复杂边界条件、不同的 PDE 参数等)难以产生精确或可靠的物理预测。为此,我们提出了一种新的图学习方法,即物理编码消息传递图网络 (PhyMPGN),用于对给定小型训练数据集的不规则网格上的时空 PDE 系统进行建模。具体而言,我们将 GNN 集成到数值积分器中,以逼近给定 PDE 系统的时空动力学的时间推进。考虑到许多物理现象受扩散过程支配,我们进一步设计了一个可学习的拉普拉斯块,该块对离散拉普拉斯-贝尔特拉米算子进行编码,以帮助和引导 GNN 在物理可行解空间中进行学习。还设计了边界条件填充策略来提高模型收敛性和准确性。大量实验表明,PhyMPGN 能够准确地预测粗糙非结构化网格上的各种时空动力学,始终如一地取得最先进的结果,并且在很大程度上优于其他基线。