用预训练+微调提升三维气动预测精度,大幅减少数据需求。
Towards a Foundation-Model Paradigm for Aerodynamic Prediction in Three-dimensional Design

- 先在海量几何数据上预训练,再用少量特定样本微调。
- 仅需450个样本即达0.36%误差,比从零训练降低84.2%错误率。
- 适合资源有限但需快速迭代气动设计的工程师和研究者。
高精度机器学习模型对加速外形优化至关重要,但复杂三维构型因训练数据生成成本高而难以构建。本文提出一种高效构建代理模型的方法:先在多样几何数据上大规模预训练,再用少量特定任务样本微调。开发了基于Transformer的AeroTransformer模型,并针对大规模训练优化其结构以学习气动特性。在跨音速机翼任务中,模型在包含近3万样本、几何多样性高的SuperWing数据集上预训练,随后微调至处理以通用研究模型为基础的扰动翼型。结果表明,仅使用450个任务特定样本,该方法在表面流场预测上误差为0.36%,相比从零训练降低84.2%。系统研究了模型配置与训练策略的影响,为有限数据与算力条件下的有效训练与部署提供指导。为促进复用,相关数据集与预训练模型已在https://github.com/tum-pbs/AeroTransformer公开。同时基于预训练模型构建的交互式设计工具在线可访问:https://webwing.pbs.cit.tum.de。
原文摘要 · Abstract (English)
Accurate machine-learning models for aerodynamic prediction are essential for accelerating shape optimization, yet remain challenging to develop for complex three-dimensional configurations due to the high cost of generating training data. This work introduces a methodology for efficiently constructing accurate surrogate models for design purposes by first pre-training a large-scale model on diverse geometries and then fine-tuning it with a few more detailed task-specific samples. A Transformer-based architecture, AeroTransformer, is developed and tailored for large-scale training to learn aerodynamics. The methodology is evaluated on transonic wings, where the model is pre-trained on SuperWing, a dataset of nearly 30000 samples with broad geometric diversity, and subsequently fine-tuned to handle specific wing shapes perturbed from the Common Research Model. Results show that, with 450 task-specific samples, the proposed methodology achieves 0.36% error on surface-flow prediction, reducing 84.2% compared to training from scratch. The influence of model configurations and training strategies is also systematically studied to provide guidance on effectively training and deploying such models under limited data and computational budgets. To facilitate reuse, we release the datasets and the pre-trained models at https://github.com/tum-pbs/AeroTransformer. An interactive design tool is also built on the pre-trained model and is available online at https://webwing.pbs.cit.tum.de.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。