Rotationally Equivariant Super-Resolution of Velocity Fields in Two-Dimensional Fluids Using Convolutional Neural Networks
This work addresses the challenge of orientation-invariant super-resolution for fluid mechanics applications, but it is incremental as it builds on existing CNNs with theoretical extensions.
This paper tackles the problem of super-resolving velocity fields in 2D fluids by focusing on rotational equivariance, showing that rotational consistency in datasets is necessary for models to achieve this property, and demonstrating through experiments that imposing equivariance via weight sharing improves inference without requiring large datasets.
This paper investigates the super-resolution (SR) of velocity fields in two-dimensional fluids from the viewpoint of rotational equivariance. SR refers to techniques that estimate high-resolution images from those in low resolution and has lately been applied in fluid mechanics. The rotational equivariance of SR models is defined as the property in which the super-resolved velocity field is rotated according to a rotation of the input, which leads to the inference covariant to the orientation of fluid systems. Generally, the covariance in physics is related to symmetries. To clarify a relationship to symmetries, the rotational consistency of datasets for SR is newly introduced as the invariance of pairs of low- and high-resolution velocity fields with respect to rotation. This consistency is sufficient and necessary for SR models to acquire rotational equivariance from large datasets with supervised learning. Such a large dataset is not required when rotational equivariance is imposed on SR models through weight sharing of convolution kernels as prior knowledge. Even if a fluid system has rotational symmetry, this symmetry may not carry over to a velocity dataset, which is not rotationally consistent. This inconsistency can occur when the rotation does not commute with the generation of low-resolution velocity fields. These theoretical suggestions are supported by the results from numerical experiments, where two existing convolutional neural networks (CNNs) are converted into rotationally equivariant CNNs and the inferences of the four CNNs are compared after the supervised training.