Structure-aware Bayesian full waveform inversion via diffusion-regularized Stein variational gradient descent

Abstract

Full waveform inversion (FWI) is a powerful tool for high-resolution subsurface model reconstruction, but it remains highly nonlinear and ill-posed, especially when the initial model is inaccurate or the acquired data are noisy and incomplete. Annealed Stein variational gradient descent (ASVGD) provides a particle-based Bayesian framework for FWI by approximating the posterior distribution with an ensemble of velocity models, thereby enabling uncertainty quantification beyond a single deterministic solution. However, conventional ASVGD-FWI commonly relies on simple handcrafted priors, such as Gaussian or smoothing assumptions, which are often insufficient to represent complex geological structures. To address this limitation, we introduce a learned diffusion model (DM) as an implicit regularization mechanism within the ASVGD-FWI framework. Instead of explicitly evaluating the diffusion prior density, the trained diffusion model acts as a learned geological projection operator that guides updated particles toward the distribution of plausible velocity models. In this way, the proposed method combines data consistency from FWI, posterior diversity from ASVGD, and structural realism from the diffusion prior. Numerical experiments demonstrate that the diffusion-regularized ASVGD-FWI produces more geologically consistent mean models, suppresses nonphysical artifacts, and yields uncertainty estimates that are more concentrated around structurally complex and poorly constrained regions. These results suggest that learned diffusion priors can effectively enhance Bayesian FWI by improving both inversion stability and the interpretability of posterior uncertainty.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Machine Learning (Statistics)

• arXiv preprint · 14 October 2026

  Ranked Right, Sized Wrong: Ensemble Spread Fails Silently at Streamflow Extremes

  Justin Gallant and Tyler Wilson
• arXiv preprint · 12 October 2026

  CartoVec: General-Purpose Location Embeddings from Open Map Data

  Furkan Gursoy
• arXiv preprint · 18 October 2026

  Emergent Sparsity: How Many Effective Variables Survive Coarse-Graining in AlphaEarth Embeddings?

  Johannes Hirn
• arXiv preprint · 12 October 2026

  Event-Centric Earthquake Monitoring with a Multi-Station Transformer

  Haoyu Wang and Weiqiang Zhu
• arXiv preprint · 19 October 2026

  Ammonia-Hydrogen Phase Equilibria in Planetary Envelopes from First-Principles-Based Machine-Learning Molecular Dynamics

  Akash Gupta, Austin Guo and Jie Deng

All new papers in Machine Learning (Statistics)