Aug, 2023

突破边界:基于可扩展物理信息神经偏微分方程求解器的分布式域分解

TL;DRMosaic Flow 是一种新颖的域分解方法,通过推理利用在小域上预训练的网络,在大域上纯粹进行偏微分方程求解,实现高度的可重用性。本文介绍了 Mosaic Flow 的端到端并行化,将数据并行训练和域并行化相结合,用于大规模问题的推理。通过优化网络架构和数据并行训练,我们将学习拉普拉斯算子的训练时间显著缩短为 32 个 GPU 上的几分钟。此外,我们的分布式域分解算法使得在 32 个 GPU 上能够在域尺度大于训练域 4096 倍的情况下解决拉普拉斯方程,展示了较强的扩展性同时保持精度。Mosaic Flow 的可重用性结合分布式内存算法所实现的性能改进,使其成为建模复杂物理现象和加速科学发现的有前景工具。