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

Transolver++:面向百万级几何的精确偏微分方程神经求解器

Transolver++: An Accurate Neural Solver for PDEs on Million-Scale Geometries

School of Software, BNRist, Tsinghua University. Huakun Luo < >

Huakun Luo, Haixu Wu, Hang Zhou, Lanxiang Xing, Yichen Di, Jianmin Wang, Mingsheng Long

AI总结:

Transolver++通过优化的并行框架和局部自适应机制,首次实现单GPU百万级网格点的PDE求解,并在标准基准和工业仿真中显著提升精度与性能。

摘要:

尽管深度模型已被广泛探索用于求解偏微分方程(PDE),但以往工作主要局限于最多数万个网格点的数据,远未达到涉及复杂几何的工业仿真所需的百万点规模。为了推动神经PDE求解器走向真实工业应用,我们提出了Transolver++,这是一种高度并行且高效的神经求解器,能够在百万级几何上精确求解PDE。在先前通过Transolver学习物理状态的进展基础上,Transolver++进一步配备了极度优化的并行框架和局部自适应机制,以高效地从海量网格点中捕获清晰的物理状态,成功应对了扩大输入网格规模时在计算和物理学习方面的棘手挑战。Transolver++首次将单GPU输入容量提升至百万级点,并且能够通过增加GPU以线性复杂度持续扩展输入规模。实验上,Transolver++在六个标准PDE基准上取得了13%的相对提升,并在百万级高保真工业仿真中实现了超过20%的性能增益,这些仿真的规模比以往基准大100倍,涵盖汽车和3D飞机设计。

英文摘要:

Although deep models have been widely explored in solving partial differential equations (PDEs), previous works are primarily limited to data only with up to tens of thousands of mesh points, far from the million-point scale required by industrial simulations that involve complex geometries. In the spirit of advancing neural PDE solvers to real industrial applications, we present Transolver++, a highly parallel and efficient neural solver that can accurately solve PDEs on million-scale geometries. Building upon previous advancements in solving PDEs by learning physical states via Transolver, Transolver++ is further equipped with an extremely optimized parallelism framework and a local adaptive mechanism to efficiently capture eidetic physical states from massive mesh points, successfully tackling the thorny challenges in computation and physics learning when scaling up input mesh size. Transolver++ increases the single-GPU input capacity to million-scale points for the first time and is capable of continuously scaling input size in linear complexity by increasing GPUs. Experimentally, Transolver++ yields 13% relative promotion across six standard PDE benchmarks and achieves over 20% performance gain in million-scale high-fidelity industrial simulations, whose sizes are 100$\times$ larger than previous benchmarks, covering car and 3D aircraft designs.

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

↑