基于自动微分的地震反演框架,支持复杂模型与灵活算法快速构建。
Automatic Differentiation-based Full Waveform Inversion with Flexible Workflows
- 利用自动微分实现多种介质波方程及梯度计算,降低开发门槛。
- 支持软动态时间规整、Wasserstein距离等新型目标函数,提升反演精度。
- 融合深度学习实现隐式参数化与不确定性估计,适合研究者快速实验新方法。
全波形反演(FWI)通过迭代最小化观测数据与模拟数据之间的差异,构建高分辨率地下模型。然而,对于复杂的波方程、目标函数或正则化方法,其实现往往繁琐。近年来,自动微分(AD)在解决各类逆问题中展现出高效性,包括FWI。本文提出一个开源的基于自动微分的FWI框架(ADFWI),旨在简化新型反演方法的设计、开发与评估流程。该框架支持从各向同性声学到垂直/水平横各向同性弹性介质的波动方程正演模拟及其梯度计算,集成多种目标函数、正则化技术与优化算法。借助先进的自动微分技术,可轻松引入传统方法难以应用的软动态时间规整(soft dynamic time warping)和Wasserstein距离等目标函数。此外,ADFWI与深度学习结合,通过神经网络实现隐式模型重参数化,不仅引入学习型正则化,还可通过丢弃法(dropout)快速估计不确定性。为应对大规模反演中自动微分带来的高内存需求,框架采用小批量(mini-batch)与检查点(checkpointing)策略。通过全面评估,验证了ADFWI的创新性、实用性与鲁棒性,可有效应对FWI挑战,并作为新反演策略的快速实验平台。
原文摘要 · Abstract (English)
Full waveform inversion (FWI) is able to construct high-resolution subsurface models by iteratively minimizing discrepancies between observed and simulated seismic data. However, its implementation can be rather involved for complex wave equations, objective functions, or regularization. Recently, automatic differentiation (AD) has proven to be effective in simplifying solutions of various inverse problems, including FWI. In this study, we present an open-source AD-based FWI framework (ADFWI), which is designed to simplify the design, development, and evaluation of novel approaches in FWI with flexibility. The AD-based framework not only includes forword modeling and associated gradient computations for wave equations in various types of media from isotropic acoustic to vertically or horizontally transverse isotropic elastic, but also incorporates a suite of objective functions, regularization techniques, and optimization algorithms. By leveraging state-of-the-art AD, objective functions such as soft dynamic time warping and Wasserstein distance, which are difficult to apply in traditional FWI are also easily integrated into ADFWI. In addition, ADFWI is integrated with deep learning for implicit model reparameterization via neural networks, which not only introduces learned regularization but also allows rapid estimation of uncertainty through dropout. To manage high memory demands in large-scale inversion associated with AD, the proposed framework adopts strategies such as mini-batch and checkpointing. Through comprehensive evaluations, we demonstrate the novelty, practicality and robustness of ADFWI, which can be used to address challenges in FWI and as a workbench for prompt experiments and the development of new inversion strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。