SLSNet: Skin lesion segmentation using a lightweight generative adversarial network

Abstract

The determination of precise skin lesion boundaries in dermoscopic images using automated methods faces many challenges, most importantly, the presence of hair, inconspicuous lesion edges and low contrast in dermoscopic images, and variability in the color, texture and shapes of skin lesions. Existing deep learning-based skin lesion segmentation algorithms are expensive in terms of computational time and memory. Consequently, running such segmentation algorithms requires a powerful GPU and high bandwidth memory, which are not available in dermoscopy devices. Thus, this article aims to achieve precise skin lesion segmentation with minimum resources: a lightweight, efficient generative adversarial network (GAN) model called SLSNet, which combines 1-D kernel factorized networks, position and channel attention, and multiscale aggregation mechanisms with a GAN model. The 1-D kernel factorized network reduces the computational cost of 2D filtering. The position and channel attention modules enhance the discriminative ability between the lesion and non-lesion feature representations in spatial and channel dimensions, respectively. A multiscale block is also used to aggregate the coarse-to-fine features of input skin images and reduce the effect of the artifacts. SLSNet is evaluated on two publicly available datasets: ISBI 2017 and the ISIC 2018. Although SLSNet has only 2.35 million parameters, the experimental results demonstrate that it achieves segmentation results on a par with the state-of-the-art skin lesion segmentation methods with an accuracy of 97.61%, and Dice and Jaccard similarity coefficients of 90.63% and 81.98%, respectively. SLSNet can run at more than 110 frames per second (FPS) in a single GTX1080Ti GPU, which is faster than well-known deep learning-based image segmentation models, such as FCN. Therefore, SLSNet can be used for practical dermoscopic applications.

arXiv commentAccepted in Expert Systems with Applications

References in corpus (17)

• Adam: A Method for Stochastic Optimization
• MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications
• Attention U-Net: Learning Where to Look for the Pancreas
• Self-Attention Generative Adversarial Networks
• DoReFa-Net: Training Low Bitwidth Convolutional Neural Networks with Low Bitwidth Gradients
• ENet: A Deep Neural Network Architecture for Real-Time Semantic Segmentation
• Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)
• A Survey of Model Compression and Acceleration for Deep Neural Networks
• XNOR-Net: ImageNet Classification Using Binary Convolutional Neural Networks
• Model compression via distillation and quantization
• Dual Attention Network for Scene Segmentation
• Attention-Gated Networks for Improving Ultrasound Scan Plane Detection
• SkinNet: A Deep Learning Framework for Skin Lesion Segmentation
• MeliusNet: Can Binary Neural Networks Achieve MobileNet-level Accuracy?
• Deep-Learning Ensembles for Skin-Lesion Segmentation, Analysis, Classification: RECOD Titans at ISIC Challenge 2018
• Improving Automatic Skin Lesion Segmentation using Adversarial Learning based Data Augmentation
• Unsupervised Medical Image Segmentation with Adversarial Networks: From Edge Diagrams to Segmentation Maps

Cited by in corpus (2)

• A Survey on Deep Learning for Skin Lesion Segmentation
• A survey, review, and future trends of skin lesion segmentation and classification