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.

Hierarchical Attention-Based Multiple Instance Learning Network for Patient-Level Lung Cancer Diagnosis

A Hybrid CNN Feature Model for Pulmonary Nodule Malignancy Risk Differentiation.

Lung Cancer Diagnosis Using Deep Attention Based on Multiple Instance Learning and Radiomics

Lung Cancer Diagnosis Using Deep Attention‐based Multiple Instance Learning and Radiomics

Cascaded-Recalibrated Multiple Instance Deep Model for Pathologic-Level Lung Cancer Prediction in CT Images

Adaptive Aggregated Attention Network for Pulmonary Nodule Classification

End-to-End Multi-Task Learning for Lung Nodule Segmentation and Diagnosis

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

Lung cancer detection using co-learning from chest CT images and clinical demographics

Multi-View Coupled Self-Attention Network for Pulmonary Nodules Classification.

Lung Nodule Malignancy Prediction Using Multi-Task Convolutional Neural Network

Multiple Instance Learning for Lymph Node Metastasis Prediction from Cervical Cancer MRI.

Multi-Task Learning for Lung Nodule Classification on Chest CT

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

MSCS-DeepLN: Evaluating Lung Nodule Malignancy Using Multi-Scale Cost-Sensitive Neural Networks

Attention Based Deep Multiple Instance Learning Approach for Lung Cancer Prediction using Histopathological Images

Lung cancer classification and identification framework with automatic nodule segmentation screening using machine learning

Interpretable Lung Cancer Diagnosis with Nodule Attribute Guidance and Online Model Debugging

An attention-based deep learning network for lung nodule malignancy discrimination

Knowledge-Guided And Hyper-Attention Aware Joint Network For Benign-Malignant Lung Nodule Classification

Multi-Scale Multi-View Model Based on Ensemble Attention for Benign-Malignant Lung Nodule Classification on Chest CT