Current browse context:
cs.CV
Change to browse by:
References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Efficient Training of 3D Seismic Image Fault Segmentation Network under Sparse Labels by Weakening Anomaly Annotation
(Submitted on 11 Oct 2021 (v1), revised 18 Oct 2021 (this version, v2), latest version 21 Jun 2022 (v6))
Abstract: Seismic data fault detection has recently been regarded as a 3D image segmentation task. The nature of fault structures in seismic image makes it difficult to manually label faults. Manual labeling often has many false negative labels (abnormal annotations), which will seriously harm the training process. In this work, we find that region-based loss significantly outperforms distribution-based loss when dealing with false negative labels, therefore we proposed Mask Dice loss (MD loss), which is the first reported region-based loss function for training 3D image segmentation networks using sparse 2D slice labels. In addition, fault is an edge feature, and the current network widely used for fault segmentation downsamples the features multiple times, which is not conducive to edge representation and thus requires many parameters and computational effort to preserve the features. We proposed Fault-Net, which uses a high-resolution and shallow structure to propagate multi-scale features in parallel, fully preserving edge features. Meanwhile, in order to efficiently fuse multi-scale features, we decouple the convolution process into feature selection and channel fusion, and proposed a lightweight feature fusion block, Multi-Scale Compression Fusion (MCF). Because the Fault-Net always keeps the edge features during propagation, only few parameters and computation are required. Experimental results show that MD loss can clearly weaken the effect of false negative labels. The Fault-Net parameter is only 0.42MB, support up to 528^3 (1.5x10^8, Float32) size cuboid inference on 16GB video ram, its inference speed on CPU and GPU is significantly faster than other networks. It works well on most of the open data seismic images, and the result of our method is the state-of-the-art in the FORCE fault identification competition.
Submission history
From: Yimin Dou [view email][v1] Mon, 11 Oct 2021 14:39:25 GMT (49401kb,D)
[v2] Mon, 18 Oct 2021 08:41:03 GMT (49377kb,D)
[v3] Tue, 19 Oct 2021 08:24:41 GMT (49377kb,D)
[v4] Sat, 26 Mar 2022 13:48:10 GMT (45756kb,D)
[v5] Thu, 7 Apr 2022 06:26:41 GMT (6110kb,D)
[v6] Tue, 21 Jun 2022 07:26:40 GMT (9302kb,D)
Link back to: arXiv, form interface, contact.