3D elastic wave propagation with a Factorized Fourier Neural Operator (F-FNO)

Numerical simulations are computationally demanding in three-dimensional (3D) settings but they are often required to accurately represent physical phenomena. Neural operators have emerged as powerful surrogate models to alleviate the computational costs of simulations. However, neural operators applications in 3D remain sparse, mainly due to the difficulty of obtaining training databases for supervised learning and the size of 3D neural operators that poses memory challenges. This work focuses on the propagation of elastic waves in 3D domains and showcases the Factorized Fourier Neural Operator (F-FNO) as an efficient and accurate surrogate model. The F-FNO is trained on the publicly available HEMEW-3D database of 30 000 wavefields simulations in realistic heterogeneous domains. The F-FNO predicts space- and time-dependent (3D) surface wavefields depending on the characteristics of the propagation domain (characterized by the velocity of shear waves). Four FNO variants are compared and extensive investigations on the influence of hyperparameters and training strategies are conducted.