Skip to main navigation Skip to search Skip to main content

GSCEU-Net: An End-to-End Lightweight Skin Lesion Segmentation Model with Feature Fusion Based on U-Net Enhancements

  • Shengnan Hao
  • , Haotian Wu
  • , Yanyan Jiang
  • , Zhanlin Ji
  • , Li Zhao
  • , Linyun Liu
  • , Ivan Ganchev
  • North China University of Science and Technology
  • University of Limerick
  • Tsinghua University
  • University of Plovdiv "Paisii Hilendarski"
  • Bulgarian Academy of Sciences

Research output: Contribution to journalArticlepeer-review

Abstract

Accurate segmentation of lesions can provide strong evidence for early skin cancer diagnosis by doctors, enabling timely treatment of patients and effectively reducing cancer mortality rates. In recent years, some deep learning models have utilized complex modules to improve their performance for skin disease image segmentation. However, limited computational resources have hindered their practical application in clinical environments. To address this challenge, this paper proposes a lightweight model, named GSCEU-Net, which is able to achieve superior skin lesion segmentation performance at a lower cost. GSCEU-Net is based on the U-Net architecture with additional enhancements. Firstly, the partial convolution (PConv) module, proposed by the FasterNet model, is modified to an SConv module, which enables channel segmentation paths of different scales. Secondly, a newly designed Ghost SConv (GSC) module is proposed for incorporation into the model’s backbone, where the Separate Convolution (SConv) module is aided by a Multi-Layer Perceptron (MLP) and the output path residuals from the Ghost module. Finally, the Efficient Channel Attention (ECA) mechanism is incorporated at different levels into the decoding part of the model. The segmentation performance of the proposed model is evaluated on two public datasets (ISIC2018 and PH2) and a private dataset. Compared to U-Net, the proposed model achieves an IoU improvement of 0.0261 points and a DSC improvement of 0.0164 points, while reducing the parameter count by 190 times and the computational complexity by 170 times. Compared to other existing segmentation models, the proposed GSCEU-Net model also demonstrates superiority, along with an advanced balance between the number of parameters, complexity, and segmentation performance.

Original languageEnglish
Article number486
JournalInformation (Switzerland)
Volume14
Issue number9
DOIs
Publication statusPublished - Sept 2023

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 3 - Good Health and Well-being
    SDG 3 Good Health and Well-being

Keywords

  • attention mechanism
  • convolutional neural network
  • lightweight
  • skin lesion segmentation

Fingerprint

Dive into the research topics of 'GSCEU-Net: An End-to-End Lightweight Skin Lesion Segmentation Model with Feature Fusion Based on U-Net Enhancements'. Together they form a unique fingerprint.

Cite this