Improving Full Waveform Inversion in Large Model Era
Not on the map. Why?
Abstract (arXiv)
Full Waveform Inversion (FWI) is a highly nonlinear and ill-posed problem that aims to recover subsurface velocity maps from surface-recorded seismic waveforms data. Existing data-driven FWI typically uses small models, as available datasets have limited volume, geological diversity, and spatial extent, leading to substantial concerns about overfitting. Although they perform well on synthetic datasets, current methods fail to generalize to more realistic geological structures. In this work, we show that a model trained entirely on simulated and relatively simple data can generalize remarkably well to challenging and unseen geological benchmarks. We provide a working recipe that tames a billion-parameter model for FWI through coordinated scaling across three axes: model capacity, data diversity, and training strategy. Our model achieves state-of-the-art performance on OpenFWI and significantly narrows the generalization gap in data-driven FWI. Across six challenging geophysical benchmarks, including Marmousi, 2D SEG/EAGE Salt and Overthrust, 2004 BP, Sigsbee, and SEAM Phase I, it infers complex structures absent from the training set and delivers significant performance improvements (SSIM from 0.5844 to 0.7669). Overall, our results demonstrate that with an appropriate scaling strategy, large models trained on simple synthetic data can achieve substantial generalization to more complex and realistic geological structures.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 20 of its 47 references that are in the map. First posted on arXiv 27 Feb 2026.
Builds on
20 of 47 references.
References · 20
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,581
- 2022Diffusion Posterior Sampling for General Noisy Inverse Problemscited by 1,375
- 2025Group Sequence Policy Optimizationcited by 788
- 2021Vector-quantized Image Modeling with Improved VQGANcited by 656
- 2021Solving Inverse Problems in Medical Imaging with Score-Based Generative Modelscited by 514
- 2025MMaDA: Multimodal Large Diffusion Language Modelscited by 241
- 2023Sequential Modeling Enables Scalable Learning for Large Vision Modelscited by 212
- 2024Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealingcited by 157
- 2018InversionNet: A Real-Time and Accurate Full Waveform Inversion with CNNs and continuous CRFscited by 60
- 2021OpenFWI: Large-Scale Multi-Structural Benchmark Datasets for Seismic Full Waveform Inversioncited by 44
8 more
- 2019Deep-Learning Inversion of Seismic Datacited by 30
- 2021Unsupervised Learning of Full-Waveform Inversion: Connecting CNN and Partial Differential Equation in a Loopcited by 29
- 2023A prior regularized full waveform inversion using generative diffusion modelscited by 29
- 2023An Empirical Study of Large-Scale Data-Driven Full Waveform Inversioncited by 9
- 2022An Intriguing Property of Geophysics Inversioncited by 8
- 2023Learned Full Waveform Inversion Incorporating Task Information for Ultrasound Computed Tomographycited by 8
- 2023Auto-Linear Phenomenon in Subsurface Imagingcited by 6
- 2024Propagating the prior from shallow to deep with a pre-trained velocity-model Generative Transformer networkcited by 2
Built on by
Not cited yet.