Physics-Constrained Three-Dimensional Swin Transformer for Gravity Data Inversion

Abstract

This paper proposes a physics-constrained 3D Swin Transformer (ST) for gravity inversion. By leveraging the self-attention mechanism in 3D ST, the method effectively models global dependencies within gravity data, enabling the network to reweight features globally and focus on critical anomalous regions. Additionally, prior gradient information is integrated into the loss function, and a hierarchical weight allocation strategy is adopted to guide the model in learning boundary information of density structures and deep-seated features more effectively. Synthetic experiments demonstrate that the proposed method achieves lower model errors, better boundary alignment, and higher inversion accuracy. The approach is further validated using gravity anomaly observations from the Gonghe Basin in Qinghai, yielding reliable and precise inversion results.

Keywords

• gravity inversion
• long-range dependencies
• self-attention mechanism
• physics constrained