Seismic wavefield solutions via physics-guided generative neural operator

Abstract

Current neural operators often struggle to generalize to complex, out-of-distribution conditions, limiting their ability in seismic wavefield representation. To address this, we propose a generative neural operator (GNO) that leverages generative diffusion models (GDMs) to learn the underlying statistical distribution of scattered wavefields while incorporating a physics-guided sampling process at each inference step. This physics guidance enforces wave equation-based constraints corresponding to specific velocity models, driving the iteratively generated wavefields toward physically consistent solutions. By training the diffusion model on wavefields corresponding to a diverse dataset of velocity models, frequencies, and source positions, our GNO enables to rapidly synthesize high-fidelity wavefields at inference time. Numerical experiments demonstrate that our GNO not only produces accurate wavefields matching numerical reference solutions, but also generalizes effectively to previously unseen velocity models and frequencies.