Deep Learning and Transfer Learning for Skin Cancer Segmentation and Classification

Abstract

According to Skin Cancer Foundation, skin cancer is by far the most common type of cancer in the United States and worldwide. Early diagnosis of skin cancer is critical because proper treatment at early stages can increase the chance of cure and recovery. However, visual inspection of dermoscopic images by dermatologists is error-prone and time-consuming. To ensure accurate diagnosis and faster treatment of skin cancer, deep learning techniques have been utilized to conduct automated skin lesion segmentation and classification. In this paper, after image processing, a Mask R-CNN model is built for lesion segmentation, where transfer learning is utilized by using the pre-trained weights from Microsoft COCO dataset. The weights of the trained Mask R-CNN model are saved and transferred to the next task - skin lesion classification, to train a Mask R-CNN model for classification. Our experiments are conducted on the benchmark datasets from the International Skin Imaging Collaboration 2018 (ISIC 2018) and evaluated by the same metrics used in ISIC 2018. The lesion boundary segmentation and lesion classification have achieved an accuracy of 96% and a balanced multiclass accuracy of 80%, respectively.