| Literature DB >> 34884146 |
Satin Jain1, Udit Singhania2, Balakrushna Tripathy1, Emad Abouel Nasr3, Mohamed K Aboudaif3, Ali K Kamrani4.
Abstract
One of the major health concerns for human society is skin cancer. When the pigments producing skin color turn carcinogenic, this disease gets contracted. A skin cancer diagnosis is a challenging process for dermatologists as many skin cancer pigments may appear similar in appearance. Hence, early detection of lesions (which form the base of skin cancer) is definitely critical and useful to completely cure the patients suffering from skin cancer. Significant progress has been made in developing automated tools for the diagnosis of skin cancer to assist dermatologists. The worldwide acceptance of artificial intelligence-supported tools has permitted usage of the enormous collection of images of lesions and benevolent sores approved by histopathology. This paper performs a comparative analysis of six different transfer learning nets for multi-class skin cancer classification by taking the HAM10000 dataset. We used replication of images of classes with low frequencies to counter the imbalance in the dataset. The transfer learning nets that were used in the analysis were VGG19, InceptionV3, InceptionResNetV2, ResNet50, Xception, and MobileNet. Results demonstrate that replication is suitable for this task, achieving high classification accuracies and F-measures with lower false negatives. It is inferred that Xception Net outperforms the rest of the transfer learning nets used for the study, with an accuracy of 90.48. It also has the highest recall, precision, and F-Measure values.Entities:
Keywords: CNN; artificial intelligence; image classification; skin lesion; transfer learning
Mesh:
Year: 2021 PMID: 34884146 PMCID: PMC8662405 DOI: 10.3390/s21238142
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Figure 1Seven different types of diseases caused from lesions.
Figure 2Occurrence of images of each type of skin cancer.
Figure 3Process flow diagram of the proposed method.
Data augmentation of the dataset.
| Disease | Frequency before Augmentation | Multiply Factor (k) | Frequency after Augmentation |
|---|---|---|---|
| Melanocytic Nevi | 3179 | 1 | 3179 |
| Benign Keratosis | 317 | 10 | 3170 |
| Melanoma | 165 | 19 | 3135 |
| Basal Cell Carcinoma | 126 | 25 | 3150 |
| Actinic Keratosis | 109 | 29 | 3161 |
| Vascular Skin Lesions | 46 | 69 | 3174 |
| Dermatofibroma | 28 | 110 | 3080 |
Performance of transfer learning nets without repetition of images.
| Model without Repetition | Accuracy | Avg. Recall | Avg. Precision | Avg. F-Measure |
|---|---|---|---|---|
| VGG19 | 0.6718 | 0.67 | 0.78 | 0.71 |
| InceptionV3 | 0.8168 | 0.82 | 0.75 | 0.78 |
| InceptionResnetV2 | 0.8114 | 0.81 | 0.82 | 0.80 |
| ResNet50 | 0.8105 | 0.81 | 0.75 | 0.77 |
| Xception | 0.8096 | 0.81 | 0.78 | 0.78 |
| MobileNet | 0.8241 | 0.82 | 0.84 | 0.80 |
Performance of transfer learning nets with repetition of images.
| Model with Repetition | Accuracy | Avg. Recall | Avg. Precision | Avg. F-Measure |
|---|---|---|---|---|
| VGG19 | 0.66 | 0.66 | 0.86 | 0.72 |
| InceptionV3 | 0.79 | 0.79 | 0.87 | 0.82 |
| InceptionResnetV2 | 0.85 | 0.86 | 0.88 | 0.86 |
| ResNet50 | 0.77 | 0.78 | 0.86 | 0.80 |
| Xception | 0.90 | 0.90 | 0.90 | 0.90 |
| MobileNet | 0.88 | 0.89 | 0.88 | 0.88 |
Figure 4Confusion matrix results of different nets: (a) VGG19; (b) InceptionV3; (c) InceptionResNetV2; (d) ResNet50; (e) Xception; (f) MobileNet.
Figure 5Fraction classified incorrectly for all seven models: (a) VGG19; (b) InceptionV3; (c) InceptionResNetV2; (d) ResNet50; (e) Xception; (f) MobileNet.
Figure 6Training and validation accuracies and training and validation loss: (a) VGG19; (b) InceptionV3; (c) InceptionResNetV2; (d) ResNet50; (e) Xception; (f) MobileNet.
Precision, recall, F-Measure, and accuracy values of the models.
| Model | Accuracy | Avg. Recall | Avg. Precision | Avg. F-Measure |
|---|---|---|---|---|
| VGG19 | 0.6754 | 0.6734 | 0.8548 | 0.7479 |
| InceptionV3 | 0.8640 | 0.8619 | 0.8769 | 0.8713 |
| InceptionResnetV2 | 0.8840 | 0.8762 | 0.8793 | 0.8845 |
| ResNet50 | 0.8232 | 0.8222 | 0.8680 | 0.8416 |
| Xception |
|
|
|
|
| MobileNet | 0.8721 | 0.8711 | 0.8838 | 0.8740 |
Xcpetion Net precision, recall, and F1-Score values.
| Disease | Avg. Precision | Avg. Recall | Avg. F-Measure |
|---|---|---|---|
| Melanocytic Nevi | 0.94 | 0.98 | 0.96 |
| Benign Keratosis | 0.68 | 0.68 | 0.68 |
| Melanoma | 0.58 | 0.48 | 0.52 |
| Basal Cell Carcinoma | 0.88 | 0.80 | 0.84 |
| Actinic Keratosis | 0.92 | 0.37 | 0.52 |
| Vascular Skin Lesions | 1.0 | 0.69 | 0.82 |
| Dermatofibroma | 0.71 | 0.62 | 0.67 |
Test accuracy and loss values of all learning nets used.
| Transfer Learning Nets | Accuracy | Loss |
|---|---|---|
| VGG19 | 66.36 | 1.0134 |
| Resnet50 | 77.60 | 0.6855 |
| InceptionResNetV2 | 85.58 | 0.6745 |
| InceptionV3 | 79.23 | 0.6665 |
| Xception | 90.48 | 0.5168 |
| MobileNet | 88.57 | 0.6347 |
Hardware specification.
| Hardware Use | Specification |
|---|---|
| NVIDIA GPU | Tesla P100 |
| CUDA Version | 9.2 |
| GPU RAM (GB) | 17.1 |
| CPU Chip | Intel Xeon CPU |
| Chip Speed (GHz) | 2.2 or 2.3 |
| CPU Cores | 2 |
| CPU RAM (Total GB) | 16.4 |
| L3 Cache (MB) | 46 |
| Disk Space (Total GB) | 220 |
Computation time.
| Model Name | Computational Time |
|---|---|
| VGG19 | 746.84069 |
| InceptionV3 | 751.12284 |
| InceptionResnetV2 | 2456.34356 |
| ResNet50 | 761.63929 |
| Xception | 834.66028 |
| MobileNet | 695.36065 |