arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.09113cs.LG
收藏

时间注意力在网格约化空间中的物理预测

Predicting Physics in Mesh-reduced Space with Temporal Attention

Tufts University(塔夫茨大学);University of Notre Dame(圣母大学);DeepMind(深度思维)

Xu Han, Han Gao, Tobias Pfaff, Jian-Xun Wang, Li-Ping Liu

摘要:

基于图的下一步预测模型最近在不规则网格上建模复杂的高维物理系统方面非常成功。然而,由于它们的时间注意力跨度短,这些模型遭受误差积累和漂移。在本文中,我们提出了一种新的方法,通过transformer风格的时间注意力模型捕获长期依赖关系。我们引入了一个编码器-解码器结构来总结功能并创建系统状态的紧凑网格表示,以允许时间模型以内存高效的方式对低维网格表示进行操作。我们的方法在几个复杂的流体动力学预测任务上优于竞争性GNN基线,从声波冲击到血管流动。我们展示了稳定的推出,而不需要训练噪声,并显示完美的相位稳定的预测,即使是很长的序列。更广泛地说,我们相信我们的方法为将基于注意力的序列模型的好处用于解决高维复杂物理任务铺平了道路。

英文摘要:

Graph-based next-step prediction models have recently been very successful in modeling complex high-dimensional physical systems on irregular meshes. However, due to their short temporal attention span, these models suffer from error accumulation and drift. In this paper, we propose a new method that captures long-term dependencies through a transformer-style temporal attention model. We introduce an encoder-decoder structure to summarize features and create a compact mesh representation of the system state, to allow the temporal model to operate on a low-dimensional mesh representations in a memory efficient manner. Our method outperforms a competitive GNN baseline on several complex fluid dynamics prediction tasks, from sonic shocks to vascular flow. We demonstrate stable rollouts without the need for training noise and show perfectly phase-stable predictions even for very long sequences. More broadly, we believe our approach paves the way to bringing the benefits of attention-based sequence models to solving high-dimensional complex physics tasks.

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

↑