Generative prediction of flow fields around an obstacle using the diffusion model

Abstract

We propose a geometry-to-flow diffusion model that utilizes obstacle shape as input to predict a flow field around an obstacle. The model is based on a learnable Markov transition kernel to recover the data distribution from the Gaussian distribution. The Markov process is conditioned on the obstacle geometry, estimating the noise to be removed at each step, implemented via a U-Net. A cross-attention mechanism incorporates the geometry as a prompt. We train the geometry-to-flow diffusion model using a dataset of flows around simple obstacles, including circles, ellipses, rectangles, and triangles. For comparison, two CNN-based models and a VAE model are trained on the same dataset. Tests are carried out on flows around obstacles with simple and complex geometries, representing interpolation and generalization on the geometry condition, respectively. To evaluate performance under demanding conditions, the test set incorporates scenarios including crosses and the characters `PKU.' Generated flow fields show that the geometry-to-flow diffusion model is superior to the CNN-based models and the VAE model in predicting instantaneous flow fields and handling complex geometries. Quantitative analysis of the accuracy and divergence demonstrates the model's robustness.

References in corpus (14)

• DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operators
• NSFnets (Navier-Stokes Flow nets): Physics-informed neural networks for the incompressible Navier-Stokes equations
• B-PINNs: Bayesian Physics-Informed Neural Networks for Forward and Inverse PDE Problems with Noisy Data
• Super-resolution reconstruction of turbulent flows with machine learning
• Enhancing Computational Fluid Dynamics with Machine Learning
• Data-driven prediction of unsteady flow fields over a circular cylinder using deep learning
• Unsupervised deep learning for super-resolution reconstruction of turbulence
• Convolutional neural network and long short-term memory based reduced order surrogate for minimal turbulent channel flow
• Machine-learning-based reduced order modeling for unsteady flows around bluff bodies of various shapes
• Grasping Extreme Aerodynamics on a Low-Dimensional Manifold
• A transformer-based synthetic-inflow generator for spatially-developing turbulent boundary layers
• Synthetic Lagrangian Turbulence by Generative Diffusion Models
• Data-driven nonlinear turbulent flow scaling with Buckingham Pi variables
• Predicting fluid-structure interaction with graph neural networks