arXiv:2607.08025cs.LG2026-07

让300万节点以上的物理仿真加速,突破显存瓶颈

PGD-NO: A Neural Operator with Precomputed Geometry Decomposition for 3D Million-scale Physics Simulations

论文配图:PGD-NO: A Neural Operator with Precomputed Geometry Decomposition for 3D Million-scale Physics Simulations
图 1 · 摘自论文原文
  • 预计算几何分解,把复杂几何处理挪到训练前
  • 支持超1000万节点网格,内存线性增长不爆显存
  • 适合需要高精度工业仿真、追求可解释性的场景

尽管神经微分方程求解器在加速工程仿真方面展现出巨大潜力,但现有架构仍受限于高内存消耗和单节点瓶颈,最大可处理网格分辨率受单个计算单元显存严格限制。为此,我们提出PGD-NO,一种带有预计算几何分解的神经算子,将几何编码的计算开销移至确定性的预计算阶段。通过迭代几何分解算法提取几何特征令牌,模型实现了特征提取与求解查询的解耦。该架构支持线性内存扩展,可在超过1000万节点的网格上实现高保真学习,而传统架构在此规模通常遭遇内存耗尽。PGD-NO在多种工业基准测试中表现优异,且通过注意力机制提供内在可解释性。有效突破传统网格尺寸限制,为下一代大规模、高保真工业设计应用提供高效可靠解决方案。

原文摘要 · Abstract (English)

While neural PDE solvers have demonstrated significant potential for accelerating engineering simulations, existing architectures remain constrained by high memory consumption and the single node bottleneck, where the maximum processable mesh resolution is strictly limited by the VRAM of a single compute unit. To address these challenges, we propose PGD-NO, a neural operator with Precomputed Geometry Decomposition, that relocates the computational overhead of geometric encoding to a deterministic pre-computation phase. By utilizing an iterative geometry decomposition algorithm to extract geometry tokens, our model decouples feature extraction from solution querying. This architecture enables linear memory scalability, allowing high fidelity learning on meshes exceeding 10 million nodes, a scale where existing architectures typically encounter memory exhaustion. PGD-NO demonstrates competitive predictive accuracy across diverse industrial benchmarks and provides intrinsic interpretability through attention mechanisms. By effectively overcoming traditional mesh-size constraints, PGD-NO offers a robust and efficient solution for the next generation of large-scale, high-fidelity industrial design applications.

神经算子物理仿真大网格可解释性

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