Gated-Dilated Networks for Lung Nodule Classification in CT scans

Abstract

Different types of Convolutional Neural Networks (CNNs) have been applied to detect cancerous lung nodules from computed tomography (CT) scans. However, the size of a nodule is very diverse and can range anywhere between 3 and 30 millimeters. The high variation of nodule sizes makes classifying them a difficult and challenging task. In this study, we propose a novel CNN architecture called Gated-Dilated (GD) networks to classify nodules as malignant or benign. Unlike previous studies, the GD network uses multiple dilated convolutions instead of max-poolings to capture the scale variations. Moreover, the GD network has a Context-Aware sub-network that analyzes the input features and guides the features to a suitable dilated convolution. We evaluated the proposed network on more than 1,000 CT scans from the LIDC-LDRI dataset. Our proposed network outperforms state-of-the-art baseline models including Multi-Crop, Resnet, and Densenet, with an AUC of >0.95. Compared to the baseline models, the GD network improves the classification accuracies of mid-range sized nodules. Furthermore, we observe a relationship between the size of the nodule and the attention signal generated by the Context-Aware sub-network, which validates our new network architecture.

arXiv commentPublished in IEEE Access

References in corpus (12)

• Adam: A Method for Stochastic Optimization
• Rethinking Atrous Convolution for Semantic Image Segmentation
• WaveNet: A Generative Model for Raw Audio
• Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition
• Wide Residual Networks
• Multi-Scale Context Aggregation by Dilated Convolutions
• DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs
• Rethinking the Inception Architecture for Computer Vision
• Aggregated Residual Transformations for Deep Neural Networks
• Neural Machine Translation in Linear Time
• Risk Stratification of Lung Nodules Using 3D CNN-Based Multi-task Learning
• Attention to Scale: Scale-aware Semantic Image Segmentation

Cited by in corpus (5)

• ProCAN: Progressive Growing Channel Attentive Non-Local Network for Lung Nodule Classification
• 3D Axial-Attention for Lung Nodule Classification
• Integration of Radiomics and Tumor Biomarkers in Interpretable Machine Learning Models
• The percolating cluster is invisible to image recognition with deep learning
• A new semi-supervised self-training method for lung cancer prediction