Attention-guided deep convolutional neural networks for skin cancer classification

Skin cancer is a silently killing disease which commonly goes unnoticed in its primitive stage but proves to be deadly later on. Hence, it needs to be detected and classified in the early stages itself. The advent of machine learning as well as deep learning based classification techniques has made this task possible. Deep convolutional neural networks (D-CNNs) have the ability to extract universal and dataset-specific features for the image classification task. But the classification of skin cancer images remains a challenging task due to the absence of balanced class images, difference between images of the same class, similarity between inter-class images and the inefficiency in focusing on the semantically significant areas of the image. To improve the performance of these D-CNNs, we incorporate the attention mechanism that focuses on the regions of importance in an image. In that regard, we propose an attention-guided D-CNN for classification of skin cancer. It is observed from the classification results that a model with attention boosts the accuracy of a normal D-CNN architecture by approximately 12%. Our research work contributes significantly to the field of biomedical image processing by providing a mechanism to improve performance of D-CNNs and facilitating early detection of skin cancer.

[1]  Malrey Lee,et al.  The skin cancer classification using deep convolutional neural network , 2018, Multimedia Tools and Applications.

[2]  D. S. Guru,et al.  Segmentation and Classification of Skin Lesions for Disease Diagnosis , 2016, ArXiv.

[3]  Ilias Maglogiannis,et al.  Overview of Advanced Computer Vision Systems for Skin Lesions Characterization , 2009, IEEE Transactions on Information Technology in Biomedicine.

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

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

[6]  Yoshua Bengio,et al.  Show, Attend and Tell: Neural Image Caption Generation with Visual Attention , 2015, ICML.

[7]  Sergey Ioffe,et al.  Rethinking the Inception Architecture for Computer Vision , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[8]  Adel Al-Jumaily,et al.  Review on automatic early skin cancer detection , 2011, 2011 International Conference on Computer Science and Service System (CSSS).

[9]  Nils Gessert,et al.  Skin Lesion Classification Using CNNs With Patch-Based Attention and Diagnosis-Guided Loss Weighting , 2019, IEEE Transactions on Biomedical Engineering.

[10]  Patrick van der Smagt,et al.  A Differentiable Transition Between Additive and Multiplicative Neurons , 2016, ArXiv.

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

[12]  Chokri Ben Amar,et al.  ChaboNet : Design of a deep CNN for prediction of visual saliency in natural video , 2019, J. Vis. Commun. Image Represent..

[13]  Sebastian Thrun,et al.  Dermatologist-level classification of skin cancer with deep neural networks , 2017, Nature.

[14]  Haofu Liao,et al.  A Deep Learning Approach to Universal Skin Disease Classification , 2015 .

[15]  Bareqa Salah,et al.  Skin Cancer Recognition by Using a Neuro-Fuzzy System , 2011, Cancer informatics.

[16]  K. S. Ravichandran,et al.  Novel Approaches for Diagnosing Melanoma Skin Lesions Through Supervised and Deep Learning Algorithms , 2016, Journal of Medical Systems.

[17]  Mahmoud A. Elgamal,et al.  AUTOMATIC SKIN CANCER IMAGES CLASSIFICATION , 2013 .