High-fidelity Three-Dimensional Flow Field Reconstruction Driven by Turbulent Fluid Dynamics
Organizations: State Key Lab of Brain-Machine Intelligence, Zhejiang University, Hangzhou, Zhejiang, China · Shanghai Innovation Institute, Shanghai, China · Zhejiang Key Laboratory of Industrial Intelligence and Digital Twin, Eastern Institute of Technology, Ningbo, Zhejiang, China · Zhejiang Sci-Tech University, Hangzhou, Zhejiang, China
Abstract
High-fidelity three-dimensional turbulent flow fields are essential for scientific and engineering analyses, yet obtaining them through direct numerical simulation or high-resolution measurements remains expensive. Super-resolution reconstruction from coarse-grained data therefore offers a promising alternative, but remains fundamentally challenging because fine-scale structures are intrinsically three-dimensional and dynamically coupled across scales. Existing data-driven approaches still struggle to recover multiscale volumetric structures and their coherent temporal evolution. To address these limitations, we propose a turbulence-aware framework for time-resolved three-dimensional flow reconstruction. The framework couples a multiscale neural operator, designed to recover volumetric structures, with a state space model to promote temporal consistency in the reconstructed dynamics. In addition, Navier-Stokes residuals are incorporated as soft constraints to regularize the reconstruction toward physically plausible dynamics. A mixture-of-experts mechanism is also introduced to support unsupervised adaptation to unknown coarse-graining. Across evaluations on canonical turbulent systems and atmospheric reanalysis data, the proposed framework reduces reconstruction errors relative to learning-based and classical baselines, while better preserving multiscale spatial structure and coherent temporal evolution. These results validate the effectiveness of the proposed framework in reconstructing high-fidelity three-dimensional flow fields from coarse-grained data.
Figures & tables
| Methods | Coarse- graining | FIT | MHDT | HBDT | |||
| 4 | 8 | 4 | 8 | 8 | 16 | ||
| Tricubic | Box/ Favre | ||||||
| RRDB | 0.0678 | 0.0940 | |||||
| RCAN | 0.0415 | 0.0262 | |||||
| EDSR | 0.2250 | ||||||
| FNO | |||||||
| nRMSE | SSIM | PSNR (dB) | ||
| 0 | 0.3 | 0.0817 | 0.9402 | 43.32 |
| 0.001 | 0.3 | 0.0807 | 0.9415 | 43.44 |
| 0.01 | 0.3 | 0.0773 | 0.9463 | 43.82 |
| 0.1 | 0.3 | 0.1130 | 0.9022 | 40.45 |
| 0.01 | 0.1 | 0.0789 | 0.9438 | 43.64 |
| 0.01 | 1 | 0.0841 | 0.9372 | 43.17 |
| nRMSE | SSIM | PSNR (dB) | ||
| 0 | 0.3 | 0.1233 | 0.7957 | 42.20 |
| 0.0001 | 0.3 | 0.1232 | 0.7960 | 42.20 |
| 0.001 | 0.3 | 0.1173 | 0.8093 | 42.61 |
| 0.01 | 0.3 | 0.1216 | 0.7985 | 42.32 |
| 0.1 | 0.3 | 0.1810 | 0.6722 | 38.66 |
| 0.001 | 0.1 | 0.1174 | 0.8090 | 42.59 |
| Dataset | Scale | Reference forcing | After filtering | RMS retained |
| FIT | ||||
| FIT | ||||
| MHDT | ||||
| MHDT |
| Model | Divergence RMS | nRMSE |
| HCP | 0.0779 | |
| w/o HCP | 0.0773 |
| System | Velocity | Pressure | Extra variable |
| FIT | ✓ | ✓ | – |
| MHDT | ✓ | ✓ | Magnetic field |
| HBDT | ✓ | ✓ | Density |
| Category | Specification |
| Reconstructed variables | Velocity (3 components), pressure |
| Governing model | Incompressible Navier–Stokes with large-scale forcing |
| Transport coefficient | Kinematic viscosity |
| Spatial grid spacing | |
| Temporal resolution |
| Category | Specification |
| Reconstructed variables | Velocity , magnetic field , pressure |
| Governing model | Incompressible MHD equations |
| Transport coefficients | Kinematic viscosity |
| Magnetic diffusivity | |
| Spatial grid spacing | |
| Temporal resolution |
| Category | Specification |
| Reconstructed variables | Density , velocity , pressure |
| Governing model | Variable-density, low-Mach-number Navier–Stokes |
| with buoyancy production term | |
| Control parameters | Reynolds number |
| Froude number | |
| Schmidt number |
| System | Samples (train/val/test/total) | Storage |
| FIT | 653/218/218/1089 | 70 GB |
| MHDT | 707/236/236/1179 | 132 GB |
| HBDT | 822/274/274/1370 | 108 GB |
| Factor | Coarse-gr. | Method | #Params. (M) | SSIM↑ | PSNR↑ | nRMSE↓ | fRMSE↓ |
| 4 | Box | Tricubic | – | ||||
| RRDB | |||||||
| RCAN | 0.9672 | 51.92 | 0.0415 | 10.23 | |||
| EDSR | |||||||
| FNO | |||||||
| PhySR |
| Factor | Coarse-gr. | Method | #Params. (M) | SSIM↑ | PSNR↑ | nRMSE↓ | fRMSE↓ |
| 4 | Box | Tricubic | – | ||||
| RRDB | 0.9482 | 47.34 | 0.0678 | 5.92 | |||
| RCAN | |||||||
| EDSR | |||||||
| FNO | |||||||
| PhySR |
| Factor | Coarse-gr. | Method | #Params. (M) | SSIM↑ | PSNR↑ | nRMSE↓ | fRMSE↓ |
| 8 | Favre | Tricubic | – | ||||
| RRDB | |||||||
| RCAN | 0.9800 | 61.57 | 0.0262 | 1.27 | |||
| EDSR | |||||||
| FNO | |||||||
| PhySR |
| Method | Noise magnitude | nRMSE | ||||
| Ours | 0.0345 | 0.0355 | 0.0400 | 0.0453 | 0.0587 | 0.0242 |
| SwinIR | 0.0447 | 0.0479 | 0.0582 | 0.0684 | 0.0912 | 0.0466 |
| PhySR | 0.0548 | 0.0556 | 0.0603 | 0.0669 | 0.0851 | 0.0302 |
| DiffSR | 0.0489 | 0.0523 | 0.0692 | 0.0834 | 0.1110 | 0.0621 |
| FNO | 0.0878 | 0.0883 | 0.0915 | 0.0976 | 0.1217 | 0.0339 |
| Velocity nRMSE | Vorticity MAE | |||||
| Model | Low | Mid | High | Low | Mid | High |
| Full | 0.0122 | 0.1725 | 0.3440 | 2.8059 | 12.6611 | 10.2234 |
| w/o | ||||||
| w/o | ||||||
| Overall | Velocity nRMSE | |||
| Spatial design | nRMSE | Low | Mid | High |
| Ours | 0.0615 | 0.0122 | 0.1725 | 0.3440 |
| MS-ESRGAN | 0.0739 | 0.0139 | 0.2174 | 0.3803 |
| FIT | FIT | |||
| Model | nRMSE | tRMSE | nRMSE | tRMSE |
| SwinIR | 0.0421 | 0.0052 | 0.1499 | 0.0103 |
| DiffSR | 0.0423 | 0.0095 | 0.1440 | 0.0244 |
| PhySR | 0.0476 | 0.0060 | 0.1486 | 0.0106 |
| Ours (w/o SSM) | 0.0340 | 0.0049 | 0.1371 | 0.0101 |
| Ours | 0.0291 | 0.0041 | 0.1195 | 0.0094 |
| Temporal model | Params (M) | nRMSE | SSIM | PSNR | tRMSE | fRMSE |
| State space model | 3.7 | 0.0291 | 0.9834 | 55.09 | 0.0041 | 7.22 |
| ConvLSTM | 3.8 | 0.0306 | 0.9819 | 54.60 | 0.0044 | 7.59 |
| Causal Transformer | 3.8 | 0.0314 | 0.9809 | 54.36 | 0.0044 | 7.79 |
| FIT | MHDT | ||||||
| Params (M) | GFLOPs | nRMSE | tRMSE | nRMSE | tRMSE | ||
| 8 | 64 | 0.968 | 63.468 | 0.0295 | 4.23 | 0.0613 | 0.90 |
| 16 | 64 | 1.034 | 67.864 | 0.0291 | 4.14 | 0.0603 | 0.88 |
| 32 | 64 | 1.166 | 76.655 | 0.0290 | 4.11 | 0.0599 | 0.87 |
| 16 | 32 | 0.484 | 31.789 | 0.0305 | 4.33 | 0.0634 | 0.92 |
| 16 | 128 | 2.329 | 152.899 | 0.0287 | 4.07 | 0.0593 | 0.86 |
| Factors | Metrics | Ours | SwinIR | FNO | Bicubic |
| vCoh | 0.9925 | 0.9906 | 0.9871 | 0.9743 | |
| vRMSE | 0.4672 | 0.5167 | 0.5925 | 0.8482 | |
| vCoh | 0.9533 | 0.9442 | 0.9259 | 0.8745 | |
| vRMSE | 1.1340 | 1.2582 | 1.4174 | 1.8646 |
| Model | SSIM | nRMSE | PSNR | tRMSE | fRMSE |
| Ours | |||||
| FNO | |||||
| PhySR | |||||
| RCAN | |||||
| EDSR | |||||
| RRDB |
| Factor | Model | MAE | MSE | tRMSE | vCoh | vRMSE |
| Ours | ||||||
| FNO | ||||||
| SwinIR | ||||||
| Ours | ||||||
| FNO | ||||||
| SwinIR |
| Coarse-graining | Expert | SSIM↑ | PSNR↑ | nRMSE↓ | fRMSE↓ |
| Tricubic | Spectral | ||||
| Box | 0.9077 | 45.02 | 0.0879 | 21.44 | |
| Gaussian | |||||
| MoE (2 experts) | 0.9406 | 47.19 | 0.0673 | 16.45 | |
| MoE (3 experts) | 0.9333 | 46.48 | 0.0731 | 17.72 | |
| Lanczos | Spectral |
Computational Fluid Dynamics
Computational fluid dynamics numerically models fluid motion, turbulence, heat and mass transport, and fluid–structure interactions.
See all 275 papers