Salt3DNet: A Self-Supervised Learning Framework for 3D Salt Segmentation

Liuqing Yang,Sergey Fomel,Shoudong Wang,Xiaohong Chen,Omar M. Saad,Yangkang Chen
DOI: https://doi.org/10.1109/tgrs.2024.3394592
IF: 8.2
2024-01-01
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Salt body segmentation is a critical part of structural interpretation and oil and gas exploration for subsalt reservoirs. Existing automatic salt body segmentation techniques mostly use supervised learning strategies. It is challenging to generate a large number of labels by manual labeling, especially for 3-D salt bodies. Here, we propose a self-supervised learning (SSL) framework called Salt3DNet, for 3-D salt body segmentation. This framework is divided into two stages: pretraining and fine-tuning of downstream tasks. In the pretraining stage, we use the Barlow twins (BTs) method to pretrain the encoder and reduce redundancy in a contrastive learning manner to learn high-level data representations. In the fine-tuning stage, we construct two encoders to reconstruct 3-D seismic data and segment salt bodies in a multitask collaborative learning way. The encoder and decoder are composed of the 3-D fully convolutional DenseNet and soft attention mechanism, where the latter represents the selective kernel block (SKB) with multiple kernels of different sizes. Salt3DNet calculates the correlation matrix of features from different perspectives in the pretraining stage and makes it close to the identity matrix to obtain a more prosperous feature representation. Then, Salt3DNet uses a limited number of labeled samples for training. According to the evaluation metrics, the proposed network has demonstrated promising salt segmentation performance in 3-D SEG advanced modeling (SEAM) synthetic data and $F3$ block real seismic data. In addition, the proposed network is demonstrated to have higher prediction accuracy than state-of-the-art salt segmentation frameworks through ablation experiments.
What problem does this paper attempt to address?