MELANOMA SKIN LESION CLASSIFICATION USING IMPROVED EFFICIENTNETB3


(Received: 4-Nov.-2021, Revised: 26-Dec.-2021 , Accepted: 12-Jan.-2022)
Malignant skin cancer is one the most common and lethal type of skin cancer. Early detection of cancerous skin lesions will increase the possibility of patient survival. In recent years, implementation of models built on deep neural networks in building medical diagnostic imaging systems is quite beneficial to medical experts. In this study, we present an improved and fine-tuned EfficientNetB3 model to classify malignant skin lesions using the concept of fine-tuning transfer learning. We have performed a comparative analysis of different deep learning pre-trained models, like ResNet50, InceptionV3, InceptionResNetV2 and EfficientNet B0-B2 models. The analysis findings signify the ability of utilizing fine-tuned EfficientNetB3 in the mission of melanoma detection and development of a computer-aided diagnostic system. All experimental procedures were carried out on ISBI-ISIC 2017 dataset. To check the efficiency of the proposed model, we compare the proposed model with EfficientNetB3 baseline model and present state-of-art pre-trained methods and approaches. The proposed EfficientNetB3 model obtained an accuracy of 87.12%, a recall of 87.00%, a precision of 87.00% and an F1 score of 85.00%. The proposed model achieved good computational results and efficaciously addressed the problem of model over- fitting and abated false negative labels.

[1] Cancer.org, "Cancer Facts and Figures 2021," [Online], Available: https://www.cancer.org/ content/ dam/cancer-org/research/cancer-facts-and-statistics/annual-cancer-facts-and-figures/2021/cancer-facts- and-figures-2021.pdf, 2021.

[2] S. Sonthalia, S. Yumeen and F. Kaliyadan, "Dermoscopy Overview and Extradiagnostic Applications," StatPearls [Internet], Treasure Island (FL): StatPearls Publishing; PMID: 30725816, [Online], Available: https://www.ncbi.nlm.nih.gov/books/NBK537131/, Jan. 2022.

[3] K. Munir, H. Elahi, A. Ayub, F. Frezza and A. Rizzi, "Cancer Diagnosis Using Deep Learning: A Bibliographic Review," Cancers (Basel), vol. 11, no. 9, p. 1235, DOI: 10.3390/cancers11091235, 2019.

[4] Isic-archive, "ISIC Challenge Datasets," [Online], Available: https://challenge.isic-archive.com/data/.

[5] A. Nazi, Zabir and Tasnim Azad Abir, "Automatic Skin Lesion Segmentation and Melanoma Detection: Transfer Learning Approach with U-Net and DCNN-SVM," Proc. Int'l conf. on Computational Intelligence, pp. 371-381. Springer, DOI: 10.1007/978-981-13-7564-4_32, Singapore, 2020.

[6] N. Gessert, M. Nielsen, M. Shaikh, R. Werner and A. Schlaefer, "Skin Lesion Classification Using Ensembles of Multi-resolution EfficientNets with Meta Data," MethodsX, vol. 7, no. 100864, p. 100864, DOI:10.1016/j.mex.2020.100864, 2020.

[7] H. Zunair and A. B. Hamza, "Melanoma Detection Using Adversarial Training and Deep Transfer Learning," Phys. Med. Biol., vol. 65, no. 13, p. 135005, DOI: 10.1088/1361-6560/ab86d3, 2020.

[8] J. R. Hagerty, R. J. Stanley, H. A. Almubarak et al., "Deep Learning and Handcrafted Method Fusion: Higher Diagnostic Accuracy for Melanoma Dermoscopy Images," IEEE Journal of Biomedical and Health Informatics, vol. 23, pp. 1385-1391, DOI: 10.1109/JBHI.2019.2891049, 2019.

[9] A. C. Foahom Gouabou, J.-L. Damoiseaux, J. Monnier, R. Iguernaissi, A. Moudafi and D. Merad, "Ensemble Method of Convolutional Neural Networks with Directed Acyclic Graph Using Dermoscopic Images: Melanoma Detection Application," Sensors (Basel), vol. 21, no. 12, p. 3999, DOI: 10.3390/s21123999, 2021.

[10] M. Frasca, M. Nappi, M. Risi, G. Tortora and A. A. Citarella, "A Comparison of Neural Network Approaches for Melanoma Classification," Proc. of the 25th IEEE International Conference on Pattern Recognition (ICPR), pp. 2110–2117, DOI:10.1109/ICPR48806.2021.9412893, Milan, Italy, 2021.

[11] S. P. G. Jasil and V. Ulagamuthalvi, "Deep Learning Architecture Using Transfer Learning for Classification of Skin Lesions," Journal of Ambient Intelligence and Humanized Computing, vol. 2021, DOI: 10.1007/s12652-021-03062-7, 2021.

[12] F. Zhuang, Q. Zhiyuan, D. Keyu, X. Dongbo, Z. Yongchun, Z. Hengshu, X. Hui and H. Qing, "A Comprehensive Survey on Transfer Learning," Proceedings of the IEEE, vol. 109, no. 1, pp. 43–76, DOI: 10.1109/JPROC.2020.3004555, 2021.

[13] A. Krizhevsky, I. Sutskever and G. E. Hinton, "ImageNet Classification with Deep Convolutional Neural Networks," Commun. of the ACM, vol. 60, no. 6, pp. 84–90, DOI: 10.1145/3065386, 2017.

[14] K. He, X. Zhang, S. Ren and J. Sun, "Deep Residual Learning for Image Recognition," Proc. of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 770–778, DOI: 10.1109/CVPR.2016.90, 2016.

[15] M. Tan and Q. V. Le, "EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks," Proc. of the Int. Conf. on Machine Learning, ArXiv, vol. abs/1905.11946, pp. 6105-6114, 2019.

[16] C. Szegedy, V. Vanhoucke, S. Ioffe, J. Shlens and Z. Wojna, "Rethinking the Inception Architecture for Computer Vision," Proc. of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), pp. 2818–2826, DOI: 10.1109/CVPR.2016.308, Las Vegas, NV, USA, 2016.

[17] C. Szegedy, S. Ioffe, V. Vanhoucke and A. Alemi, "Inception-v4, Inception-ResNet and the Impact of Residual Connections on Learning," Proc. of the 31st AAAI Conference on Artificial Intelligence (AAAI'17), vol. 31, no. 1, pp. 4278–4284, 2017.

[18] H. Lim, "A Study on Dropout Techniques to Reduce Overfitting in Deep Neural Networks," Proc. of Advanced Multimedia and Ubiquitous Engineering, Part of the Lecture Notes in Electrical Engineering Book Series, vol. 716, pp. 133-139, DOI: 10.1007/978-981-15-9309-3_20, 2021.

[19] T.-Y. Hsiao, Y.-C. Chang, H.-H. Chou and C.-T. Chiu, "Filter-based Deep-compression with Global Average Pooling for Convolutional Networks," Journal of Systems Architecture, vol. 95, pp. 9–18, DOI: 10.1016/j.sysarc.2019.02.008, 2019.

[20] A. El-Halees and M. Tafish, "Breast Cancer Severity Predication Using Deep Learning Techniques," Jordanian Journal of Computers and Information Technology (JJCIT), vol. 06, no. 01, pp. 94-102, 2020.

[21] S. F. Abuowaida and H. Y. Chan, "Improved Deep Learning Architecture for Depth Estimation from Single Image," Jordanian Journal of Computers and Information Technology (JJCIT), vol. 06, no. 04, pp. 434-445, 2020.