ProCAN: Progressive Growing Channel Attentive Non-Local Network for Lung Nodule Classification

Abstract

Lung cancer classification in screening computed tomography (CT) scans is one of the most crucial tasks for early detection of this disease. Many lives can be saved if we are able to accurately classify malignant/cancerous lung nodules. Consequently, several deep learning based models have been proposed recently to classify lung nodules as malignant or benign. Nevertheless, the large variation in the size and heterogeneous appearance of the nodules makes this task an extremely challenging one. We propose a new Progressive Growing Channel Attentive Non-Local (ProCAN) network for lung nodule classification. The proposed method addresses this challenge from three different aspects. First, we enrich the Non-Local network by adding channel-wise attention capability to it. Second, we apply Curriculum Learning principles, whereby we first train our model on easy examples before hard ones. Third, as the classification task gets harder during the Curriculum learning, our model is progressively grown to increase its capability of handling the task at hand. We examined our proposed method on two different public datasets and compared its performance with state-of-the-art methods in the literature. The results show that the ProCAN model outperforms state-of-the-art methods and achieves an AUC of 98.05% and an accuracy of 95.28% on the LIDC-IDRI dataset. Moreover, we conducted extensive ablation studies to analyze the contribution and effects of each new component of our proposed method.

References in corpus (6)

• Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Self-Attention Generative Adversarial Networks
• Deep Learning: A Critical Appraisal
• Gated-Dilated Networks for Lung Nodule Classification in CT scans
• Data Augmentation Revisited: Rethinking the Distribution Gap between Clean and Augmented Data

Cited by in corpus (7)

• Is attention all you need in medical image analysis? A review
• Unsupervised Contrastive Learning based Transformer for Lung Nodule Detection
• 3D Axial-Attention for Lung Nodule Classification
• Deep fusion of gray level co-occurrence matrices for lung nodule classification
• Integration of Radiomics and Tumor Biomarkers in Interpretable Machine Learning Models
• A new semi-supervised self-training method for lung cancer prediction
• Learning from Ambiguous Labels for Lung Nodule Malignancy Prediction