FSAN: Flow State Attention Network for Aerodynamic Prediction
Organizations: Shanghai Key Laboratory of Scalable Computing and Systems · School of Computer Science, Shanghai Jiao Tong University
Abstract
Accurate aerodynamic prediction is critical for designing fuel-efficient and safe transportation systems such as aircraft and automobiles, yet traditional computational fluid dynamics (CFD) simulations remain computationally expensive and expertise-intensive, severely limiting their use in iterative design and real-time analysis. Existing deep learning surrogates suffer from two major limitations: (i) they are evaluated on datasets with narrow flow-condition ranges, leaving their performance under complex flow conditions undemonstrated; (ii) they treat global flow conditions as a single vector injected uniformly across all surface points, ignoring that different geometric regions experience distinct local flow phenomena, which degrades prediction accuracy under complex flow conditions. To address these limitations, we propose the Flow State Attention Network (FSAN). FSAN separately encodes point cloud and flow conditions, then partitions the geometry into multiple flow states via learnable soft assignments, and uses flow features to update these state representations, which in turn influence point cloud features through state changes. This enables fine-grained, state-specific interaction between geometry and flow information. Extensive experiments on two well-recognized aerodynamic benchmarks demonstrate that FSAN achieves the highest accuracy among the methods compared in this work at a higher computational cost. On Emmi-Wing, FSAN reduces the Relative L2 (REL-L2) error by over 20% compared to the strongest baseline (Transolver), and on DrivAerNet++, it achieves a 10% reduction compared to the strongest baseline (AdaField). These results establish FSAN as a promising neural surrogate on public benchmarks with diverse flow conditions and geometries.
Figures & tables
| Dataset | Method | MSE | MAE | REL-L1 | REL-L2 |
|---|---|---|---|---|---|
| ( ) | ( ) | (%) | (%) | ||
| Emmi-Wing | PointTransformerV3 | ||||
| Transolver | |||||
| AdaField | |||||
| FSAN (Ours) | |||||
| DrivAerNet++ | PointTransformerV3 |
| Model | Params | Training Time | Inference Time | Train Peak Mem. | Infer. Peak Mem. |
|---|---|---|---|---|---|
| (M) | (s / batch) | (s / batch) | (GB) | (GB) | |
| PointTransformerV3 | 46.17 | 0.165 0.004 | 0.050 0.000 | 2.20 | 0.47 |
| Transolver | 18.34 | 0.122 0.002 | 0.037 0.000 | 3.85 | 0.17 |
| AdaField | 247.09 | 0.978 0.021 | 0.455 0.004 | 8.63 | 1.34 |
| FSAN (Ours) | 15.74 | 0.339 0.001 | 0.153 0.000 | 11.53 | 0.99 |
| Variant | MSE ( ) | MAE ( ) | REL-L1 (%) | REL-L2 (%) |
|---|---|---|---|---|
| FSAN (flow state attention) | 0.297 | 0.456 | 1.40 | 2.59 |
| FSAN + FiLM | 0.334 | 0.501 | 1.65 | 2.83 |
| FSAN + AdaLN-Zero | 0.325 | 0.499 | 1.61 | 2.80 |
| FSAN + Cross-Attn | 0.374 | 0.602 | 1.87 | 3.16 |
| FSAN + Physics-Attn | 0.298 | 0.484 | 1.49 | 2.68 |
Appendix figures & tables21 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Domain | 3D? | Flow Condition | Geometric Diversity |
|---|---|---|---|---|
| DrivAerNet++ | Automotive | Yes | Fixed velocity | High |
| ( Elrefaie et al., 2024 ) | (30 m/s). | (8,000+ shapes). | ||
| Emmi-Wing | Aircraft | Yes | Variable AoA, | Moderate |
| ( Paischer et al., 2025 ) | Ma, Re. | (parameterized wings). | ||
| DrivAerML | Automotive | Yes | Variable speed | Moderate. |
| ( Ashton et al., 2024b ) | (20–50 m/s). |
| Status | Method | Rationale |
|---|---|---|
| Included | PointTransformerV3 ( Wu et al., 2024b ) | Public implementation; flow conditions injected via coordinate concatenation (same protocol as other geometry-only models). |
| Transolver ( Wu et al., 2024a ) | Public implementation; physical attention mechanism; parameter count increased to 18M for capacity matching. | |
| AdaField ( Zou et al., 2026 ) | Public implementation; natively supports flow-conditioned adapters for surface pressure prediction. | |
| Excluded | FIGConvNet ( Choy et al., 2025 ) | No native flow condition input; concatenation would cause spatial misreading of physical parameters. |
| DoMINO ( Ranade et al., 2025 ) | Single-parameter design (velocity only); extending to multi-parameter conditions (AoA, Ma, Re) would require architectural modification of the parameter encoder. | |
| TripNet ( Chen et al., 2025 ) | Code not publicly available; triplane encoding not validated under multi-parameter flow conditions. |
| Model | Params | MSE | MAE | Rel L2 | Rel L1 |
|---|---|---|---|---|---|
| AB-UPT | 8.9M | 0.0113 | 0.0434 | 0.0762 | 0.0568 |
| FSAN | 7.9M | 0.0102 | 0.0421 | 0.0747 | 0.0542 |
| Conditioning | MSE | MAE | REL-L1 | REL-L2 |
|---|---|---|---|---|
| ( ) | ( ) | (%) | (%) | |
| Concatenation | 0.375 | 0.517 | 1.58 | 3.24 |
| FiLM | 0.452 | 0.613 | 1.88 | 3.68 |
| AdaLN-Zero | 1.047 | 1.048 | 3.23 | 6.44 |
| Model | Params | Training Time | Inference Time | Train Peak Mem. | Infer. Peak Mem. |
|---|---|---|---|---|---|
| (M) | (s / batch) | (s / batch) | (GB) | (GB) | |
| PointTransformerV3 | 46.16 | 0.172 | 0.053 | 2.17 | 0.47 |
| Transolver | 18.34 | 0.123 | 0.038 | 3.85 | 0.17 |
| AdaField | 247.08 | 0.977 | 0.465 | 8.63 | 1.34 |
| FSAN (Ours) | 15.74 | 0.338 | 0.146 | 11.53 | 0.99 |
| Model | 8,192 pts | 16,384 pts | 32,768 pts | 65,536 pts | 131,072 pts | 262,144 pts | Full surface |
|---|---|---|---|---|---|---|---|
| (ms / GB) | (ms / GB) | (ms / GB) | (ms / GB) | (ms / GB) | (ms / GB) | (s / batches) | |
| PointTransformerV3 | 49.94 / 0.47 | 51.43 / 0.72 | 57.27 / 1.26 | 70.99 / 2.34 | 100.35 / 4.50 | 161.59 / 8.82 | 0.31 / 3 |
| Transolver | 36.73 / 0.17 | 70.04 / 0.25 | 141.55 / 0.42 | 291.61 / 0.77 | 580.76 / 1.46 | 1171.78 / 2.84 | 1.67 / 1 |
| AdaField | 453.82 / 1.34 | 579.00 / 1.96 | 1004.70 / 5.01 | / | / | / | 11.57 / 12 |
| FSAN (Ours) | 152.90 / 0.99 | 396.81 / 1.86 | 1182.81 / 4.14 | / | / | / | 13.22 / 12 |
| Module | 8192 pts | 16384 pts | 32768 pts |
|---|---|---|---|
| Local attention layer | 90.1% / 81.4% | 95.4% / 87.2% | 97.6% / 91.5% |
| Flow condition encoder | 1.9% / 0.0% | 0.8% / 0.0% | 0.3% / 0.0% |
| Flow state attention | 3.2% / 0.0% | 1.3% / 0.0% | 0.4% / 0.0% |
| Others | 4.8% / 18.6% | 2.5% / 12.8% | 1.6% / 8.5% |
| Points | MSE | MAE | REL-L1 | REL-L2 |
|---|---|---|---|---|
| ( ) | ( ) | (%) | (%) | |
| 1,024 | 1.927 | 2.417 | 7.77 | 9.62 |
| 2,048 | 0.747 | 1.216 | 3.86 | 5.46 |
| 4,096 | 0.365 | 0.607 | 1.88 | 3.18 |
| 8,192 | 0.297 | 0.456 | 1.40 | 2.59 |
| 16,384 | 0.330 | 0.566 | 1.75 | 2.93 |
| Evaluation | MSE | MAE | REL-L1 | REL-L2 |
|---|---|---|---|---|
| ( ) | ( ) | (%) | (%) | |
| Sampled (8,192 pts) | 0.297 | 0.456 | 1.40 | 2.59 |
| Full-surface (stitched) | 0.298 | 0.457 | 1.40 | 2.63 |
| Fig. | Method | Clipped | Max | P95 |
|---|---|---|---|---|
| (%) | ||||
| 6 | PointTransformerV3 | 8.8 | 2.17 | 0.029 |
| Transolver | 1.0 | 1.38 | 0.006 | |
| AdaField | 1.4 | 0.90 | 0.007 | |
| FSAN (Ours) | 1.0 | 0.38 | 0.006 | |
| 7 | PointTransformerV3 | 18.6 | 2.32 | 0.060 |
| Case | S1 | S2 | S3 | S4 | S5 | S6 | S7 | S8 |
|---|---|---|---|---|---|---|---|---|
| 1 | 1.15 | 1.31 | 1.07 | 0.45 | 0.24 | 2.17 | 0.15 | 0.78 |
| 2 | 1.11 | 1.47 | 1.15 | 0.33 | 0.14 | 1.97 | 0.09 | 0.58 |
| 3 | 1.10 | 1.23 | 1.46 | 0.36 | 0.22 | 1.95 | 0.12 | 0.81 |
Neural Surrogate Modeling
Neural surrogate modeling uses learned models to approximate costly simulations, enabling faster prediction, design, control, and inference.
See all 401 papers