Multiclass skin lesion classification using image augmentation technique and transfer learning models

PurposeThe mortality rate due to skin cancers has been increasing over the past decades. Early detection and treatment of skin cancers can save lives. However, due to visual resemblance of normal skin and lesion and blurred lesion borders, skin cancer diagnosis has become a challenging task even for skilled dermatologists. Hence, the purpose of this study is to present an image-based automatic approach for multiclass skin lesion classification and compare the performance of various models.Design/methodology/approachIn this paper, the authors have presented a multiclass skin lesion classification approach based on transfer learning of deep convolutional neural network. The following pre-trained models have been used: VGG16, VGG19, ResNet50, ResNet101, ResNet152, Xception, MobileNet and compared their performances on skin cancer classification.FindingsThe experiments have been performed on HAM10000 dataset, which contains 10,015 dermoscopic images of seven skin lesion classes. The categorical accuracy of 83.69%, Top2 accuracy of 91.48% and Top3 accuracy of 96.19% has been obtained.Originality/valueEarly detection and treatment of skin cancer can save millions of lives. This work demonstrates that the transfer learning can be an effective way to classify skin cancer images, providing adequate performance with less computational complexity.

[1]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[2]  R. Lew,et al.  Prevention and early detection strategies for melanoma and skin cancer. Current status. , 1996, Archives of dermatology.

[3]  Tim Holland-Letz,et al.  Superior skin cancer classification by the combination of human and artificial intelligence. , 2019, European journal of cancer.

[4]  Harald Kittler,et al.  Descriptor : The HAM 10000 dataset , a large collection of multi-source dermatoscopic images of common pigmented skin lesions , 2018 .

[5]  P. Sasieni,et al.  13. Cancers attributable to solar (ultraviolet) radiation exposure in the UK in 2010 , 2011, British Journal of Cancer.

[6]  Saket S. Chaturvedi,et al.  Skin Lesion Analyser: An Efficient Seven-Way Multi-Class Skin Cancer Classification Using MobileNet , 2019, Advances in Intelligent Systems and Computing.

[7]  Vimal K. Shrivastava,et al.  Segmentation and Border Detection of Melanoma Lesions Using Convolutional Neural Network and SVM , 2018, Computational Intelligence: Theories, Applications and Future Directions - Volume I.

[8]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[9]  A. Enk,et al.  Deep learning outperformed 11 pathologists in the classification of histopathological melanoma images. , 2019, European journal of cancer.

[10]  A. Enk,et al.  Systematic outperformance of 112 dermatologists in multiclass skin cancer image classification by convolutional neural networks. , 2019, European journal of cancer.

[11]  Vimal K. Shrivastava,et al.  Transfer Learning-Based Framework for Classification of Pest in Tomato Plants , 2020, Appl. Artif. Intell..

[12]  Muhammad Sharif,et al.  Attributes based skin lesion detection and recognition: A mask RCNN and transfer learning-based deep learning framework , 2021, Pattern Recognit. Lett..

[13]  Peyman Hosseinzadeh Kassani,et al.  A comparative study of deep learning architectures on melanoma detection. , 2019, Tissue & cell.

[14]  François Chollet,et al.  Xception: Deep Learning with Depthwise Separable Convolutions , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).