arXiv:2502.02414cs.LG2025-02ICML被引 98

首个可处理百万级网格的神经微分方程求解器,突破工业仿真规模瓶颈。

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

  • 基于优化并行框架与局部自适应机制,实现大规模网格高效求解。
  • 在六项标准测试中提升13%,百万级工业仿真性能提升超20%。
  • 适合需要高保真复杂几何仿真的工业领域研究者使用。

尽管深度模型已被广泛用于求解偏微分方程(PDE),但以往工作主要局限于最多数万网格点的数据,远低于涉及复杂几何的工业仿真所需的百万级规模。为推动神经PDE求解器向真实工业应用迈进,我们提出Transolver++,一种高度并行且高效的神经求解器,可准确求解百万级网格上的PDE。在先前基于Transolver学习物理状态的基础上,Transolver++进一步引入极优化的并行框架与局部自适应机制,高效捕捉海量网格点中的动态物理状态,成功应对大规模输入时计算与物理学习的棘手挑战。Transolver++首次将单卡输入容量扩展至百万级点,并可通过增加GPU实现输入规模的线性持续扩展。实验表明,其在六项标准PDE基准上相对提升13%,在百万级高保真工业仿真中性能提升超20%,其规模比此前基准大100倍,涵盖汽车与三维飞机设计等场景。

原文摘要 · Abstract (English)

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.

PDE求解神经算子工业仿真大规模建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。