Operator learning of RANS equations: a Graph Neural Network closure model

Michele Quattromini,Michele Alessandro Bucci,Stefania Cherubini,Onofrio Semeraro
2023-03-07
Abstract:The spread of machine learning (ML) techniques in combination with the availability of high-quality experimental and numerical data boosted in recent years numerous applications in fluid mechanics. Among those, examples of closure models for turbulent flows or data-assimilation based on neural networks (NN) are already numerous. However, it is well known that these techniques are prone to over-fit and necessitate an exceedingly large amount of data, unless enforcing physical constraints. We address those limitations by applying graph neural networks (GNN). This architecture is characterized by a net of nodes that can be easily adapted to unstructured meshes. Moreover, it is known GNN can show remarkable generalization capabilities as compared to standard network models. Here, we demonstrate the use of GNN by interfacing them with a finite elements (FEM) solver for the supervised learning of Reynolds-averaged Navier--Stokes equations. We consider as a test-case the flow past bluff bodies; we train the model using the wake past a cylinder, at different Reynolds numbers 40<Re<150 and resolved on different grids. The GNN model successfully predicts the Reynolds stress also in unseen cases characterized by different geometric configurations and Reynolds numbers. Interestingly, a small data-set is used for achieving these performances, thus suggesting the applications of GNN in replacement of less flexible techniques such as convolutional NN.
Fluid Dynamics
What problem does this paper attempt to address?
The paper aims to address the problem of turbulence simulation in fluid mechanics, particularly focusing on the closure models in Reynolds-Averaged Navier-Stokes (RANS) equations. Specifically, the authors attempt to overcome some limitations encountered by traditional machine learning methods when dealing with such problems, such as overfitting and the need for large amounts of training data. To achieve this goal, the research team adopted a novel approach—Graph Neural Networks (GNN). Compared to traditional Convolutional Neural Networks, GNNs can better adapt to unstructured grids and exhibit stronger generalization capabilities. By combining GNN with the Finite Element Method (FEM), the researchers developed a supervised learning framework to predict Reynolds stress in RANS equations. The main contributions of the paper include: 1. **Proposing a GNN model**: Utilizing the characteristics of GNN, a closure model capable of adapting to different geometric configurations and Reynolds numbers was constructed. 2. **Reducing data requirements**: Experiments demonstrated that GNN could achieve good prediction results even with a small dataset. 3. **Validating model performance**: The effectiveness and generalization capability of the model were proven through validation on known cases and testing on cases with unknown geometric configurations and Reynolds number ranges. In summary, this paper proposes a new framework combining GNN and FEM to solve the closure problem in RANS equations, and demonstrates the effectiveness and flexibility of this approach in handling fluid flows with complex geometries and varying Reynolds numbers.