MORPH: Shape-agnostic PDE Foundation Models

AI4S PINNs Other AI4S
2025年09月25日
我们提出MORPH,一种对几何形状无依赖的、用于偏微分方程(PDE)的自回归基础模型。MORPH基于卷积视觉Transformer架构构建,能够无缝处理具有不同数据维度(一维至三维)、不同分辨率以及包含多个混合标量与矢量物理场的异构时空数据集。该架构结合了三个关键设计:(i)逐分量卷积,可联合处理标量和矢量通道,以捕捉局部相互作用;(ii)跨场交叉注意力机制,用于建模不同物理场之间的信息并选择性地进行传播;(iii)轴向注意力机制,沿各个空间和时间轴分别分解完整的时空自注意力,从而在保持表达能力的同时降低计算开销。我们在多种多样的异构PDE数据集上预训练了多个模型变体,并评估其在一系列下游预测任务中的迁移性能。无论是通过全模型微调,还是采用参数高效的低秩适配器(LoRA)方法,MORPH在零样本和全样本泛化场景下均优于从头训练的模型。在广泛的实验评估中,MORPH达到或超过了强基线方法以及近期最先进的模型表现。综上所述,这些能力使MORPH成为一个灵活而强大的骨干网络,可用于学习科学观测中普遍存在的异构性与多模态特性,为实现可扩展且数据高效利用的科学机器学习指明了发展方向。
We introduce MORPH, a shape-agnostic, autoregressive foundation model for partial differential equations (PDEs). MORPH is built on a convolutional vision transformer backbone that seamlessly handles heterogeneous spatiotemporal datasets of varying data dimensionality (1D--3D) at different resolutions, multiple fields with mixed scalar and vector components. The architecture combines (i) component-wise convolution, which jointly processes scalar and vector channels to capture local interactions, (ii) inter-field cross-attention, which models and selectively propagates information between different physical fields, (iii) axial attentions, which factorizes full spatiotemporal self-attention along individual spatial and temporal axes to reduce computational burden while retaining expressivity. We pretrain multiple model variants on a diverse collection of heterogeneous PDE datasets and evaluate transfer to a range of downstream prediction tasks. Using both full-model fine-tuning and parameter-efficient low-rank adapters (LoRA), MORPH outperforms models trained from scratch in both zero-shot and full-shot generalization. Across extensive evaluations, MORPH matches or surpasses strong baselines and recent state-of-the-art models. Collectively, these capabilities present a flexible and powerful backbone for learning from heterogeneous and multimodal nature of scientific observations, charting a path toward scalable and data-efficient scientific machine learning.
许愿