| Literature DB >> 35746208 |
Iftikhar Naseer1, Sheeraz Akram1, Tehreem Masood1, Arfan Jaffar1, Muhammad Adnan Khan2, Amir Mosavi3,4,5.
Abstract
The convolutional neural network (CNN) has become a powerful tool in machine learning (ML) that is used to solve complex problems such as image recognition, natural language processing, and video analysis. Notably, the idea of exploring convolutional neural network architecture has gained substantial attention as well as popularity. This study focuses on the intrinsic various CNN architectures: LeNet, AlexNet, VGG16, ResNet-50, and Inception-V1, which have been scrutinized and compared with each other for the detection of lung cancer using publicly available LUNA16 datasets. Furthermore, multiple performance optimizers: root mean square propagation (RMSProp), adaptive moment estimation (Adam), and stochastic gradient descent (SGD), were applied for this comparative study. The performances of the three CNN architectures were measured for accuracy, specificity, sensitivity, positive predictive value, false omission rate, negative predictive value, and F1 score. The experimental results showed that the CNN AlexNet architecture with the SGD optimizer achieved the highest validation accuracy for CT lung cancer with an accuracy of 97.42%, misclassification rate of 2.58%, 97.58% sensitivity, 97.25% specificity, 97.58% positive predictive value, 97.25% negative predictive value, false omission rate of 2.75%, and F1 score of 97.58%. AlexNet with the SGD optimizer was the best and outperformed compared to the other state-of-the-art CNN architectures.Entities:
Keywords: AlexNet; LUNA16; LeNet; artificial intelligence; big data; cancer research; deep learning; lung cancer; machine learning; medical image analysis
Mesh:
Year: 2022 PMID: 35746208 PMCID: PMC9227226 DOI: 10.3390/s22124426
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.847
Literature survey on computational intelligence-based lung cancer detection methods.
| Publications | Method | Dataset | Accuracy% | Weakness |
|---|---|---|---|---|
| Khehrah et al. [ | ANN | LIDC-IDRI | 92.0% | Requires handcrafted features |
| Xie et al. [ | ANN | LIDC-IDRI | 89.62% | Requires handcrafted features |
| Naseer et al. [ | ANN | Private Lung Dataset | 96.67% | Requires handcrafted features |
| [ | ANN | Private Lung Dataset | 96.67% | Requires handcrafted features |
| S. Zhang et al. [ | LeNet-5 | LIDC-IDRI | 97.04% | Complexity required |
| Zhao et al. [ | hybrid CNN of LeNet and | LIDC-IDRI | 87.7% | Complexity required |
| Agarwal et al. [ | AlexNet CNN | Private Lung Dataset | 96.0% | Less number images |
| Polat et al. [ | Hybrid 3D-CNN RBF-based | LUNA16 | 91.81% | Complexity required |
| Al-Yasriy et al. [ | AlexNet CNN | (IQ-OTH/NCCD) lung cancer dataset | 93.548% | Use of imbalance dataset |
| A. Elnakib et al. [ | VGG19 architecture and SVM classifier | Early Lung Cancer Action Project (ELCAP) database | 96.25% | Less number images |
| Nibali et al. [ | ResNet-18 architecture | LIDC-IDRI | 89.90% | Needs to improve accuracy |
| Zheng et al. [ | Inception CNN classifier | AIA-INF | 88.67% | Needs to improve accuracy |
| Haibo et al. [ | DarkNet-53 CNN architecture | LUNA16 | 73.9% | Needs to improve accuracy |
Figure 1Convolutional neural network architecture.
LeNet confusion matrix (validation).
| CNN Architecture | Optimizer | True Negative | False Positive | False Negative | True Positive |
|---|---|---|---|---|---|
| LeNet | RMSprop | 204 | 14 | 9 | 239 |
| LeNet | Adam | 211 | 7 | 15 | 233 |
| LeNet | SGD | 212 | 6 | 13 | 235 |
Validation statistical analysis of the LeNet model.
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| LeNet RMSprop | 95.06% | 96.37% | 93.58% | 94.47% | 95.77% | 4.23% | 95.41% |
| LeNet Adam | 95.18% | 93.7% | 96.79% | 96.96% | 93.36% | 6.63% | 95.3% |
| LeNet SGD | 95.92% | 94.76% | 97.25% | 97.51% | 94.22% | 5.78% | 96.11% |
AlexNet confusion matrix (validation).
| CNN Architecture | Optimizer | True Negative | False Positive | False Negative | True Positive |
|---|---|---|---|---|---|
| AlexNet | RMSprop | 206 | 12 | 10 | 238 |
| AlexNet | Adam | 205 | 13 | 7 | 241 |
| AlexNet | SGD | 212 | 6 | 6 | 242 |
Validation statistical analysis of the AlexNet model.
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| AlexNet RMSprop | 95.28% | 95.97% | 94.5% | 95.20% | 95.37% | 4.63% | 95.58% |
| AlexNet Adam | 95.71% | 97.18% | 94.04% | 94.88% | 96.70% | 3.30% | 96.02% |
| AlexNet SGD | 97.42% | 97.58% | 97.25% | 97.58% | 97.25% | 2.75% | 97.58% |
VGG16 confusion matrix (validation).
| CNN Architecture | Optimizer | True Negative | False Positive | False Negative | True Positive |
|---|---|---|---|---|---|
| VGG16 | RMSprop | 204 | 14 | 21 | 227 |
| VGG16 | Adam | 203 | 15 | 19 | 229 |
| VGG16 | SGD | 209 | 9 | 21 | 227 |
Validation Statistical Analysis of the VGG16 model.
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| VGG16 RMSprop | 92.49% | 91.53% | 93.58% | 94.19% | 90.67% | 9.33% | 92.84% |
| VGG16 Adam | 92.70% | 92.34% | 93.12% | 93.85% | 91.44% | 8.56% | 93.09% |
| VGG16 SGD | 93.56% | 91.53% | 95.87% | 96.19% | 90.87% | 9.13% | 93.80% |
ResNet 50 confusion matrix (validation).
| CNN Architecture | Optimizer | True Negative | False Positive | False Negative | True Positive |
|---|---|---|---|---|---|
| ResNet 50 | RMSprop | 207 | 11 | 18 | 230 |
| ResNet 50 | Adam | 211 | 7 | 16 | 234 |
| ResNet 50 | SGD | 212 | 6 | 11 | 237 |
Validation statistical analysis of the ResNet 50 model.
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| ResNet 50 RMSprop | 93.78% | 92.74% | 94.95% | 95.44% | 92.0% | 8.0% | 94.07% |
| ResNet 50 Adam | 95.09% | 93.60% | 96.79% | 97.10% | 92.95% | 7.04% | 95.32% |
| ResNet 50 SGD | 96.35% | 95.56% | 97.25% | 97.53% | 95.07% | 4.93% | 96.54% |
Inception-V1 confusion matrix (validation).
| CNN Architecture | Optimizer | True Negative | False Positive | False Negative | True Positive |
|---|---|---|---|---|---|
| Inception-V1 | RMSprop | 206 | 12 | 28 | 220 |
| Inception-V1 | Adam | 210 | 8 | 20 | 228 |
| Inception-V1 | SGD | 211 | 7 | 16 | 232 |
Validation statistical analysis of the Inception-V1 model.
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| Inception-V1 RMSprop | 91.42% | 88.71% | 94.50% | 94.83% | 88.03% | 11.97% | 91.67% |
| Inception-V1 Adam | 93.99% | 91.94% | 96.33% | 96.61% | 91.30% | 8.70% | 94.21% |
| Inception-V1 SGD | 95.06% | 93.55% | 96.79% | 97.07% | 92.95% | 7.05% | 95.28% |
Detection results of AlexNet with the SGD optimizer on the LUNA16 dataset.
| Detection Class | |||
|---|---|---|---|
| Benign | Malignant | ||
|
| Benign |
|
|
| Malignant |
|
| |
Figure 2Performance evaluation with statistical parameters for LeNet, AlexNet, VGG16, ResNet-50, and Inception-V1 (training accuracy).
Figure 3Performance evaluation with statistical parameters for LeNet, AlexNet, VGG16, ResNet-50, and Inception-V1 (validation accuracy).
Validation statistical analysis of five CNN architectures with the SGD optimizer confusion matrix (5-fold cross-validation).
| CNN Architecture | Accuracy | Sensitivity | Specificity | PPV | NPV | FOR | F1-Score |
|---|---|---|---|---|---|---|---|
| LeNet SGD | 93.56% | 92.74% | 94.50% | 95.04% | 91.96% | 8.045% | 93.88% |
| AlexNET SGD | 95.73% | 95.20% | 96.33% | 96.75% | 94.59% | 5.41% | 95.97% |
| VGG16 SGD | 93.56% | 89.92% | 97.71% | 97.81% | 89.50% | 10.50% | 93.70% |
| ResNet-50 SGD | 95.28% | 96.77% | 93.58% | 94.49% | 96.23% | 3.77% | 95.62% |
| Inception-V1 SGD | 91.85% | 87.1% | 97.25% | 97.30% | 86.89% | 13.11% | 91.91% |
Comparison of AlexNet with the SGD model with previously published approaches.
| Publications | Method | Dataset | Accuracy% | Misclassification Rate |
|---|---|---|---|---|
| Khehrah et al. [ | ANN | LIDC-IDRI | 92.0% | 8.0% |
| Xie et al. [ | ANN | LIDC-IDRI | 89.62% | 10.38% |
| Naseer et al. [ | ANN | Private Lung Dataset | 96.67% | 3.33% |
| [ | ANN | Private Lung Dataset | 96.67% | 3.33% |
| S. Zhang et al. [ | LeNet-5 | LIDC-IDRI | 97.04% | 2.96% |
| Zhao et al. [ | Hybrid CNN of LeNet and AlexNet | LIDC-IDRI | 87.7% | 12.3% |
| Agarwal et al. [ | AlexNet CNN | Private Lung Dataset | 96.0% | 4.0% |
| Polat et al. [ | Hybrid 3D-CNN RBF-based | LUNA16 | 91.81% | 8.19% |
| Al-Yasriy et al. [ | AlexNet CNN | (IQ-OTH/NCCD) Lung Cancer Dataset | 93.548% | 6.45% |
| A. Elnakib et al. [ | VGG19 architecture and SVM classifier | Early Lung Cancer Action Project (ELCAP) Database | 96.25% | 3.75% |
| Nibali et al. [ | ResNet-18 architecture | LIDC-IDRI | 89.90% | 10.1% |
| Zheng et al. [ | Inception CNN classifier | AIA-INF | 88.67% | 11.33% |
| Haibo et al. [ | DarkNet-53 CNN architecture | LUNA16 | 73.9% | 26.1% |
| Best Model: AlexNet SGD with 5-fold Cross-Validation | AlexNet with SGD | LUNA16 | 95.73% | 4.27% |
| Best Model: AlexNet with SGD | AlexNet with SGD | LUNA16 | 97.42% | 2.58% |