Skin Cancer Diagnosis Using VGG16 and Transfer Learning: Analyzing the Effects of Data Quality over Quantity on Model Efficiency
Source abstract
Abstract
The recent increase in the prevalence of skin cancer, along with its significant impact on individuals’ lives, has garnered the attention of many researchers in the field of deep learning models, especially following the promising results observed using these models in the medical field. This study aimed to develop a system that can accurately diagnose one of three types of skin cancer: basal cell carcinoma (BCC), melanoma (MEL), and nevi (NV). Additionally, it emphasizes the importance of image quality, as many studies focus on the quantity of images used in deep learning. In this study, transfer learning was employed using the pre-trained VGG-16 model alongside a dataset sourced from Kaggle. Three models were trained while maintaining the same hyperparameters and script to ensure a fair comparison. However, the quantity of data used to train each model was varied to observe specific effects and to hypothesize about the importance of image quality in deep learning models within the medical field. The model with the highest validation score was selected for further testing using a separate test dataset, which the model had not seen before, to evaluate the model’s performance accurately. This work contributes to the existing body of research by demonstrating the critical role of image quality in enhancing diagnostic accuracy, providing a comprehensive evaluation of the VGG-16 model’s performance in skin cancer detection and offering insights that can guide future improvements in the field.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: AI in cancer detection · Cutaneous Melanoma Detection and Management · Digital Imaging for Blood Diseases
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Computer Science Department, University of Batna 2, Batna 05078, Algeria
Computer Science Department, IUT of Colmar University of Haute Alsace, 68008 Colmar, France
References (47)
Show source-provided references
1. Incidence estimate of nonmelanoma skin cancer (keratinocyte carcinomas) in the US population, 201210.1001/jamadermatol.2015.1187 ↗
2. American Cancer Society (2024, January 17). Cancer Facts and Figures 2024. Available online: https://www.cancer.org/research/cancer-facts-statistics/all-cancer-facts-figures/2024-cancer-facts-figures.html.
3. The treatment of actinic keratoses—The rule rather than the exception
4. EANO guidelines on the diagnosis and treatment of diffuse gliomas of adulthood10.1038/s41571-020-00447-z ↗
5. Agrawal, A., and Bhatt, J. (2023). Diagnostic Error. Patient Safety, Springer.10.1007/978-3-031-35933-0 ↗
6. Ahsan, M.M., Luna, S.A., and Siddique, Z. (2022). Machine-Learning-Based Disease Diagnosis: A Comprehensive Review. Healthcare, 10.10.3390/healthcare10030541 ↗
7. Galić, I., Habijan, M., Leventić, H., and Romić, K. (2023). Machine Learning Empowering Personalized Medicine: A Comprehensive Review of Medical Image Analysis Methods. Electronics, 12.10.3390/electronics12214411 ↗
8. Deep learning-aided decision support for diagnosis of skin disease across skin tones10.1038/s41591-023-02728-3 ↗
9. Dwivedi, A., Agarwal, N., Ray, L., and Tripathi, A. (2019). Understanding Cellular and Molecular Events of Skin Aging and Cancer: An Integrative Perspective. Skin Aging & Cancer, Springer.10.1007/978-981-13-2541-0 ↗
10. Swathi, B., Kannan, K., Chakravarthi, S.S., Ruthvik, G., Avanija, J., and Reddy, C.C.M. (2023, January 6–8). Skin Cancer Detection using VGG16, InceptionV3 and ResUNet. Proceedings of the 2023 4th International Conference on Electronics and Sustainable Communication Systems (ICESC), Coimbatore, India.10.1109/ICESC57686.2023.10193609 ↗
11. Automatic Classification of Melanoma Skin Cancer with Deep Convolutional Neural Networks10.3390/ai3020029 ↗
12. A systematic review of worldwide incidence of nonmelanoma skin cancer10.1111/j.1365-2133.2012.10830.x ↗
13. Incidence of basal cell and squamous cell carcinomas in a population younger than 40 years10.1001/jama.294.6.681 ↗
14. Melanoma recognition in Dermoscopy images via aggregated deep convolutional features10.1109/TBME.2018.2866166 ↗
15. Malignant melanoma10.1007/s00105-019-04514-0 ↗
16. Meta-analysis of risk factors for cutaneous melanoma: I. Common and atypical naevi10.1016/j.ejca.2004.10.015 ↗
17. Deep Learning Applications in Medical Image Analysis10.1109/ACCESS.2017.2788044 ↗
18. Guest Editorial Deep Learning in Medical Imaging: Overview and Future Promise of an Exciting New Technique10.1109/TMI.2016.2553401 ↗
19. Ghanem, N.M., Attallah, O., Anwar, F., and Ismail, M.A. (2022). Artificial Intelligence in Cancer Diagnosis and Prognosis, Volume 2: Breast and Bladder Cancer, IOP Publishing.
20. Krizhevsky, A., Sutskever, I., and Hinton, G.E. (2012). Imagenet classification with deep convolutional neural networks. Advances in Neural Information Processing Systems, Available online: https://dl.acm.org/doi/10.1145/3065386.
21. Driver inattention monitoring system for intelligent vehicles: A review10.1109/TITS.2010.2092770 ↗
22. Stacked denoising autoencoders: Learning useful representations in a deep network with a local denoising criterion
23. Goodfellow, I., Pouget-Abadie, J., Mirza, M., Xu, B., Warde-Farley, D., Ozair, S., and Bengio, Y. (2014). Generative adversarial nets. Advances in Neural Information Processing Systems, Available online: https://dl.acm.org/doi/10.1145/3422622.
24. Kingma, D.P., and Welling, M. (2013). Auto-encoding variational bayes. arXiv.
25. Dinh, L., Sohl-Dickstein, J., and Bengio, S. (2016). Density estimation using real nvp. arXiv.
26. Lipton, Z.C., Berkowitz, J., and Elkan, C. (2015). A critical review of recurrent neural networks for sequence learning. arXiv.
27. Vaswani, A., Shazeer, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, Ł., and Polosukhin, I. (2017). Attention is all you need. Advances in Neural Information Processing Systems, Available online: https://arxiv.org/abs/1706.03762.
28. Nodular melanoma: Patients’ perceptions of presenting features and implications for earlier detection10.1067/mjd.2003.216 ↗
29. Deep learning10.1038/nature14539 ↗
30. Liu, S., Liu, S., Cai, W., Pujol, S., Kikinis, R., and Feng, D. (May, January 29). Early diagnosis of Alzheimer’s disease with deep learning. Proceedings of the International Symposium on Biomedical Imaging, Beijing, China.
31. Manifold learning of brain MRIs by deep learning
32. Gradient-based learning applied to document recognition10.1109/5.726791 ↗
33. A learning algorithm for continually running fully recurrent neural networks10.1162/neco.1989.1.2.270 ↗
34. Smolensky, P. (1986). Information Processing in Dynamical Systems: Foundations of Harmony Theory, Colorado University at Boulder, Department of Computer Science.
35. Reducing the dimensionality of data with neural networks10.1126/science.1127647 ↗
36. A Survey of the Recent Architectures of Deep Convolutional Neural Networks10.1007/s10462-020-09825-6 ↗
37. Nazari, S., and Garcia, R. (2023). Automatic Skin Cancer Detection Using Clinical Images: A Comprehensive Review. Life, 13.10.3390/life13112123 ↗
38. Dildar, M., Akram, S., Irfan, M., Khan, H.U., Ramzan, M., Mahmood, A.R., Alsaiari, S.A., Saeed, A.H.M., Alraddadi, M.O., and Mahnashi, M.H. (2021). Skin Cancer Detection: A Review Using Deep Learning Techniques. Int. J. Environ. Res. Public Health, 18.10.3390/ijerph18105479 ↗
39. Naqvi, M., Gilani, S.Q., Syed, T., Marques, O., and Kim, H.-C. (2023). Skin Cancer Detection Using Deep Learning—A Review. Diagnostics, 13.10.3390/diagnostics13111911 ↗
40. Abadi, M. (2016, January 18–24). TensorFlow: Learning functions at scale. Proceedings of the 21st ACM SIGPLAN International Conference on Functional Programming, Nara, Japan.10.1145/2951913.2976746 ↗
41. Dermatologist-level classification of skin cancer with deep neural networks10.1038/nature21056 ↗
42. Simonyan, K., and Zisserman, A. (2015, January 7–9). Very Deep Convolutional Networks for Large-Scale Image Recognition. Proceedings of the International Conference on Learning Representations (ICLR), San Diego, CA, USA.
43. Madinakhon, R., Mukhtorov, D., and Cho, Y.-I. (2024). Integrating Principal Component Analysis and Multi-Input Convolutional Neural Networks for Advanced Skin Lesion Cancer Classification. Appl. Sci., 14.10.3390/app14125233 ↗
44. Saini, A., Guleria, K., and Sharma, S. (2023, January 3–4). Skin Cancer Classification Using Transfer Learning-Based Pre-Trained VGG 16 Model. Proceedings of the 2023 IEEE International Conference on Computer Communication and Information Systems (ICCCIS), Greater Noida, India.10.1109/ICCCIS60361.2023.10425341 ↗
45. A Visually Interpretable Deep Learning Framework for Histopathological Image-Based Skin Cancer Diagnosis10.1109/JBHI.2021.3052044 ↗
46. Khamsa, D., Pascal, L., Zakaria, B., Lokman, M., and Zakaria, M.Y. (2023, January 5–7). Skin Cancer Diagnosis and Detection Using Deep Learning. Proceedings of the 2023 International Conference on Electrical Engineering and Advanced Technology (ICEEAT), Batna, Algeria.10.1109/ICEEAT60471.2023.10426611 ↗
47. The HAM10000 dataset: A large collection of multi-source dermatoscopic images of common pigmented skin lesions