突破工业级几何模拟的计算瓶颈,实现超1.6亿网格高精度预测。
Transolver-3: Scaling Up Transformer Solvers to Industrial-Scale Geometries
- 通过矩阵乘法优化与几何切片分块,降低内存开销。
- 支持超过1.6亿网格的仿真,三类基准任务表现优异。
- 适合航空航天、汽车等复杂工程场景的高保真模拟需求。
深度学习已成为求解偏微分方程(PDE)的神经代理建模工具,即神经PDE求解器。然而,由于高分辨率网格带来的巨大内存开销,将此类求解器扩展到包含超过10^8个单元的工业级几何结构仍面临根本性挑战。本文提出Transolver-3,作为Transolver系列的新成员,是一个专为高保真物理仿真设计的可扩展框架。为弥合有限GPU容量与复杂工程任务分辨率需求之间的差距,我们引入两项关键架构优化:利用矩阵乘法结合律实现更快的切片与逆切片操作,以及基于几何切片分块的物理状态计算分区策略。结合在原始高分辨率网格随机子集上进行的摊销训练策略和推理阶段的物理状态缓存技术,Transolver-3实现了对工业级网格的高保真场预测。大量实验表明,Transolver-3可处理超过1.6亿个网格单元,在三个具有挑战性的仿真基准任务中表现卓越,涵盖飞机与汽车设计任务。代码已开源:https://github.com/thuml/Transolver-3。
原文摘要 · Abstract (English)
Deep learning has emerged as a transformative tool for the neural surrogate modeling of partial differential equations (PDEs), known as neural PDE solvers. However, scaling these solvers to industrial-scale geometries with over $10^8$ cells remains a fundamental challenge due to the prohibitive memory complexity of processing high-resolution meshes. We present Transolver-3, a new member of the Transolver family as a highly scalable framework designed for high-fidelity physics simulations. To bridge the gap between limited GPU capacity and the resolution requirements of complex engineering tasks, we introduce two key architectural optimizations: faster slice and deslice by exploiting matrix multiplication associative property and geometry slice tiling to partition the computation of physical states. Combined with an amortized training strategy by learning on random subsets of original high-resolution meshes and a physical state caching technique during inference, Transolver-3 enables high-fidelity field prediction on industrial-scale meshes. Extensive experiments demonstrate that Transolver-3 can handle meshes with over 160 million cells, achieving impressive performance across three challenging simulation benchmarks, including aircraft and automotive design tasks. Code is available at https://github.com/thuml/Transolver-3.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。