Subgrid Stress Modelling with Multi-dimensional State Space Sequence Models

TL;DR

This paper introduces a novel neural network-based subgrid stress model for Large Eddy Simulations that generalizes well across different grid sizes and Reynolds numbers, outperforming traditional models.

Contribution

The paper proposes a S4ND Unet neural network architecture for subgrid stress modeling that generalizes across grid resolutions and Reynolds numbers, improving LES accuracy.

Findings

S4ND model outperforms traditional and ANN models on training grid sizes.

The model generalizes to coarser grids both a priori and a posteriori.

It remains stable at Reynolds numbers 500,000 times higher than training data.

Abstract

Large Eddy Simulations (LES) are becoming increasingly viable due to the growth in computational power the last few decades, and subgrid stress modelling plays a large role in the accuracy of LES. A new class of neural network models, S4 and S4ND models, allow for learning a continuous representation of the discrete dataset, which facilitates a principled approach to incorporating grid dependence in neural network subgrid stress modelling. A S4ND Unet neural network architecture is proposed and trained on both forced Homogeneous Isotropic Turbulence (HIT) and channel flow, where a priori, it is shown to generalize to grid spacings that are coarser than the training set grid spacings, while simpler artificial neural network (ANN) models fail. A posteriori tests on both forced HIT and channel flow indicate that the S4ND model is more accurate than traditional models and ANN-based models on grid sizes that are in the training set. The S4ND model is also able to generalize to grid sizes that are coarser than the training set a posteriori, and is more accurate than many traditional and neural network subgrid stress models. Finally, the proposed model is also evaluated on flows at increasing Reynolds numbers, where it is seen that the proposed neural network remains stable even at a Reynolds number that is 500,000 times that seen in the training set.