Light-weighted foundation model for seismic data processing based on representative and non-redundant pre-training dataset

Abstract

In the fields of computer vision (CV) and remote sensing (RS), foundational models typically follow the "big data + large model parameters" paradigm. However, the application of this strategy in seismic data processing faces several challenges: seismic data is difficult to obtain and the scarcity of publicly available datasets make it difficult to construct large-scale datasets. Additionally, the high computational cost associated with a large number of model parameters restricts widespread research in this domain. Therefore, we propose a lightweight seismic processing foundational model paradigm (SPFM), which aims to overcome the limitations of traditional methods by data engineering and network architecture innovation. Specifically, we propose an innovative dataset construction strategy that generates more seismic data by data augmentation techniques, including collecting publicly available field data and using generative diffusion models (GDM) for data enhancement. Furthermore, we optimize the data distribution by employing dimensionality reduction, cluster analysis, and stratified sampling methods, reducing redundant information while preserving important seismic features, thus constructing a comprehensive dataset. In terms of network architecture design, we introduce the selective structured state-space model (Mamba) structure, which effectively captures global features of seismic data and alleviates the quadratic growth of computational complexity inherent in Transformer-based models, thereby improving computational efficiency. This model, pre-trained with only four A800 GPUs, outperforms traditional methods across multiple tasks, including denoising, interpolation, frequency-band extrapolation, and resolution enhancement. The lightweight paradigm provides an solution for seismic data processing, advancing the generalization and accessibility of seismic data processing.