Log In Sign Up

Two-Stage Convolutional Neural Network Architecture for Lung Nodule Detection

by   Haichao Cao, et al.

Early detection of lung cancer is an effective way to improve the survival rate of patients. It is a critical step to have accurate detection of lung nodules in computed tomography (CT) images for the diagnosis of lung cancer. However, due to the heterogeneity of the lung nodules and the complexity of the surrounding environment, robust nodule detection has been a challenging task. In this study, we propose a two-stage convolutional neural network (TSCNN) architecture for lung nodule detection. The CNN architecture in the first stage is based on the improved UNet segmentation network to establish an initial detection of lung nodules. Simultaneously, in order to obtain a high recall rate without introducing excessive false positive nodules, we propose a novel sampling strategy, and use the offline hard mining idea for training and prediction according to the proposed cascaded prediction method. The CNN architecture in the second stage is based on the proposed dual pooling structure, which is built into three 3D CNN classification networks for false positive reduction. Since the network training requires a significant amount of training data, we adopt a data augmentation method based on random mask. Furthermore, we have improved the generalization ability of the false positive reduction model by means of ensemble learning. The proposed method has been experimentally verified on the LUNA dataset. Experimental results show that the proposed TSCNN architecture can obtain competitive detection performance.


page 2

page 5

page 8

page 12

page 13

page 17


Accurate Pulmonary Nodule Detection in Computed Tomography Images Using Deep Convolutional Neural Networks

Early detection of pulmonary cancer is the most promising way to enhance...

Automated Pulmonary Nodule Detection via 3D ConvNets with Online Sample Filtering and Hybrid-Loss Residual Learning

In this paper, we propose a novel framework with 3D convolutional networ...

Towards Single-phase Single-stage Detection of Pulmonary Nodules in Chest CT Imaging

Detection of pulmonary nodules in chest CT imaging plays a crucial role ...

Transfer Learning by Cascaded Network to identify and classify lung nodules for cancer detection

Lung cancer is one of the most deadly diseases in the world. Detecting s...

Unsupervised Contrastive Learning based Transformer for Lung Nodule Detection

Early detection of lung nodules with computed tomography (CT) is critica...

MHSnet: Multi-head and Spatial Attention Network with False-Positive Reduction for Pulmonary Nodules Detection

The mortality of lung cancer has ranked high among cancers for many year...

No Surprises: Training Robust Lung Nodule Detection for Low-Dose CT Scans by Augmenting with Adversarial Attacks

Detecting malignant pulmonary nodules at an early stage can allow medica...