Abstract:Objective: Lung cancer is a significant public health problem worldwide, and its mortality and morbidity rates are among the highest of cancers. At the same time, early diagnosis of nodules can significantly improve the survival rate of patients. Therefore, this paper proposes the Lung-YOLO algorithm for lung CT image detection based on YOLOv6. Methods: First, to enable the network to detect nodules of different sizes and to minimize missed detections, we introduce the Multi-scale Dual-branch Attention (MSDA) mechanism in the feature extraction part of the network. The input features undergo continual dilation convolutions, effectively establish remote dependencies, and the fused multiscale contextual information expands the receptive field while enhancing the model's ability to detect targets of different sizes, which not only contains precise category and location information but also enables the allocation of attentional weights to generate more pixel-level attention. Then, the fused features access the dual-branch attention module to shift the model's attention to the target nodules, and the composed dual-branch structure captures cross-dimensional interactions and realizes inter-dimensional dependencies, effectively improving the detection performance. Second, during the feature transfer process, the original Bidirectional Feature Pyramid structure (RepBiFPAN) suffers from the loss of detailed information such as texture and color, making it challenging to localize target nodules accurately. To address this, we propose the Cross-layer Aggregation Module (CLAM), by cross-layer aggregating the multi-level feature layer of the backbone with the multi-level detection layer of the head, which preserves the multi-level fine-grained information that may be lost during the feature transfer process, which is crucial for the detection of small targets. Finally, the module proposed in this paper can be easily incorporated into any detection framework for plug-and-play. Results: Our method achieves accuracy, precision, recall, and mAP of 97.5 %, 96.5 %, 96.9 %, and 97.9 % on the LUNA16 dataset, and 95.1 %, 94.3 %, 93.4 %, and 95.9 % on the LIDC-IDRI dataset, respectively, surpassing many existing state-of-the-art detection methods. Moreover, the inference speed is 22.8 ms and 28.4 ms per image with 30.6 M parameters, respectively.

A hierarchical approach for pulmonary nodules identification from CT images using YOLO v5s nodule detection and 3D neural network classifier

Hierarchical approach for pulmonary-nodule identification from CT images using YOLO model and a 3D neural network classifier

A Deep 3D Residual CNN for False-Positive Reduction in Pulmonary Nodule Detection

Identification of Pulmonary Nodules Via CT Images with Hierarchical Fully Convolutional Networks.

An End-to-end Framework for Pulmonary Nodule Detection and False Positive Reduction from CT Images

3DFPN-HS$^2$: 3D Feature Pyramid Network Based High Sensitivity and Specificity Pulmonary Nodule Detection

Pulmonary Nodule Detection Using 3-D Residual U-Net Oriented Context-Guided Attention and Multi-Branch Classification Network

An Embedded Multi-branch 3D Convolution Neural Network for False Positive Reduction in Lung Nodule Detection.

Accurate and Robust Pulmonary Nodule Detection by 3D Feature Pyramid Network with Self-supervised Feature Learning

Simultaneous Accurate Detection of Pulmonary Nodules and False Positive Reduction Using 3D CNNs

Lung-YOLO: Multiscale Feature Fusion Attention and Cross-Layer Aggregation for Lung Nodule Detection

An Attentive and Adaptive 3D CNN for Automatic Pulmonary Nodule Detection in CT Image

Explainable AI for Lung Nodule Detection and Classification in CT Images

Automated Pulmonary Nodule Detection in CT Images Using Deep Convolutional Neural Networks.

A Cascade and Heterogeneous Neural Network for CT Pulmonary Nodule Detection and Its Evaluation on Both Phantom and Patient Data.

High precision localization of pulmonary nodules on chest CT utilizing axial slice number labels

A Pulmonary Nodule Detection Model Based on Progressive Resolution and Hierarchical Saliency

A Novel Multi-Scale CNNs for False Positive Reduction in Pulmonary Nodule Detection

Novel convolutional neural network architecture for improved pulmonary nodule classification on computed tomography

Lung Nodule Detection in Medical Images Based on Improved YOLOv5s

P3.13-037 Deep Learning System for Lung Nodule Detection