Application of Optimized 3D U-Net Residual Network with CBAM and MEGA Modules in Seismic Fault Detection

With the continuous development of deep learning technologies, fault prediction techniques based on various neural networks have been evolving. The deep learning modules based on U-Net residual networks have shown significant advantages in both learning efficiency and effectiveness. In this paper, we propose a deep learning model that integrates a 3D U-Net residual architecture, Convolutional Block Attention Module (CBAM), and Multi-scale Enhanced Global Attention (MEGA) module for automatic seismic fault detection and segmentation. This model can effectively handle complex 3D seismic data, fully exploiting both spatial and channel information, significantly improving the prediction accuracy for small faults, while only slightly increasing the computational cost.

Firstly, the model uses the 3D U-Net as the backbone framework, where the residual blocks (BasicRes) extract features through multiple convolution layers. The CBAM module is incorporated to apply attention weighting, enhancing the model's ability to focus on critical information. The CBAM module combines channel attention and spatial attention, effectively adjusting the importance of feature maps from different dimensions, enabling the model to identify potential fault features in complex seismic data.

Secondly, the MEGA module is introduced into the model, which further improves the model's feature representation ability by fusing multi-scale features and applying a global attention mechanism. By weighting global information, the MEGA module helps the model better capture key seismic fault features during feature fusion. This design allows the model to focus not only on local details but also to fully utilize the global contextual information in 3D data, thereby enhancing the accuracy of fault detection.

After validation, the model achieved promising results in seismic fault detection tasks, automatically identifying and segmenting fault structures in seismic data. The accuracy was improved from 80% with the original 3D U-Net residual network to 85%-87%. This provides strong support for applications such as seismic exploration and subsurface imaging.

Keywords: Seismic Fault Detection, 3D U-Net, Convolutional Block Attention Module (CBAM), Multi-scale Enhanced Global Attention (MEGA), Deep Learning