Enhanced Classification of Skin Lesions Using Fine-Tuned MobileNet and DenseNet121 Models with Ensemble Learning

Yıl 2024, Cilt: 17 Sayı: 3, 870 - 883, 31.12.2024

Yasin Sancar

https://doi.org/10.18185/erzifbed.1581918

Öz

This study presents a deep learning approach for early detection of melanoma, one of the most dangerous skin cancers. In this article, all pre-trained models of the Keras library are trained with the ISIC skin cancer dataset available on Kaggle and the accuracy of each model is analyzed in detail. With the results obtained from the trained models, the models were fine-tuned to further optimize the performance of each model. After re-evaluation with fine-tuning, the accuracy rates were compared: DenseNet121 and MobileNet were found to be the two best models with high accuracy among the fine-tuned models. As such, these two models were combined in an ensemble approach to achieve a better overall accuracy. The skin cancer detection rate obtained with this ensemble approach is 93.03%. Therefore, the deep learning-based ensemble method appears to be a reliable and powerful technique that can be used to diagnose serious diseases such as skin cancer. This model can be used to provide a powerful support system with great potential to assist dermatologists in the early detection phase by easing workload and improving patient outcomes.

Anahtar Kelimeler

Transfer Learning, Skin Cancer Classification, Fine Tuning, Model Ensemble

Kaynakça

• [1] Nisar, D.M., Amin, R., Shah, N.H., Al Ghamdi, M. A., Almotiri, S. H., & Alruily, M. (2021). Healthcare techniques through deep learning: issues, challenges and opportunities. IEEE Access, 9, 98523–98541.
• [2] Alfi, I. A., Rahman, M. M., Shorfuzzaman, M., & Nazir, A. (2022) A non-invasive interpretable diagnosis of melanoma skin cancer using deep learning and ensemble stacking of machine learning models. Diagnostics, 12(3), 726.
• [3] Anand, V., Gupta, S., Koundal, D., & Singh, K. (2023) Fusion of U-Net and CNN model for segmentation and classification of skin lesion from dermoscopy images. Expert Systems with Applications, 213, 119230.
• [4] Başaran, E., & Çelik, Y. (2022) Skin cancer diagnosis using CNN features with Genetic Algorithm and Particle Swarm Optimization methods. Transactions of the Institute of Measurement and Control, 01423312241253926.
• [5] Bazgir, E., Haque, E., Maniruzzaman, M., & Hoque, R. (2024) Skin cancer classification using Inception Network. World Journal of Advanced Research and Reviews, 21(2), 839–849.
• [6] Dimen, L., Tahâs, V.-S., Borşan, T., & Ferencz, Z. (2016) Ultraviolet Rays and Its Environmental Effects. International Multidisciplinary Scientific GeoConference: SGEM, 1, 787–792.
• [7] Hussein, Z. J., Hussein, A. M., Maki, G. I., & Gheni, H. Q. (2023) Improved model for skin illnesses classification utilizing gray-level co-occurrence matrix and convolution neural network. Journal of Advances in Information Technology, 14(6), 1273–1279.
• [8] Ker, J., Wang, L., Rao, J., & Lim, T. (2017) Deep learning applications in medical image analysis. Ieee Access, 6, 9375–9389.
• [9] Kilic, U., Karabey Aksakalli, I., Tumuklu Ozyer, G., Aksakalli, T., Ozyer, B., & Adanur, S. (2023) Exploring the Effect of Image Enhancement Techniques with Deep Neural Networks on Direct Urinary System (DUSX) Images for Automated Kidney Stone Detection. International Journal of Intelligent Systems, 2023(1), 3801485.
• [10] Latif, J., Xiao, C., Tu, S., Rehman, S. U., Imran, A., & Bilal, A. (2020) Implementation and use of disease diagnosis systems for electronic medical records based on machine learning: A complete review. IEEE Access, 8, 150489–150513.
• [11] Mohania, D., Chandel, S., Kumar, P., Verma, V., Digvijay, K., Tripathi, D., Shah, D. (2017) Ultraviolet radiations: Skin defense-damage mechanism. Ultraviolet Light in Human Health. Diseases and Environment, 71–87.
• [12] Sethanan, K., Pitakaso, R., Srichok, T., Khonjun, S., Thannipat, P., Wanram, S., … Others. (2023) Double AMIS-ensemble deep learning for skin cancer classification. Expert Systems with Applications, 234, 121047.
• [13] Shekar, B. H., & Hailu, H. (2023) Fusion of features extracted from transfer learning and handcrafted methods to enhance skin cancer classification performance. In Computer Vision and Machine Intelligence: Proceedings of CVMI 2022, Springer, 243–257.
• [14] Tumpa, P. P., & Kabir, M. A. (2021) An artificial neural network based detection and classification of melanoma skin cancer using hybrid texture features. Sensors International, 2, 100128.
• [15] Tuncer, T., Barua, P. D., Tuncer, I., Dogan, S., & Acharya, U. R. (2024) A lightweight deep convolutional neural network model for skin cancer image classification. Applied Soft Computing, 111794.
• [16] Panneerselvam, R., Balasubramaniam, S. (2023) Multi-Class Skin Cancer Classification using a hybrid dynamic salp swarm algorithm and weighted extreme learning machines with transfer learning. Acta Informatica Pragensia, 12(1), 141–159.
• [17] Prasad, C. R., Bilveni, G., Priyanka, B., Susmitha, C., Abhinay, D., & Kollem, S. (2024) Skin Cancer Prediction using Modified EfficientNet-B3 with Deep Transfer Learning. 2024 IEEE International Conference for Women in Innovation, Technology & Entrepreneurship (ICWITE), 519–523.
• [18] Srivastava, N., Hinton, G., Krizhevsky, A., Sutskever, I., & Salakhutdinov, R. (2014) Dropout: a simple way to prevent neural networks from overfitting. The Journal of Machine Learning Research, 15(1), 1929–1958.
• [19] Dietterich, T. G. (2000) Ensemble methods in machine learning. International Workshop on Multiple Classifier Systems, Springer, 1–15.
• [20] Lin, M. (2013) Network in network. arXiv Preprint arXiv:1312. 4400.
• [21] Fanconi, C. (2019) Skin cancer: malignant vs. benign-processed skin cancer pictures of the ISIC archive.