Lightweight UNet with Multi-module Synergy and Dual-domain Attention for Precise Skin Lesion Segmentation

Read the full article See related articles

Listed in

This article is not in any list yet, why not save it to one of your lists.
Log in to save this article

Abstract

Skin cancer poses a significant threat to life, necessitating early detection. Skin lesion segmentation, a critical step in diagnosis, remains challenging due to variations in lesion size and edge blurring. Despite recent advancements in computational efficiency, edge detection accuracy remains a bottleneck. In this paper, we propose a lightweight UNet with multi-module synergy and dual-domain attention for precise skin lesion segmentation to address these issues. Our model combines the Swin Transformer (Swin-T) block, Multi-Axis External Weighting (MEWB), Group multi-axis Hadamard Product Attention (GHPA), and Group Aggregation Bridge (GAB) within a lightweight framework. Swin-T reduces complexity through parallel processing, MEWB incorporates frequency domain information for comprehensive feature capture, GHPA extracts pathological information from diverse perspectives, and GAB enhances multi-scale information extraction. On the ISIC2017 and ISIC2018 datasets, our model achieves mIoU and DSC scores of 81.22% and 89.64%, and 81.65% and 89.90%, respectively. These results demonstrate improved segmentation accuracy with low parameter count and computational cost, aiding physicians in diagnosis and treatment. Our modeling code is available at https://github.com/SolitudeWolf/ESMDL-UNet.git.

Article activity feed