Skin Cancer Classification Using Fine-Tuned Transfer Learning of DENSENET-121
Source abstract
Abstract
Skin cancer diagnosis greatly benefits from advanced machine learning techniques, particularly fine-tuned deep learning models. In our research, we explored the impact of traditional machine learning and fine-tuned deep learning approaches on prediction accuracy. Our findings reveal significant improvements in predictability and accuracy with fine-tuning, particularly evident in deep learning models. The CNN, SVM, and Random Forest Classifier achieved high accuracy. However, fine-tuned deep learning models such as EfficientNetB0, ResNet34, VGG16, Inception _v3, and DenseNet121 demonstrated superior performance. To ensure comparability, we fine-tuned these models by incorporating additional layers, including one flatten layer and three densely interconnected layers. These layers play a crucial role in enhancing model efficiency and performance. The flatten layer preprocesses multidimensional feature maps, facilitating efficient information flow, while subsequent dense layers refine feature representations, capturing intricate patterns and relationships within the data. Leveraging LeakyReLU activation functions in the dense layers mitigates the vanishing gradient problem and promotes stable training. Finally, the output dense layer with a sigmoid activation function simplifies decision making for healthcare professionals by providing binary classification output. Our study underscores the significance of incorporating additional layers in fine-tuned neural network models for skin cancer classification, offering improved accuracy and reliability in diagnosis.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Cutaneous Melanoma Detection and Management · AI in cancer detection · Nonmelanoma Skin Cancer Studies
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Computing and Information Science, Faculty of Science and Engineering, Anglia Ruskin University, Cambridge CB1 1PT, UK
References (37)
Show source-provided references
1. (2023, June 05). Difference between Melanoma & Nonmelanoma Skin Cancer. Available online: https://moffitt.org/cancers/skin-cancer-nonmelanoma/faqs/what-is-the-difference-between-melanoma-and-nonmelanoma-skin-cancer/.
2. (2023, June 05). Radiation: Ultraviolet (UV) Radiation and Skin Cancer (No Date) World Health Organization. Available online: https://www.who.int/news-room/questions-and-answers/item/radiation-ultraviolet-(uv)-radiation-and-skin-cancer.
3. (2023, June 05). Skin Cancer Statistics: World Cancer Research Fund International. Available online: https://www.wcrf.org/cancer-trends/skin-cancer-statistics/.
4. Skin cancer prediction using Deep Learning10.48175/IJARSCT-8541 ↗
5. Transcription of human papillomaviruses in non-melanoma skin cancers of the immunosuppressed10.1002/ijc.33683 ↗
6. Vardasca, R., and Magalhaes, C. (2022). Towards an effective imaging-based decision support system for skin cancer. Handbook of Research on Applied Intelligence for Health and Clinical Informatics, IGI Global.10.4018/978-1-7998-7709-7.ch021 ↗
7. Larre Borges, A., Nicoletti, S., Dufrechou, L., and Nicola Centanni, A. (2017). Dermatoscopy in the Public Health Environment. Dermatology in Public Health Environments, Springer International Publishing.10.1007/978-3-319-33919-1_56 ↗
8. Chatterjee, S., Dey, D., Munshi, S., and Gorai, S. (2019). Extraction of features from cross correlation in space and frequency domains for classification of skin lesions. Biomed. Signal Process. Control, 53.10.1016/j.bspc.2019.101581 ↗
9. Nanoengineered targeting strategy for cancer immunotherapy10.1038/s41401-020-0417-3 ↗
10. Pathological brain detection based on AlexNet and transfer learning10.1016/j.jocs.2018.11.008 ↗
11. Smart clothing system with multiple sensors based on digital twin technology10.1109/JIOT.2022.3224947 ↗
12. Bello, A., Ng, S.-C., and Leung, M.-F. (2023). A Bert framework to sentiment analysis of Tweets. Sensors, 23.10.3390/s23010506 ↗
13. Tensor Factorization with Sparse and Graph Regularization for Fake News Detection on Social Networks10.1109/TCSS.2023.3296479 ↗
14. A comprehensive implementation of road surface classification for vehicle driving assistance: Dataset, models, and deployment10.1109/TITS.2023.3264588 ↗
15. Intelligent wearable system with motion and emotion recognition based on digital twin technology10.1109/JIOT.2024.3394244 ↗
16. Donges, N. (2023, June 09). What Is Transfer Learning? Exploring the Popular Deep Learning Approach. Built In. Available online: https://builtin.com/data-science/transfer-learning.
17. Chaturvedi, S.S., Gupta, K., and Prasad, P.S. (2021). Skin Lesion Analyser: An Efficient Seven-Way Multi-Class Skin Cancer Classification Using MobileNet. arXiv.10.1007/978-981-15-3383-9_15 ↗
18. Deep neural networks are superior to dermatologists in melanoma image classification10.1016/j.ejca.2019.05.023 ↗
19. Dermatologist-level classification of skin cancer with deep neural networks10.1038/nature21056 ↗
20. Milton, M.A.A. (2019). Automated Skin Lesion Classification Using Ensemble of Deep Neural Networks in ISIC 2018: Skin Lesion Analysis Towards Melanoma Detection Challenge. arXiv.
21. Skin cancer identification system of HAM10000 skin cancer dataset using convolutional neural network10.1063/1.5141652 ↗
22. Agrahari, P., Agrawal, A., and Subhashini, N. (2022). Skin Cancer Detection Using Deep Learning. Futuristic Communication and Network Technologies, Springer.10.1007/978-981-16-4625-6_18 ↗
23. Automated melanoma recognition in dermoscopy images via very deep residual networks10.1109/TMI.2016.2642839 ↗
24. Codella, N., Cai, J., Abedini, M., Garnavi, R., Halpern, A., and Smith, J.R. (2015, January 5–9). Deep learning, sparse coding, and SVM for melanoma recognition in Dermoscopy Images. Proceedings of the 6th International Workshop on Machine Learning in Medical Imaging, Munish, Germany.10.1007/978-3-319-24888-2_15 ↗
25. A review on protection against fileless malware attacks using gateway10.17762/turcomat.v12i10.5620 ↗
26. Uçkuner, M., and Erol, H. (2021, January 15–17). A New Deep Learning Model for Skin Cancer Classification. Proceedings of the 2021 6th International Conference on Computer Science and Engineering (UBMK), Ankara, Turkey.10.1109/UBMK52708.2021.9558936 ↗
27. Filali, Y., El Khoukhi, H., Sabri, M.A., and Aarab, A. (2022, January 18–20). Analysis and classification of skin cancer based on deep learning approach. Proceedings of the 2022 International Conference on Intelligent Systems and Computer Vision (ISCV), Fez, Morocco.10.1109/ISCV54655.2022.9806087 ↗
28. Fanconic (2023, February 10). Skin Cancer—Malignant vs. Benign. Kaggle. Available online: https://www.kaggle.com/datasets/fanconic/skin-cancer-malignant-vs-benign.
29. Dhankhar, N. (2024, August 18). ISIC 2020 JPG 224X224 Resized, Kaggle. Available online: https://www.kaggle.com/datasets/nischaydnk/isic-2020-jpg-224x224-resized/data.
30. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep Residual Learning for Image Recognition. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.90 ↗
31. Xin, C., Liu, Z., Zhao, K., Miao, L., Ma, Y., Zhu, X., Zhou, Q., Wang, S., Li, L., and Yang, F. (2022). An improved transformer network for skin cancer classification. Comput. Biol. Med., 1.10.1016/j.compbiomed.2022.105939 ↗
32. Research on SVM and KNN classifiers for Skin cancer detection10.35940/ijeat.B5117.129219 ↗
33. Huang, G., Liu, Z., van der Maaten, L., and Weinberger, K.Q. (2017, January 21–26). Densely connected convolutional networks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Honolulu, HI, USA.10.1109/CVPR.2017.243 ↗
34. Szegedy, C., Vanhoucke, V., Ioffe, S., Shlens, J., and Wojna, Z. (2016, January 27–30). Rethinking the inception architecture for computer vision. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.10.1109/CVPR.2016.308 ↗
35. Hassanien, A., Bhatnagar, R., and Darwish, A. (2020). Advanced Machine Learning Technologies and Applications. Advances in Intelligent Systems and Computing, Springer.10.1007/978-981-15-3383-9 ↗
36. Graph regularization methods in soft detector fusion10.1109/ACCESS.2023.3344776 ↗
37. Multiclass alpha integration of scores from multiple classifiers10.1162/neco_a_01169 ↗