Skin Cancer Classification and Comparison of Pre-trained Models Performance using Transfer Learning
Downloads
Background: Skin cancer can quickly become fatal. An examination and biopsy of dermoscopic pictures are required to determine if skin cancer is malignant or benign. However, these examinations can be costly.
Objective: In this research, we proposed deep learning (DL)-based approach to identify a melanoma, the most dangerous kind of skin cancer. DL is particularly excellent in learning traits and predicting cancer. However, DL requires a vast number of images.
Method: We used image augmentation and transferring learning to categorise images into benign and malignant. We used the public ISIC 2020 database to train and test our models. The ISIC 2020 dataset classifies melanoma as malignant. Along with the categorization, the dataset was examined for variation. The training and validation accuracy of three of the best pre-trained models were compared. To minimise the loss, three optimizers were used: RMSProp, SGD, and ADAM.
Results: We attained training accuracy of 98.73%, 99.12%, and 99.76% using ResNet, VGG16, and MobileNetV2, respectively. We achieved a validation accuracy of 98.39% using these three pre-trained models.
Conclusion: The validation accuracy of 98.39% outperforms the prior pre-trained model. The findings of this study can be applied in medical science to help physicians diagnose skin cancer early and save lives.
Keywords: Deep Learning, ISIC 2020, Pre-trained Model, Skin Cancer, Transfer Learning
S. Hosseinzadeh Kassani and P. Hosseinzadeh Kassani, “A comparative study of deep learning architectures on melanoma detection,” Tissue and Cell, vol. 58, pp. 76–83, Jun. 2019, doi: 10.1016/j.tice.2019.04.009.
F. Zou, L. Shen, Z. Jie, W. Zhang, and W. Liu, “A Sufficient Condition for Convergences of Adam and RMSProp,” 2019, pp. 11127–11135. Accessed: Aug. 23, 2022. [Online]. Available: https://openaccess.thecvf.com/content_CVPR_2019/html/Zou_A_Sufficient_Condition_for_Convergences_of_Adam_and_RMSProp_CVPR_2019_paper.html
D. P. Kingma and J. Ba, “Adam: A Method for Stochastic Optimization,” Dec. 2014, doi: 10.48550/arXiv.1412.6980.
M. S. Ali, M. S. Miah, J. Haque, M. M. Rahman, and M. K. Islam, “An enhanced technique of skin cancer classification using deep convolutional neural network with transfer learning models,” Machine Learning with Applications, vol. 5, p. 100036, Sep. 2021, doi: 10.1016/j.mlwa.2021.100036.
F. A. Spanhol, L. S. Oliveira, C. Petitjean, and L. Heutte, “Breast cancer histopathological image classification using Convolutional Neural Networks,” in 2016 International Joint Conference on Neural Networks (IJCNN), Jul. 2016, pp. 2560–2567. doi: 10.1109/IJCNN.2016.7727519.
N. M. Astudillo, R. Bolman, and N. M. Sirakov, “Classification with Stochastic Learning Methods and Convolutional Neural Networks,” SN COMPUT. SCI., vol. 1, no. 3, p. 119, Apr. 2020, doi: 10.1007/s42979-020-00126-x.
T.-C. Pham, C.-M. Luong, M. Visani, and V.-D. Hoang, “Deep CNN and Data Augmentation for Skin Lesion Classification,” in Intelligent Information and Database Systems, Cham, 2018, pp. 573–582. doi: 10.1007/978-3-319-75420-8_54.
A. Esteva et al., “Dermatologist-level classification of skin cancer with deep neural networks,” Nature, vol. 542, no. 7639, Art. no. 7639, Feb. 2017, doi: 10.1038/nature21056.
R. R. Winkelmann, A. S. Farberg, N. Tucker, R. White, and D. S. Rigel, “Enhancement of International Dermatologists’ Pigmented Skin Lesion Biopsy Decisions Following Dermoscopy with Subsequent Integration of Multispectral Digital Skin Lesion Analysis,” J Clin Aesthet Dermatol, vol. 9, no. 7, pp. 53–55, Jul. 2016.
U.-O. Dorj, K.-K. Lee, J.-Y. Choi, and M. Lee, “The skin cancer classification using deep convolutional neural network,” Multimed Tools Appl, vol. 77, no. 8, pp. 9909–9924, Apr. 2018, doi: 10.1007/s11042-018-5714-1.M. Mateen, J. Wen, Nasrullah, S. Song, and Z. Huang, “Fundus Image Classification Using VGG-19 Architecture with PCA and SVD,” Symmetry, vol. 11, no. 1, Art. no. 1, Jan. 2019, doi: 10.3390/sym11010001.
Q. Gao, S. Lim, and X. Jia, “Hyperspectral Image Classification Using Convolutional Neural Networks and Multiple Feature Learning,” Remote Sensing, vol. 10, no. 2, Art. no. 2, Feb. 2018, doi: 10.3390/rs10020299.
Z. A. Oraibi, H. Yousif, A. Hafiane, G. Seetharaman, and K. Palaniappan, “Learning Local and Deep Features for Efficient Cell Image Classification Using Random Forests,” in 2018 25th IEEE International Conference on Image Processing (ICIP), Oct. 2018, pp. 2446–2450. doi: 10.1109/ICIP.2018.8451287.
D. Theckedath and R. R. Sedamkar, “Detecting Affect States Using VGG16, ResNet50 and SE-ResNet50 Networks,” SN COMPUT. SCI., vol. 1, no. 2, p. 79, Mar. 2020, doi: 10.1007/s42979-020-0114-9.
K. M. Hosny, M. A. Kassem, and M. M. Fouad, “Classification of Skin Lesions into Seven Classes Using Transfer Learning with AlexNet,” J Digit Imaging, vol. 33, no. 5, pp. 1325–1334, Oct. 2020, doi: 10.1007/s10278-020-00371-9.
I. Z. Mukti and D. Biswas, “Transfer Learning Based Plant Diseases Detection Using ResNet50,” in 2019 4th International Conference on Electrical Information and Communication Technology (EICT), Dec. 2019, pp. 1–6. doi: 10.1109/EICT48899.2019.9068805.
M. Sandler, A. Howard, M. Zhu, A. Zhmoginov, and L.-C. Chen, “MobileNetV2: Inverted Residuals and Linear Bottlenecks,” 2018, pp. 4510–4520. Accessed: Sep. 28, 2022. [Online]. Available: https://openaccess.thecvf.com/content_cvpr_2018/html/Sandler_MobileNetV2_Inverted_Residuals_CVPR_2018_paper.html
D. A. van Dyk and X.-L. Meng, “The Art of Data Augmentation,” Journal of Computational and Graphical Statistics, vol. 10, no. 1, pp. 1–50, Mar. 2001, doi: 10.1198/10618600152418584.
S. J. Pan and Q. Yang, “A Survey on Transfer Learning,” IEEE Transactions on Knowledge and Data Engineering, vol. 22, no. 10, pp. 1345–1359, Oct. 2010, doi: 10.1109/TKDE.2009.191.
T. Kattenborn, J. Leitloff, F. Schiefer, and S. Hinz, “Review on Convolutional Neural Networks (CNN) in vegetation remote sensing,” ISPRS Journal of Photogrammetry and Remote Sensing, vol. 173, pp. 24–49, Mar. 2021, doi: 10.1016/j.isprsjprs.2020.12.010.
A. Krizhevsky, I. Sutskever, and G. E. Hinton, “ImageNet classification with deep convolutional neural networks,” Commun. ACM, vol. 60, no. 6, pp. 84–90, May 2017, doi: 10.1145/3065386.
H. Dalianis, “Evaluation Metrics and Evaluation,” in Clinical Text Mining: Secondary Use of Electronic Patient Records, H. Dalianis, Ed. Cham: Springer International Publishing, 2018, pp. 45–53. doi: 10.1007/978-3-319-78503-5_6.
S. Banerjee, S. K. Singh, A. Chakraborty, A. Das, and R. Bag, “Melanoma Diagnosis Using Deep Learning and Fuzzy Logic,” Diagnostics, vol. 10, no. 8, Art. no. 8, Aug. 2020, doi: 10.3390/diagnostics10080577.
J. Rashid et al., “Skin Cancer Disease Detection Using Transfer Learning Technique,” Applied Sciences, vol. 12, no. 11, Art. no. 11, Jan. 2022, doi: 10.3390/app12115714.
R. Indraswari, R. Rokhana, and W. Herulambang, “Melanoma image classification based on MobileNetV2 network,” Procedia Computer Science, vol. 197, pp. 198–207, Jan. 2022, doi: 10.1016/j.procs.2021.12.132.
Copyright (c) 2022 The Authors. Published by Universitas Airlangga.
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
All accepted papers will be published under a Creative Commons Attribution 4.0 International (CC BY 4.0) License. Authors retain copyright and grant the journal right of first publication. CC-BY Licenced means lets others to Share (copy and redistribute the material in any medium or format) and Adapt (remix, transform, and build upon the material for any purpose, even commercially).
