| Literature DB >> 35795700 |
Ritu Tandon1, Shweta Agrawal1, Arthur Chang2, Shahab S Band3.
Abstract
Detection of malignant lung nodules from Computed Tomography (CT) images is a significant task for radiologists. But, it is time-consuming in nature. Despite numerous breakthroughs in studies on the application of deep learning models for the identification of lung cancer, researchers and doctors still face challenges when trying to deploy the model in clinical settings to achieve improved accuracy and sensitivity on huge datasets. In most situations, deep convolutional neural networks are used for detecting the region of the main nodule of the lung exclusive of considering the neighboring tissues of the nodule. Although the accuracy achieved through CNN is good enough but this models performance degrades when there are variations in image characteristics like: rotation, tiling, and other abnormal image orientations. CNN does not store relative spatial relationships among features in scanned images. As CT scans have high spatial resolution and are sensitive to misalignments during the scanning process, there is a requirement of a technique which helps in considering spatial information of image features also. In this paper, a hybrid model named VCNet is proposed by combining the features of VGG-16 and capsule network (CapsNet). VGG-16 model is used for object recognition and classification. CapsNet is used to address the shortcomings of convolutional neural networks for image rotation, tiling, and other abnormal image orientations. The performance of VCNeT is verified on the Lung Image Database Consortium (LIDC) image collection dataset. It achieves higher testing accuracy of 99.49% which is significantly better than MobileNet, Xception, and VGG-16 that has achieved an accuracy of 98, 97.97, and 96.95%, respectively. Therefore, the proposed hybrid VCNet framework can be used for the clinical purpose for nodule detection in lung carcinoma detection.Entities:
Keywords: CT; MobileNet; VCNet; VGG-16; Xception; capsule network; convolutional neural networks
Mesh:
Year: 2022 PMID: 35795700 PMCID: PMC9251197 DOI: 10.3389/fpubh.2022.894920
Source DB: PubMed Journal: Front Public Health ISSN: 2296-2565
Figure 1(A) Benign lung image. (B) Malignant lung.
Summary of the research articles on lung cancer diagnosis and classification using the DL model by CNN.
|
|
|
|
|
|
|
|
|---|---|---|---|---|---|---|
| Jakimovski et al. ( | 2019 | Lung cancer stage detection | CT images | LONI database | 99.63% | A deep neural network has been designed to detect the stage of lung cancer. The pre-classification of photos is done using the K-mean approach |
| Xie et al. ( | 2019 | Automated nodule detection | CT images | LUNA-16 | 86.42% | An automated pulmonary nodule detection-based model is proposed using two-dimensional CNN |
| Qi et al. ( | 2020 | Classification of lung cancer | PET and CT images | Department of Radiology of the Henan Provincial People's Hospital | 72% (Balanced Accuracy) | Used CT and PET multimodality noninvasive clinical images for lung cancer classification |
| Nakrani et al. ( | 2020 | Lung nodule detection | CT images | LIDC-IDRI | 95.24% | ResNet architecture of 2D convolutional neural network is used for detection of lung nodule |
| Neal Joshua et al. ( | 2021 | Lung nodule detection | CT images | LUNA-16 dataset | 97.17% | The 3D AlexNet architecture is being used. The 3D multi-dimensional convolutional neural network is used for lung nodule detection |
| Bharati et al. ( | 2020 | Lung disease detection | X-ray images | Chest X-ray dataset of NIH collected from kaggle | 73% | Proposed a hybrid model for lung disease detection using a modified capsule network |
| Afshar et al. ( | 2020 | Lung nodule malignancy prediction | CT images | LIDC-IDRI | 83% | The 3D multi-dimensional convolutional neural network is used for lung nodule detection |
Figure 2Methodology.
Figure 3Example of max & average pooling.
Figure 4ReLu activation function.
Model architecture.
|
|
|
|
|
| |
|---|---|---|---|---|---|
| 1 | MobileNet | 17 MB | 0.665 | 0.871 | 4,253,864 |
| 2 | Xception | 88 MB | 0.790 | 0.945 | 22,910,480 |
| 3 | VGG-16 | 528 MB | 0.715 | 0.901 | 138,357,544 |
| 4 | ResNet | 99 MB | 0.759 | 0.929 | 25,636,712 |
| 5 | Inception V3 | 92 MB | 0.788 | 0.944 | 23,851,784 |
Figure 5Architecture of VCNet.
Figure 6Proposed VCNet for the classification of lung nodule.
Figure 7CapsNet architecture.
Figure 8(A) MobileNet accuracy. (B) MobileNet loss.
Figure 9(A) Xception accuracy. (B) Xception loss.
Figure 10(A) VGG accuracy. (B) VGG model loss.
Figure 11(A) VCNet accuracy. (B) VCNet loss.
Comparative analyses in terms of accuracy among the proposed VCNET and the State-of-the-Art models.
|
|
|
|
|
|---|---|---|---|
| Jakimovski et al. ( | 2019 | LONI Database | 99.63% |
| Xie et al. ( | 2019 | LUNA-16 | 86.42% |
| Qi et al. ( | 2020 | Department of Radiology of the Henan Provincial People's Hospital | 72% |
| Nakrani et al. ( | 2020 | LIDC-IDRI | 95.24% |
| Neal Joshua et al. ( | 2021 | LUNA-16 dataset | 97.17% |
| Afshar et al. ( | 2020 | Chest X-ray dataset of NIH collected from kaggle | 73% |
| Bharati et al. ( | 2020 | LIDC-IDRI | 83% |
|
|
|
|
|
Classification matrix of different models using LIDC dataset.
|
|
|
|
|
|
|
|---|---|---|---|---|---|
| CNN | 95.88 ± 2.01 | 95 ± 1.84 | 96 ± 1.85 | 96 ± 1.92 | 96 ± 1.02 |
| Xception | 97.97 ± 1.02 | 96.87 ± 1.82 | 97.47 ± 1.05 | 97.97 ± 1.02 | 96.97 ± 1.02 |
| VGG-16 | 96.95 ± 1.75 | 96.95 ± 1.55 | 95.95 ± 1.75 | 96.95 ± 1.25 | 95.95 ± 1.84 |
| MobileNet | 98 ± 0.40 | 98 ± 0.55 | 98 ± 0.65 | 98 ± 0.40 | 98 ± 0.35 |
| ResNet | 93.5 ± 1.38 | 93.5 ± 1.55 | 93.5 ± 1.28 | 93.5 ± 1.4 | 93.5 ± 1.38 |
| Inception V3 | 94.35 ± 1.1 | 94.35 ± 1.15 | 94.35 ± 1.2 | 94.35 ± 1.10 | 94.35 ± 1.1 |
| VCNet |
Figure 12(A) Confusion Matrix MobileNet. (B) Confusion Matrix XceptionNet. (C) Confusion matrix VGG-16. (D) Confusion matrix VCNet.