State estimation in homogeneous isotropic turbulence using super-resolution with a 4DVar training algorithm

Abstract

Variational data assimilation and machine-learning based super-resolution are two alternative approaches to state estimation in turbulent flows. The former is an optimisation problem featuring a time series of coarse observations, the latter usually requires a library of high-resolution 'ground truth' data. We show that the classic '4DVar' data assimilation algorithm can be used to train neural networks for super-resolution in three-dimensional isotropic turbulence without the need for high-resolution reference data. To do this, we adapt a pseudo-spectral version of the fully-differentiable JAX-CFD solver (Kochkov et al, Proc. Nat. Acad. Sci. 118, 2021) to three-dimensional flows and combine it with a convolutional neural network for super-resolution. As a result we are able to include entire trajectories in our loss function which is minimised with gradient-based optimisation to define the neural network weights. We show that the resulting neural networks outperform 4DVar for state estimation at initial time over a wide variety of metrics, though 4DVar leads to more robust predictions towards the end of its assimilation window. We also present a hybrid approach in which the trained neural network output is used to initialise 4DVar. The resulting performance is more than twice as accurate as other state estimation strategies for all times and performs well even beyond known limiting lengthscales, all without requiring access to high-resolution measurements at any point.