An improved V-Net lung nodule segmentation model based on pixel threshold separation and attention mechanism

Xiaopu Ma,Handing Song,Xiao Jia,Zhan Wang

DOI: https://doi.org/10.1038/s41598-024-55178-3

IF: 4.6

2024-02-28

Scientific Reports

Abstract:Accurate labeling of lung nodules in computed tomography (CT) images is crucial in early lung cancer diagnosis and before nodule resection surgery. However, the irregular shape of lung nodules in CT images and the complex lung environment make it much more challenging to segment lung nodules accurately. On this basis, we propose an improved V-Net segmentation method based on pixel threshold separation and attention mechanism for lung nodules. This method first offers a data augment strategy to solve the problem of insufficient samples in 3D medical datasets. In addition, we integrate the feature extraction module based on pixel threshold separation into the model to enhance the feature extraction ability under different thresholds on the one hand. On the other hand, the model introduces channel and spatial attention modules to make the model pay more attention to important semantic information and improve its generalization ability and accuracy. Experiments show that the Dice similarity coefficients of the improved model on the public datasets LUNA16 and LNDb are 94.9% and 81.1% respectively, and the sensitivities reach 92.7% and 76.9% respectively. which is superior to most existing UNet architecture models and comparable to the manual level segmentation results by medical technologists.

multidisciplinary sciences

What problem does this paper attempt to address?

The paper primarily addresses the precise segmentation of pulmonary nodules in computed tomography (CT) images. Due to the irregular shape of pulmonary nodules and the complex lung environment, accurately segmenting pulmonary nodules is highly challenging. To solve this problem, the authors propose an improved V-Net segmentation method (named Dig-CS-VNet), which is based on pixel threshold separation and attention mechanisms. Specifically, the main contributions of this method include: 1. **Data Augmentation Strategy**: A data augmentation method suitable for 3D medical datasets is proposed. By using slice reverse order and matching plane geometric transformations, the dataset size is increased, addressing the issue of insufficient samples in 3D medical image datasets. 2. **Model Construction**: - A feature extraction module based on pixel threshold separation (Dig_Sep) is proposed, enhancing the model's ability to extract hierarchical features at different thresholds. - Channel and spatial attention mechanisms are introduced into the model, enabling the model to focus more on important semantic information, thereby improving its generalization ability and accuracy. Experimental results show that the improved model achieved Dice similarity coefficients of 94.9% and 81.1% on the LUNA16 and LNDb public datasets, respectively, with sensitivities of 92.7% and 76.9%, respectively. This outperforms most existing U-Net architecture models and is comparable to the manual segmentation level of medical technicians. In summary, this study aims to improve the accuracy and efficiency of pulmonary nodule segmentation in CT images by proposing an improved V-Net model that combines pixel threshold separation and attention mechanisms.

An improved V-Net lung nodule segmentation model based on pixel threshold separation and attention mechanism

Detection and segmentation of pulmonary nodules based on improved 3D VNet algorithm

Improved 3D Multiscale V-Net Network-based Segmentation of Lung Nodules.

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

Automatic Segmentation of Lung Noudles Using Improved U-Net NetWork

Segmentation of Lung Nodules in CT Images Using Improved U-Net++

Segmentation of Pulmonary Nodules Based on Improved UNet++

An Intelligent Model For Benign And Malignant Pulmonary Nodule Analysis Using U-Net Networks And Multilevel Attention Mechanisms

Lung Nodule Segmentation and Recognition Algorithm Based on Multiposition U-Net

Attention-guided deep neural network with a multichannel architecture for lung nodule classification

MSA-unet: A Multiscale Lightweight U-net Lung CT Image Segmentation Algorithm Under Attention Mechanism

Uncertainty Analysis Based Attention Network for Lung Nodule Segmentation from CT Images

Multilevel Attention Unet Segmentation Algorithm for Lung Cancer Based on CT Images

SM-RNet: A Scale-Aware-Based Multiattention-Guided Reverse Network for Pulmonary Nodules Segmentation.

3D Segmentation of Pulmonary Nodules Based on Multi-View and Semi-Supervised

Dual‐branch feature fusion S3D V‐Net network for lung nodules segmentation

MAU-Net: Multiple Attention 3D U-Net for Lung Cancer Segmentation on CT Images

Design of Lung Nodules Segmentation and Recognition Algorithm Based on Deep Learning

Detection of Pulmonary Nodules in CT Images Based on 3D Multi-Scale Retina Vnet

CTBP-Net: Lung Nodule Segmentation Model Based on the Cross-Transformer and Bidirectional Pyramid

A Dual-Attention V-network for Pulmonary Lobe Segmentation in CT Scans