Skin Cancer Classification Using Transfer Learning by VGG16 Architecture (Case Study on Kaggle Dataset)
- 1 Department of Computer Science, Sudan Technical University, Khartoum, Sudan
- 2 Faculty of Computer Science and Information Technology, Alzaiem Alazhri University, Khartoum, Sudan
- 3 Applied College, Najran University, Najran, Saudi Arabia
- 4 Faculty of Computer Science and Information Technology, Omdurman Islamic University, Omdurman, Sudan
- 5 Department of Emergency Medicine, King Abdullah Medical City, Makkah, Saudi Arabia
Abstract
Skin cancer is a serious and potentially life-threatening disease that affects millions of people worldwide. Early detection and accurate diagnosis are critical for successful treatment and improved patient outcomes. In recent years, deep learning has emerged as a powerful tool for medical image analysis, including the diagnosis of skin cancer. The importance of using deep learning in diagnosing skin cancer lies in its ability to analyze large amounts of data quickly and accurately. This can help doctors make more informed decisions about patient care and improve overall outcomes. Additionally, deep learning models can be trained to recognize subtle patterns and features that may not be visible to the human eye, leading to earlier detection and more effective treatment. The pre-trained Visual Geometry Group 16 (VGG16) architecture has been used in this study to classification of skin cancer images, and the images have been converted into other color scales, there are named: 1) Hue Saturation Value (HSV), 2) YCbCr, 3) Grayscale for evaluation. Results show that the dataset created with RGB and YCbCr images in field condition was promising with a classification accuracy of 84.242%. The dataset has also been evaluated with other popular architectures and compared. The performance of VGG16 with images of each color scale is analyzed. In addition, feature parameters have been extracted from the different layers. The extracted layers were felt with the VGG16 to evaluate the ability of the feature parameters in classifying the disease.
- Ali, M.S., Miah, M.S., Haque, J., Rahman, M.M. and Islam, M.K. (2021) An Enhanced Technique of Skin Cancer Classification Using Deep Convolutional Neural Network with Transfer Learning Models. Machine Learning with Applications, 5, Article ID: 100036. https://doi.org/10.1016/j.mlwa.2021.100036
- Carcagnì, P., Leo, M., Cuna, A., Mazzeo, P.L., Spagnolo, P., Celeste, G., et al. (2019) Classification of Skin Lesions by Combining Multilevel Learnings in a DenseNet Architecture. In: Ricci, E., Rota Bulò, S., Snoek, C., Lanz, O., Messelodi, S. and Sebe, N., Eds., Image Analysis and Processing—ICIAP 2019, Springer, Cham, 335-344. https://doi.org/10.1007/978-3-030-30642-7_30
- Dorj, U.O., Lee, K.K., Choi, J.Y. and Lee, M. (2018) The Skin Cancer Classification Using Deep Convolutional Neural Network. Multimedia Tools and Applications, 77, 9909-9924. https://doi.org/10.1007/s11042-018-5714-1
- Pacheco, A.G. and Krohling, R.A. (2019) Recent Advances in Deep Learning Applied to Skin Cancer Detection. arXiv: 1912.03280.
- Ashraf, R., Afzal, S., Rehman, A.U., Gul, S., Baber, J., Bakhtyar, M., Mehmood, I., Song, O.Y. and Maqsood, M. (2020) Region-of-Interest Based Transfer Learning Assisted Framework for Skin Cancer Detection. IEEE Access, 8, 147858-147871. https://doi.org/10.1109/ACCESS.2020.3014701
- Byrd, A.L., Belkaid, Y. and Segre, J.A. (2018) The Human Skin Microbiome. Nature Reviews Microbiology, 16, 143-155. https://doi.org/10.1038/nrmicro.2017.157
- Jeong, M.K., Lu, J.C., Huo, X., Vidakovic, B. and Chen, D. (2006) Wavelet-Based Data Reduction Techniques for Process Fault Detection. Technometrics, 48, 26-40. https://doi.org/10.1198/004017005000000553
- Namey, E., Guest, G., Thairu, L. and Johnson, L. (2008) Data Reduction Techniques for Large Qualitative Data Sets. Handbook for Team-Based Qualitative Research, 2, 137-161.
- Mahbod, A., Schaefer, G., Wang, C., Ecker, R. and Ellinge, I. (2019) Skin Lesion Classification Using Hybrid Deep Neural Networks. ICASSP 2019—2019 IEEE International Conference on Acoustics, Speech and Signal Processing, Brighton, 12-17 May 2019, 1229-1233. https://doi.org/10.1109/ICASSP.2019.8683352
- Sagar, A. and Dheeba, J. (2020) Convolutional Neural Networks for Classifying Melanoma Images. bioRxiv. https://www.biorxiv.org/content/10.1101/2020.05.22.110973v2
- Shaikh, J., Khan, R., Ingle, Y. and Shaikh, N. (2022) Skin Cancer Detection: A Review Using AI Techniques. International Journal of Health Sciences, 6, 14339-14346.