摘要
深度模型最近已成为解决偏微分方程(PDE)的有前景工具,被称为神经网络偏微分方程求解器。虽然从模拟数据或物理信息损失函数训练的神经网络求解器能够很好地解决偏微分方程,但它们主要局限于少数几种偏微分方程,例如具有有限系数集的特定方程。这限制了神经网络求解器对不同偏微分方程的泛化能力,阻碍了它们成为数值求解器的实用替代模型。在本文中,我们提出了通用偏微分方程求解器(Unisolver),它能够通过在一个新型Transformer模型上训练多样化数据并以不同的偏微分方程为条件来解决范围广泛的偏微分方程。Unisolver并非仅仅是扩大数据和参数规模,而是源于对偏微分方程求解过程的理论分析。我们的关键发现是,偏微分方程的解从根本上受到一系列偏微分方程组件的控制,例如方程符号、系数和边界条件。受偏微分方程数学结构的启发,我们定义了一套完整的偏微分方程组件,并将它们灵活地嵌入为Transformer偏微分方程求解器的域内(例如方程符号)和点内(例如边界)条件。通过将物理洞察与最新的Transformer进展相结合,Unisolver在三个具有挑战性的大型基准测试中取得了一致的最先进的结果,展现了令人印象深刻的性能提升和良好的偏微分方程泛化能力。