Physics-informed conditional diffusion model for generalizable elastic wave-mode separation

Abstract: Traditional elastic wavefield separation methods, while accurate, often demand substantial computational resources, especially for large geological models or 3D scenarios. Purely data-driven neural network approaches can be more efficient, but may fail to generalize and maintain physical consistency due to the absence of explicit physical constraints. Here, we propose a physics-informed conditional diffusion model for elastic wavefield separation that seamlessly integrates domain-specific physics equations into both the training and inference stages of the reverse diffusion process. Conditioned on full elastic wavefields and subsurface P- and S-wave velocity profiles, our method directly predicts clean P-wave modes while enforcing Laplacian separation constraints through physics-guided loss and sampling corrections. Numerical experiments on diverse scenarios yield the separation results that closely match conventional numerical solutions but at a reduced cost, confirming the effectiveness and generalizability of our approach.