While deep learning accelerates expensive partial differential equation solving in computational fluid dynamics (CFD), existing methods like PINNs and FNOs often struggle with generalization, noise robustness, and physical consistency.
We introduce the Transformer-Mamba for Flow Field (TM4FF) framework, a physics-constrained operator learning model with three key innovations:
- a Residual Wavelet Mamba (RWM) layer for feature denoising,
- a Transformer-based attention mechanism for enhanced feature fusion,
- a physics-informed loss using Fourier derivatives to enforce the Navier-Stokes equations.
Experiments on four CFD datasets show TM4FF achieves high accuracy and robust generalization across varying flow conditions.