Abstract:Background and Objective : Lung cancer counts among diseases with the highest global morbidity and mortality rates. The automatic segmentation of lung tumors from CT images is of vast significance. However, the segmentation faces several challenges, including variable shapes and different sizes, as well as complicated surrounding tissues. Methods : We propose a multi-scale segmentation squeeze-and-excitation UNet with a conditional random field (M-SegSEUNet-CRF) to automatically segment lung tumors from CT images. M-SegSEUNet-CRF employs a multi-scale strategy to solve the problem of variable tumor size. Through the spatially adaptive attention mechanism, the segmentation SE blocks embedded in 3D UNet are utilized to highlight tumor regions. The dense connected CRF framework is further added to delineate tumor boundaries at a detailed level. In total, 759 CT scans of patients with lung cancer were used to train and evaluate the M-SegSEUNet-CRF model (456 for training, 152 for validation, and 151 for test). Meanwhile, the public NSCLC-Radiomics and LIDC datasets have been utilized to validate the generalization of the proposed method. The role of different modules in the M-SegSEUNet-CRF model is analyzed by the ablation experiments, and the performance is compared with that of UNet, its variants and other state-of-the-art models. Results : M-SegSEUNet-CRF can achieve a Dice coefficient of 0.851±0.071, intersection over union (IoU) of 0.747±0.102, sensitivity of 0.827±0.108, and positive predictive value (PPV) of 0.900±0.107. Without a multi-scale strategy, the Dice coefficient drops to 0.820±0.115; without CRF, it drops to 0.842±0.082, and without both, it drops to 0.832. M-SegSEUNet-CRF presented a higher Dice coefficient than 3D UNet (0.782±0.115) and its variants (ResUNet, 0.797±0.132; DenseUNet, 0.792±0.111, and UNETR, 0.794±0.130). Although the performance slightly declines with the decrease in tumor volume, M-SegSEUNet-CRF exhibits more obvious advantages than the other comparative models. Conclusions : Our M-SegSEUNet-CRF model improves the segmentation ability of UNet through the multi-scale strategy and spatially adaptive attention mechanism. The CRF enables a more precise delineation of tumor boundaries. The M-SegSEUNet-CRF model integrates these characteristics and demonstrates outstanding performance in the task of lung tumor segmentation. It can furthermore be extended to deal with other segmentation problems in the medical imaging field.

A scale-aware UNet++ model combined with attentional context supervision and adaptive Tversky loss for accurate airway segmentation

Multi-Stage Airway Segmentation in Lung CT Based on Multi-scale Nested Residual UNet

Detail-sensitive 3D-UNet for pulmonary airway segmentation from CT images

Accurate Airway Tree Segmentation in CT Scans via Anatomy-aware Multi-class Segmentation and Topology-guided Iterative Learning

A CT-Based Airway Segmentation Using U$^2$-net Trained by the Dice Loss Function

An end-to-end multi-scale airway segmentation framework based on pulmonary CT image

Two-stage contextual transformer-based convolutional neural network for airway extraction from CT images

AirwayNet: A Voxel-Connectivity Aware Approach for Accurate Airway Segmentation Using Convolutional Neural Networks

Fuzzy Attention Neural Network to Tackle Discontinuity in Airway Segmentation

Deep anatomy learning for lung airway and artery-vein modeling with contrast-enhanced CT synthesis

Towards Connectivity-Aware Pulmonary Airway Segmentation.

An 8-Layer Residual U-Net with Deep Supervision for Segmentation of the Left Ventricle in Cardiac CT Angiography

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

Learning Tubule-Sensitive CNNs for Pulmonary Airway and Artery-Vein Segmentation in CT

Multi-Scale Segmentation Squeeze-and-Excitation UNet with Conditional Random Field for Segmenting Lung Tumor from CT Images

Alleviating Class-Wise Gradient Imbalance for Pulmonary Airway Segmentation

Coarse-to-fine Airway Segmentation Using Multi information Fusion Network and CNN-based Region Growing

A Branch-sensitive 3D Convolutional Neural Networks for Airway Segmentation

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

A U-Shaped Network Based on Multi-level Feature and Dual-Attention Coordination Mechanism for Coronary Artery Segmentation of CCTA Images

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