Modified U-NET Architecture for Segmentation of Skin Lesion

Sensors (Basel). 2022 Jan 24;22(3):867. doi: 10.3390/s22030867.

Abstract

Dermoscopy images can be classified more accurately if skin lesions or nodules are segmented. Because of their fuzzy borders, irregular boundaries, inter- and intra-class variances, and so on, nodule segmentation is a difficult task. For the segmentation of skin lesions from dermoscopic pictures, several algorithms have been developed. However, their accuracy lags well behind the industry standard. In this paper, a modified U-Net architecture is proposed by modifying the feature map's dimension for an accurate and automatic segmentation of dermoscopic images. Apart from this, more kernels to the feature map allowed for a more precise extraction of the nodule. We evaluated the effectiveness of the proposed model by considering several hyper parameters such as epochs, batch size, and the types of optimizers, testing it with augmentation techniques implemented to enhance the amount of photos available in the PH2 dataset. The best performance achieved by the proposed model is with an Adam optimizer using a batch size of 8 and 75 epochs.

Keywords: U-Net; convolutional neural network; deep learning; image analysis; segmentation; skin disease.

MeSH terms

  • Algorithms
  • Dermoscopy
  • Humans
  • Image Processing, Computer-Assisted
  • Melanoma*
  • Neural Networks, Computer
  • Skin Diseases*
  • Skin Neoplasms*