Skin Cancer Classification Application Using Machine Learning

Main Article Content

Palash Chaudhari


Melanoma is one of the predominant types of skin cancer. The affected number has been increasing year after year. Although the deaths can be minimized by early detection and there is where the problem exists and consulting a dermatologist may not always guarantee the success of early detection and diagnoses. At first, the dermatologist examines the skin visually and decides whether it’s a type of skin cancer or a skin allergy. The accuracy of the diagnosis directly corresponds to the experience of the dermatologist. Even a small error in the inspection of the skin might end a life of a person so it is really necessary to have a standard and supporting system which can help dermatologists to identify and diagnose the patients is necessary. So with the advancements in image processing and deep learning algorithms have unleashed the potential to classify and identify the type of skin cancer with a single click of an image. The traditional method involves a lot of pre-processing steps and if something goes wrong in that step the model doesn’t perform well. The accuracy won’t be up to the mark this is where the Convolutional Neural Networks come into the picture. These models don’t require any feature extraction or with some minimal pre-processing steps to be done and it consumes a huge amount of data to be well trained. In this paper, we will compare the transfer learning with end-to-end trained custom deep learning models. It classifies the images into seven different classes. The model is deployed on the web locally which will be handy for the dermatologist to use it as a User Interface for assisting with the identification. The model with the transfer learning shows good results than the one which is trained from scratch. The plots show the difference between them and the way in which they train.

Article Details

How to Cite
P. Chaudhari, “Skin Cancer Classification Application Using Machine Learning”, Int. J. Data. Science., vol. 2, no. 1, pp. 47-55, Sep. 2021.


Codella N, Rotemberg V, Tschandl P, Celebi ME, Dusza S, Gutman D, et al. Skin lesion analysis toward melanoma detection: A challenge at the 2017 International Symposium on Biomedical Imaging (ISBI), hosted by the International Skin Imaging Collaboration (ISIC). 2018 Presented at: IEEE 15th International Symposium on Biomedical Imaging (ISBI 2018); April 4; Washington, DC, USA.

XU Meifeng, GUO Leida, SONG Panpan, et al. Skin Disease Recognition Method Based on Multi-Model Fusion of Convolutional Neural Network[J]. Journal of Xi'an Jiaotong University, 2019, 53(11):125-130.

Noel Codella, Veronica Rotemberg, Philipp Tschandl, M. Emre Celebi, Stephen Dusza, David Gutman, Brian Helba, Aadi Kalloo, Konstantinos Liopyris, Michael Marchetti, Harald Kittler, Allan Halpern: “Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)”, 2018;

Tschandl, P., Rosendahl, C. & Kittler, H. The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions. Sci. Data 5, 180161 DOI:10.1038/sdata.2018.161 (2018).

Codella N, Cai J, Abedini M, Garnavi R, Halpern A, Smith JR. Deep Learning, Sparse Coding, and SVM for Melanoma Recognition in Dermoscopy Images. In: Machine Learning In Medical Imaging. Cham: Springer; 2015:118-126

S.Mabrouk M, A.Sheha M, Sharawy A. Automatic Detection of Melanoma Skin Cancer using Texture Analysis. IJCA 2012 Mar 31;42(20):22-26.

Hosny KM, Kassem MA, Foaud MM. Skin Cancer Classification using Deep Learning and Transfer Learning. 2018 Presented at: 9th Cairo International Biomedical Engineering Conference (CIBEC); December 20-22; Cairo, Egypt p. 90-93.

Esteva A, Kuprel B, Novoa RA, Ko J, Swetter SM, Blau HM, et al. Dermatologist-level classification of skin cancer with deep neural networks. Nature 2017 Jan 25;542(7639):115-118.

Navarrete-Dechent C, Dusza SW, Liopyris K, Marghoob AA, Halpern AC, Marchetti MA. Automated dermatological diagnosis: Hype or reality? J Invest Dermatol 2018 Oct; 138(10):2277-2279.

Vogado, L. H., Veras, R. M., Araujo, F. H., Silva, R. R., & Aires, K. R. (2018). “Leukemia diagnosis in blood slides using transfer learning in CNNs and SVM for classification”. Engineering Applications of Artificial Intelligence, 72, 415-42

Krizhevsky A, Sutskever I, Hinton GE. Imagenet classification with deep convolutional neural networks. Communications of the ACM 2017;60(6):84-90.

Dwivedi, A. K. (2018). “Artificial neural network model for effective cancer classification using microarray gene expression data”. Neural Computing and Applications, 29(12), 1545-1554.

Fan, H., Zhang, F., Xi, L., Li, Z., Liu, G., & Xu, Y. (2019). “LeukocyteMask: An automated localization and segmentation method for leukocyte in blood smear images using deep neural networks”. Journal of biophotonics, 12(7), e201800488.

Gao, Z., Wang, L., Zhou, L., & Zhang, J. (2016). “HEp-2 cell image classification with deep convolutional neural networks”. IEEE Journal of biomedical and health informatics, 21(2), 416-428.

He, K., Gkioxari, G., Dollar, P., & Girshick, R. (2017). “Mask R-CNN”. In Proceedings of the IEEE international conference on computer vision (pp.2961-2969).

Jiang, H., & Learned-Miller, E. (2017, May). “Face detection with the faster R-CNN”. In 2017 12th IEEE International Conference on Automatic Face & Gesture Recognition (FG 2017) (pp. 650-657). IEEE.

Kazemi, F., Najafabadi, T. A., & Araabi, B. N. (2016). “Automatic recognition of acute myelogenous leukemia in blood microscopic images using k-means clustering and support vector machine”. Journal of medical signals and sensors, 6(3), 183.

Kusunose, K., Abe, T., Haga, A., Fukuda, D., Yamada, H., Harada, M., & Sata, M. (2020). “A deep learning approach for assessment of regional wall motion abnormality from echocardiographic images”. JACC: Cardiovascular Imaging, 13(2), 374-381.

Krizhevsky A, Sutskever I, Hinton GE. ImageNet classification with deep convolutional neural networks. Adv Neural Inf Process Syst 2012 Jan:1-23.

Peng, Y., Wang, N., Wang, Y., & Wang, M. (2019). “Segmentation of dermoscopy image using adversarial networks”. Multimedia Tools and Applications, 78(8), 10965-10981.

Ren, S., He, K., Girshick, R., & Sun, J. (2015). “Faster R-CNN: Towards real-time object detection with region proposal networks”. In Advances in neural information processing systems (pp. 91-99).

Sun, X., Wu, P., & Hoi, S. C. (2018). “Face detection using deep learning: An improved faster RCNN approach”. Neurocomputing, 299, 42-50.

Unver, H. M., & Ayan, E. (2019). “Skin lesion segmentation in dermoscopic images with combination of YOLO and grabcut algorithm”. Diagnostics, 9(3), 7

Han SS, Kim MS, Lim W, Park GH, Park I, Chang SE. Classification of the clinical images for benign and malignant cutaneous tumors using a deep learning algorithm. J Invest Dermatol 2018 Jul;138(7):1529-1538.

Dorj U, Lee K, Choi J, Lee M. The skin cancer classification using deep convolutional neural network. Multimed Tools Appl 2018 Feb 22;77(8):9909-9924.