Title:Rotation Equivariant Fourier Neural Operators for Learning Symmetry Preserving Transformations on Scalar Fields, Vector Fields, and Higher Order Tensor Fields

Abstract:We introduce equivariant neural operators for learning resolution invariant, rotation equivariant and consequently symmetry preserving transformations between arbitrary sets of tensor fields including scalar fields, vector fields, and higher order fields. Our tensor field convolution layers emulate any linear operator by learning its impulse response or Green's function as the convolution kernel. Our tensor field attention layers emulate pairwise field coupling via local tensor products. Convolutions and associated adjoints can be in real or Fourier space allowing for linear scaling. By unifying concepts from Euclidean tensor field networks (E3NN), tensor basis networks (TBNN) and Fourier neural operators (FNO), we achieve good predictive performance on a wide range of PDEs and dynamical systems in engineering and quantum chemistry. Code is in Julia and available upon request from authors.