Multiscale lung nodule segmentation based on 3D coordinate attention and edge enhancement

Abstract

An important prerequisite for improving the reliability of lung cancer surveillance and clinical interventions is accurate lung nodule segmentation. Although deep learning is effective at performing medical image segmentation, lung CT image heterogeneity, nodule size, shape, and location variations, convolutional localized feature extraction characteristics, the receptive field limitations of continuous downsampling, lesion edge information losses, fuzzy boundary segmentation challenges, and the low segmentation accuracy achieved when segmenting lung CT images using deep learning remain. An edge-enhanced multiscale Sobel coordinate attention-atrous spatial convolutional pooling pyramid V-Net (SCA-VNet) algorithm for lung nodule segmentation was proposed to solve these problems. First, a residual edge enhancement module was designed, which was used to enhance the edges of the original data. Using an edge detection operator in combination with a residual module, this module could reduce data redundancy and alleviate the gray level similarity between the foreground and background. Then, a 3D atrous spatial convolutional pooling pyramid module set different expansion rates, which could obtain feature maps under different receptive fields and capture the multiscale information of the segmentation target. Finally, a three-dimensional coordinate attention network (3D CA-Net) module was added to the encoding and decoding paths to extract channel weights from multiple dimensions. This step propagated the spatial information in the coding layer to the subsequent layers, and it could reduce the loss of information during the forward propagation process. The proposed method achieved a Dice coefficient of 87.50% on the lung image database consortium and image database resource initiative (LIDC-IDRI). It significantly outperformed the existing lung nodule segmentation models (UGS-Net, REMU-Net, and multitask models) and compared favorably with the Med3D, CENet, and PCAM_Net segmentation models in terms of their Dice coefficients, which were 3.37%, 2.2%, and 1.43%, respectively. The experimental results showed that the proposed SCA-VNet model attained improved lung nodule segmentation accuracy and laid a good foundation for improving the early detection rate of lung cancer.

Keywords

• pulmonary nodule segmentation
• deep learning
• attention mechanism
• 3d multiscale feature extraction
• hollow-space convolution pooling
• residual edge enhancement
• convolutional neural network