Automatic Skin Lesion Segmentation and Melanoma Detection: Transfer Learning Approach with U-Net and DCNN-SVM

Abstract

Industrial pollution resulting in ozone layer depletion has influenced increased UV radiation in recent years which is a major environmental risk factor for invasive skin cancer, melanoma, and other keratinocyte cancers. The incidence of deaths from melanoma has risen worldwide in the past two decades. Deep learning has been employed successfully for dermatologic diagnosis. In this work, we present a deep learning-based scheme to automatically segment skin lesions and detect melanoma from dermoscopy images. U-Net was used for segmenting out the lesion from surrounding skin. The limitation of utilizing deep neural networks with limited medical data was solved with data augmentation and transfer learning. In our experiments, U-Net was used with spatial dropout to solve the problem of overfitting, and different augmentation effects were applied to the training images to increase data samples. The model was evaluated on two different datasets. It achieved a mean dice score of 0.87 and a mean Jaccard index of 0.80 on ISIC 2018 dataset. The trained model was assessed on PH2 dataset where it achieved a mean dice score of 0.93 and a mean Jaccard index of 0.87 with transfer learning. For classification of malignant melanoma, a DCNN-SVM model was used where we compared state-of-the-art deep nets as feature extractors to find the applicability of transfer learning in dermatologic diagnosis domain. Our best model achieved a mean accuracy of 92% on PH2 dataset. The findings of this study are expected to be useful in cancer diagnosis research.