ICML2026

UniDrag: Unified Multi-Field Prediction and Robust Shape Optimization for Vehicle Aerodynamics

Ye Liu, Shouyi Liu, Ding Wang, Huiyu Yang, Ruizhe DENG, Qian Li, Yuxiao Hu, Jianghang gu, Yongzheng Liu, Quanshi Zhang, Shiyi Chen, Yuntian Chen

摘要

High-fidelity vehicle aerodynamics analysis is bottlenecked by costly CFD simulations. Neural surrogates accelerate prediction but lack inverse design capabilities, while existing generative optimization methods suffer from unstable convergence and frequent engineering constraint violations. We present UniDrag, a unified framework bridging multi-field aerodynamic prediction with robust differentiable shape optimization. Given a vehicle geometry, UniDrag predicts surface pressure, volume flow fields, drag coefficient CdC_d, and a streamwise build-up profile localizing drag contributions. Our architecture combines enhanced Physics-Sliced Attention (ePSA), Gated Expert Routing, and Modality-Protected Learning to prevent negative transfer across output modalities. At deployment, the frozen surrogate enables gradient-based optimization via Free-Form Deformation with engineering constraints. We introduce Expectation-over-Transformation to prevent adversarial exploitation of surrogate fragility. We curate a large-scale dataset of 15,000 vehicle geometries spanning four body types with GPU-accelerated LBM simulations. On this benchmark, UniDrag achieves CdC_d prediction R2R^2 of 0.937 (+7.6% over baselines) and 13.7% mean CFD-verified drag reduction with 100% success rate and only 21.3 mm average displacement (0.46% vehicle length).