arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.16014cs.LGcs.AI
收藏

OmniArch:构建面向科学计算的基础模型

OmniArch: Building Foundation Model For Scientific Computing

Tianyu Chen, Haoyi Zhou, Ying Li, Hao Wang, Chonghan Gao, Rongye Shi, Shanghang Zhang, Jianxin Li

AI总结:

本文提出首个面向多尺度与多物理场科学计算的基础模型OmniArch,通过统一架构结合Fourier编解码器、Transformer主干与PDE-Aligner,在PDEBench上实现1D-2D-3D联合预训练并创下性能新基准,展现卓越的零样本适应能力。

摘要:

基础模型已彻底改变了语言建模,然而这一成功能否在科学计算中复制仍有待探索。我们提出OmniArch,这是首个旨在以物理对齐方式求解多尺度与多物理场科学计算问题的原型。我们用统一架构应对了全部三个挑战。其预训练阶段包含一个Fourier Encoder-decoder,用于消除分离维度间的不一致性,以及一个Transformer主干,通过时间动态整合物理量;此外,新颖的PDE-Aligner在灵活条件下执行物理信息微调。据我们所知,我们首次在PDEBench上进行1D-2D-3D联合预训练,不仅为1D、2D和3D PDE设定了新的性能基准,还通过上下文和零样本学习方法展现了对新物理现象的卓越适应能力,从而支持现实工程应用与前瞻性物理发现。

英文摘要:

Foundation models have revolutionized language modeling, while whether this success is replicated in scientific computing remains unexplored. We present OmniArch, the first prototype aiming at solving multi-scale and multi-physics scientific computing problems with physical alignment. We addressed all three challenges with one unified architecture. Its pre-training stage contains a Fourier Encoder-decoder fading out the disharmony across separated dimensions and a Transformer backbone integrating quantities through temporal dynamics, and the novel PDE-Aligner performs physics-informed fine-tuning under flexible conditions. As far as we know, we first conduct 1D-2D-3D united pre-training on the PDEBench, and it sets not only new performance benchmarks for 1D, 2D, and 3D PDEs but also demonstrates exceptional adaptability to new physics via in-context and zero-shot learning approaches, which supports realistic engineering applications and foresight physics discovery.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑