Automated Lung Nodule Candidate Detection Using An Iteratively Optimized Multi-Resolution 3D Depthwise Separable Cnns With Effective Training Initialization

Document Type : Primary Research paper

Authors

1 Assistant Professor, Department of Computer Applications, Sri Ramakrishna College of Arts and Science, Coimbatore, India

2 Assistant Professor, Department of Computer Science, Sri Ramakrishna College of Arts and Science, Coimbatore, India

Abstract

An earlier detection and diagnosis of lung cancer requires a major task known as lung nodule candidate classification. To detect the lung nodule candidate, a Multi-Resolution 3-Dimensional Convolutional Neural Network and Knowledge Transfer (MR3DCNN-KT) model has been designed that can extract the contextual information between multiple samples of lung nodule image for increasing the detection accuracy. But, this model was not able to classify few types of nodules that may cause the false detection. Also, the training data preparation was high difficult due to the manual labeling that consumes more time and the label mistakes were introduced while using large scale datasets since 3D-CNN requires more number of samples. Hence this article proposes an Iteratively Optimized MR3DCNN-KT (IO-MR3DCNN-KT) model that establishes automated weak label initialization to classify the large scale lung nodule image datasets. This model is trained on dynamically updated training datasets in an iterative manner. A fast and automatic weak labeling scheme is applied to generate the initial training dataset. Nonetheless, the computational complexity of 3D-CNN structure is extremely high since it requires the significant number of computational resources. As a result, an IO-MR3D Depthwise Separable CNN and KT (IO-MR3D-DSCNN-KT) model is proposed that introduces the bottleneck-based 3D-DSCNN structure to reduce the computational complexity. This model can extract both spatial and temporal features using basic depthwise convolution and pointwise convolution, accordingly. Based on this model, the number of parameters used in the 3D-CNN structure is significantly reduced to automatically classify the lung nodule candidates. Finally, the experimental results show that the proposed model promises more accuracy and robustness compared to the MR3DCNN-KT model.

Keywords