用分阶段多分辨率训练,24小时完成高精度汽车气动预测。
PMRT: A Training Recipe for Fast, 3D High-Resolution Aerodynamic Prediction
- 分阶段逐步提升分辨率训练,缓解数据稀缺问题
- 512x128x128分辨率速度场预测仅需24小时,成本降7倍
- 适配多数据源与不同求解器,适合工业级气动设计优化
汽车气动优化需工程师与设计师协作,但传统仿真耗时昂贵。代理模型可在训练空间内准确预测气动性能,但受限于3D问题复杂性与数据不足,难以扩展至高分辨率。本文提出渐进式多分辨率训练(PMRT),通过概率调度在训练中动态采样低、中、高分辨率数据,使U-Net在单张NVIDIA H100 GPU上24小时内完成512×128×128高分辨率速度场与阻力系数(c_d)预测,相比纯高分辨率基线成本降低7倍,精度相当。该方法可适配其他骨干网络。此外,通过条件建模,单个模型可在五个不同求解器生成的数据集上训练,包括真实世界数据集。在DrivAerML数据集上,模型达到c_d的R²=0.975,媲美文献基准,训练成本大幅下降。
原文摘要 · Abstract (English)
The aerodynamic optimization of cars requires close collaboration between aerodynamicists and stylists, while slow, expensive simulations remain a bottleneck. Surrogate models have been shown to accurately predict aerodynamics within the design space for which they were trained. However, many of these models struggle to scale to higher resolutions because of the 3D nature of the problem and data scarcity. We propose Progressive Multi-Resolution Training (PMRT), a probabilistic multi-resolution training schedule that enables training a U-Net to predict the drag coefficient ($c_d$) and high-resolution velocity fields (512 x 128 x 128) in 24 hours on a single NVIDIA H100 GPU, 7x cheaper than the high-resolution-only baseline, with similar accuracy. PMRT samples batches from three resolutions based on probabilities that change during training, starting with an emphasis on lower resolutions and gradually shifting toward higher resolutions. Since this is a training methodology, it can be adapted to other high-resolution-focused backbones. We also show that a single model can be trained across five datasets from different solvers, including a real-world dataset, by conditioning on the simulation parameters. In the DrivAerML dataset, our models achieve a $c_d$ $R^2$ of 0.975, matching literature baselines at a fraction of the training cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。