Hostname: page-component-74656f9884-2dmjx Total loading time: 0 Render date: 2026-11-07T20:30:32.351Z Has data issue: false hasContentIssue false

Subgrid stress modelling with multidimensional state space sequence models

Published online by Cambridge University Press:  27 August 2026

Andy Wu*
Affiliation:
Department of Aeronautics and Astronautics, Stanford University , Stanford, CA, USA
Sanjiva Lele
Affiliation:
Department of Aeronautics and Astronautics, Stanford University , Stanford, CA, USA Department of Mechanical Engineering, Stanford University, Stanford, CA, USA
*
Corresponding author: Andy Wu, awu1018@stanford.edu

Abstract

Content of image described in text.

Large eddy simulations (LES) are becoming increasingly viable due to the growth in computational power over 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 U-Net neural network architecture is proposed and trained on both forced homogeneous isotropic turbulence (HIT) and channel flow. A priori, this architecture is able to extrapolate 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 extrapolate robustly to grid sizes that are coarser than the training set a posteriori, and is more accurate than many traditional 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 larger than what was seen in the training set.

Information

Type
JFM Papers
Creative Commons
Creative Common License - CCCreative Common License - BYCreative Common License - NCCreative Common License - ND
This is an Open Access article, distributed under the terms of the Creative Commons Attribution-NonCommercial-NoDerivatives licence (https://creativecommons.org/licenses/by-nc-nd/4.0/), which permits non-commercial re-use, distribution, and reproduction in any medium, provided that no alterations are made and the original article is properly cited. The written permission of Cambridge University Press or the rights holder(s) must be obtained prior to any commercial use and/or adaptation of the article.
Copyright
© The Author(s), 2026. Published by Cambridge University Press
Figure 0

Figure 1. The S4ND extension of the S4 model. A S4 model is instantiated in each direction, and then an outer product is taken to produce a continuous convolution kernel. When training and during inference, the continuous convolution kernel is discretized.

Figure 1

Figure 2. The overall neural network architecture. The whole three-dimensional (3-D) domain is inputted into the neural network, and the output is the corresponding subgrid stress at each point. For each S4ND layer, a state space dimension of eight was chosen to keep the trainable parameters comparable to existing models in the literature. The average pooling operation will downsample all spatial dimensions by a factor of two. As seen, after the final S4ND layer, the neural network architecture splits into two ‘prediction heads’ to predict the ‘magnitude’ and ‘structure’ of the subgrid stress tensor, and both prediction heads use the tensor basis (each prediction head predicts its own set of coefficients to multiply the tensor basis components). See Wu (2025) for downloadable version of the model and an example input/output.

Figure 2

Table 1. Training data.Table 1 long description.

Figure 3

Table 2. Various neural network model inputs and trainable parameters.Table 2 long description.

Figure 4

Table 3. Filter widths and energy filtered for HIT.Table 3 long description.

Figure 5

Figure 3. Spatial distribution of τ11$\tau _{11}$ for a 2-D slice of the 3-D domain for a filter width of 16 ΔDNS$\varDelta _{\textit{DNS}}$ for the forced HIT case. All numerical values are ensemble averaged over 10 ensembles, while the spatial fields are chosen randomly for visual reference. The letter r$r$ denotes the correlation coefficient. The correlation coefficients are highest for S4ND, while the other, less complex neural networks have lower correlation coefficients. The x and y axis correspond to the coordinates in the physical domain.

Figure 6

Figure 4. Figure 4 long description.Spatial distribution of −τijS¯ij$-\tau _{\textit{ij}}\bar {S}_{\textit{ij}}$ for a 2-D slice of the 3-D domain for a filter width of 16 ΔDNS$\varDelta _{\textit{DNS}}$. All numerical values are ensemble averaged over 10 ensembles, while the spatial fields are chosen randomly for visual reference. The x and y axis correspond to the coordinates in the physical domain.

Figure 7

Figure 5. The PDFs for 16 ΔDNS$\varDelta _{\textit{DNS}}$ forced HIT. As seen, the S4ND model is more able to accurately capture the distribution of τ11$\tau _{11}$ as compared with other models. The S4ND model and NN model (convolution-based) both capture more backscatter than traditional models, although it is still less than filtered DNS (FDNS). Here (a) PDF of ϵ$\epsilon$; (b) PDF of τ11$\tau _{11}$.

Figure 8

Figure 6. The RMSE and DRMSE for various neural network models evaluated on 24ΔDNS$24 \varDelta _{\textit{DNS}}$ HIT data. As seen, the S4ND is able to handle this case the best (no increase in the RMSE and DRMSE as compared with the test set at other filter widths), followed by the NN model and then the traditional TBNN, which shows a noticeable increase in both the RMSE and DRMSE when compared with the S4ND model.

Figure 9

Figure 7. The RMSE and DRMSE for various neural network models evaluated on 64ΔDNS$64 \varDelta _{\textit{DNS}}$ HIT data. As seen, the S4ND model is able to extrapolate to this more aggressive filter width without any significant increase in the RMSE and DRMSE, while the NN model has around a factor of two increase in the RMSE and a factor of 1.3 in DRMSE. Meanwhile, the traditional TBNN has a loss increase of almost a factor of four for RMSE and a loss increase of a factor of 2.5 for DRMSE.

Figure 10

Figure 8. Forced HIT spectra for various SGS models a posteriori on a grid size corresponding to a filter width of 16 ΔDNS$\varDelta _{\textit{DNS}}$.

Figure 11

Figure 9. Figure 9 long description.Forced HIT spatial, temporal, and space–time correlations for various SGS models a posteriori on a grid size corresponding to a filter width of 16 ΔDNS$\varDelta _{\textit{DNS}}$.

Figure 12

Figure 10. Channel flow streamwise spectra at z/H=0.46875$z/H = 0.46875$ for various SGS models compared with filtered DNS. Both the static Smagorinsky model and TBNN fail at predicting the overall spectra shape.

Figure 13

Figure 11. Figure 11 long description.Channel flow macroscale quantities analysed for various SGS models. The mean velocity profiles and spatial correlations for the S4ND and NN model look the closest compared with filtered DNS. As seen, the TBNN completely fails in this setting, predicting jagged spatial correlations. (a) Mean streamwise velocity profiles. (b) Spatial correlations at z/H=0.46875$z/H = 0.46875$.

Figure 14

Figure 12. Channel flow Reynolds stress profiles for various SGS models at a filter width corresponding to 16 ΔDNS$\varDelta _{\textit{DNS}}$. (a) Here R11$R_{11}$ varying as a function of wall normal distance. (b) Here R13$R_{13}$ varying as a function of wall normal distance. (c) Here R33$R_{33}$ varying as a function of wall normal distance.

Figure 15

Figure 13. Figure 13 long description.Spectra for various SGS models on a grid corresponding to an extrapolative filter width of 64ΔDNS$64 \varDelta _{\textit{DNS}}$.

Figure 16

Figure 14. Figure 14 long description.Forced HIT spatial, temporal and space–time correlations for various SGS models on a grid corresponding to an extrapolative filter width of 64ΔDNS$64 \varDelta _{\textit{DNS}}$.

Figure 17

Figure 15. Channel flow mean profiles for various SGS models on a grid corresponding to an extrapolative filter width of 64ΔDNS$64 \varDelta _{\textit{DNS}}$.

Figure 18

Figure 16. Channel flow Reynolds stress profiles for various SGS models on a grid corresponding to an extrapolative filter width of 64ΔDNS$64 \varDelta _{\textit{DNS}}$. (a) Here R11$R_{11}$ varying as a function of wall normal distance. (b) Here R13$R_{13}$ varying as a function of wall normal distance. (c) Here R33$R_{33}$ varying as a function of wall normal distance.

Figure 19

Figure 17. Figure 17 long description.Macroscale quantities analysed for forced HIT at a higher Reynolds number not seen in the training set. (a) Forced HIT spectra. (b) Forced HIT spatial correlations.

Figure 20

Figure 18. Compensated spectra for forced HIT at a higher Reynolds number not seen in the training set.

Figure 21

Figure 19. Figure 19 long description.Macroscale quantities analysed for channel flow at a higher Reynolds number not seen in the training set. (a) Streamwise spectra at z/H = 0.46875. (b) Spatial correlations at z/H=0.46875$z/H = 0.46875$.

Figure 22

Figure 20. Channel flow Reynolds stress profiles for various SGS models at a higher Reynolds number not seen in the training set. (a) Here R11$R_{11}$ varying as a function of wall normal distance. (b) Here R13$R_{13}$ varying as a function of wall normal distance. (c) Here R33$R_{33}$ varying as a function of wall normal distance.

Figure 23

Figure 21. Extreme Reynolds number scaling for HIT. (a) Energy spectra at various Reynolds numbers. (b) Spatial correlations at various Reynolds numbers.

Figure 24

Figure 22. Various mean profiles for channel flow extrapolation to extreme Reynolds numbers. (a) Channel flow mean profiles for various Reynolds numbers. (b) Channel flow mean profiles for various Reynolds numbers with sigma model and fits.

Figure 25

Figure 23. Figure 23 long description.Reynolds stress profiles for various extreme Reynolds numbers for the S4ND model. (a) Here R11$R_{11}$ varying as a function of wall normal distance. (b) Here R13$R_{13}$ varying as a function of wall normal distance. (c) Here R33$R_{33}$ varying as a function of wall normal distance.

Figure 26

Figure 24. Implied wall skin friction coefficient versus a first-order trend, with the x-axis being friction Reynolds number. The trends show that the error is likely due to the wall model, as both the S4ND model (plotted in blue) and the sigma model (plotted in red) show similar trends and deviation from the expected trend (plotted in yellow (Schultz & Flack 2013)).

Figure 27

Figure 25. Computational cost of various subgrid stress models. Overhead refers to the time it takes to load the neural network from disk and send the inputs from the LES to the neural network for computation, and then returning the neural network subgrid stress output back to the LES.

Figure 28

Figure 26. Bandlimit parameter sensitivity sweeps for HIT and channel flow on the hold out test set. The bandlimit value of α=0.05$\alpha = 0.05$ has a slight performance degradation as compared with the other two larger values. (a) Losses on the test set for HIT when varying α$\alpha$. (b) Losses on the test set for channel flow when varying α$\alpha$.

Figure 29

Figure 27. Sensitivity study of the sigma model to coefficients in the recommended range and also non-standard coefficients. The values in the legend correspond to the coefficient used. (a) Spectra of forced HIT for various sigma model coefficient values. (b) Forced HIT spatial correlations for various sigma model values.