该研究探讨了利用任意维度机器学习模型(如图神经网络)为偏微分方程求解器提供热启动的方法。作者从理论上给出了在特定对称性条件下,低维训练的PDE求解器可零样本直接迁移至高维问题的数学条件;当方程满足对称性但初始数据不满足时,该理论可指导对低维求解器进行热启动以求解高维PDE。研究在热方程、Burgers方程和可压缩Navier-Stokes方程上验证了该方法,在零样本和高维常规训练场景下均提升了性能。例如,仅用2D Navier-Stokes数据训练的代理模型在3D测试数据上的表现优于用3D数据训练的基线模型,且仅消耗12%的浮点运算量和20%的数据量。这一结果说明,借助对称性实现跨维度迁移可显著降低高维PDE求解的计算与数据成本。
| 图神经网络 (GNN) | 一种能处理图结构数据的深度学习模型,可自然适应不同大小和维度的输入。 |
| 偏微分方程 (PDE) | 描述物理量随空间和时间变化的数学方程,常用于建模连续介质和场问题。 |
| 零样本 (zero-shot) | 模型无需在目标维度数据上训练,直接应用于新维度任务的能力。 |
| 热启动 (warm-starting) | 利用低维求解器的结果作为高维求解的初始猜测,以加速收敛或提升性能。 |
| Navier-Stokes方程 | 描述流体运动的基本方程,包括可压缩和不可压缩形式,广泛应用于流体力学。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅