arXiv Machine Learning

Continually learning neural-operator surrogate for three-dimensional airborne electromagnetic Bayesian inversion

The paper presents a continually learning neural‑operator surrogate for the three‑dimensional forward operator used in time‑domain airborne electromagnetic (AEM) Bayesian inversion. By training on successive geological priors and employing an ensemble‑disagreement validity check, the surrogate replaces the expensive forward solver, enabling the Markov chain Monte Carlo sampler to reproduce the full‑solver posterior with credible intervals within 2.6 % of the truth. Applied to the 2013 Capricorn TEMPEST survey, the surrogate inverts over two million soundings in seconds, making uncertainty‑quantified conductivity imaging at survey scale feasible for near real‑time mineral‑systems targeting.

arXiv AI
Jul 20

Energy-based Transport for Amortized Bayesian Inference

arXiv:2605. 15407v3 Announce Type: replace-cross Abstract: We consider amortized Bayesian inference for nonlinear inverse problems using only samples from the joint distribution of parameters and observations, including problems with unknown functions in a Banach space.

By Ricardo Baptista, Hojjat Kaveh, Andrew M. Stuart
arXiv Machine Learning
Sep 24

Inverse Problems Conditioned on Observation Ensembles: Applications and Methods

The paper introduces the Ensemble-conditioned Inverse Problem (EIP), a multivariate statistical framework for inferring an ensemble that follows the pushforward of a prior through a forward process. It applies to fields such as high‑energy physics, full waveform inversion, and inverse imaging, and proposes non‑iterative inference‑time methods using ensemble inverse generative models that avoid explicit forward model use during inference. The authors demonstrate the approach on synthetic and real datasets and provide code for replication.

By Zhengyan Huan, Camila Pazos, Martin Klassen, Vincent Croft, Pierre-Hugues Beauchemin, Shuchin Aeron
arXiv Machine Learning
Sep 14

Physical-State-Guided Diffusion Sampling for Full-Waveform Inversion

Physical-State-Guided Diffusion Sampling (PSG) couples a persistent physical velocity model to a diffusion prior via a Gaussian bridge, allowing the physical state to be refined by waveform fitting while guiding the reverse diffusion process. This approach separates wave‑equation and denoiser gradients, preserving conventional FWI initialization and optimization history. PSG outperforms classical and diffusion‑based baselines on four OpenFWI families, maintains strong structural recovery under noise, and supports large‑scale models like Marmousi, Overthrust, and BP2004 Salt without retraining.

By Chen Min, Haowen Jiang, Zheng Ma, Xiongbin Yan
arXiv Machine Learning
Jun 16

Amortized mean-shift interacting particles

arXiv:2606. 15871v1 Announce Type: cross Abstract: Bayesian inference for inverse problems is run to evaluate integrals -- posterior expectations, tail probabilities, and risks -- across a stream of observations.

By Ali Siahkoohi