LSSF-Net: Lightweight Segmentation with Self-Awareness, Spatial Attention, and Focal Modulation

TL;DR

LSSF-Net is a lightweight, efficient skin lesion segmentation model designed for mobile devices, combining advanced attention mechanisms and split channel-shuffle to improve accuracy on challenging dermoscopic images.

Contribution

The paper introduces a novel lightweight segmentation network with only 0.8 million parameters, integrating conformer-based focal modulation and self-aware spatial attention for improved performance.

Findings

Achieves state-of-the-art results on four benchmark datasets.

Demonstrates high Jaccard index indicating accurate segmentation.

Operates effectively with minimal parameters on mobile platforms.

Abstract

Accurate segmentation of skin lesions within dermoscopic images plays a crucial role in the timely identification of skin cancer for computer-aided diagnosis on mobile platforms. However, varying shapes of the lesions, lack of defined edges, and the presence of obstructions such as hair strands and marker colors make this challenge more complex. \textcolor{red}Additionally, skin lesions often exhibit subtle variations in texture and color that are difficult to differentiate from surrounding healthy skin, necessitating models that can capture both fine-grained details and broader contextual information. Currently, melanoma segmentation models are commonly based on fully connected networks and U-Nets. However, these models often struggle with capturing the complex and varied characteristics of skin lesions, such as the presence of indistinct boundaries and diverse lesion appearances, which can lead to suboptimal segmentation performance.To address these challenges, we propose a novel lightweight network specifically designed for skin lesion segmentation utilizing mobile devices, featuring a minimal number of learnable parameters (only 0.8 million). This network comprises an encoder-decoder architecture that incorporates conformer-based focal modulation attention, self-aware local and global spatial attention, and split channel-shuffle. The efficacy of our model has been evaluated on four well-established benchmark datasets for skin lesion segmentation: ISIC 2016, ISIC 2017, ISIC 2018, and PH2. Empirical findings substantiate its state-of-the-art performance, notably reflected in a high Jaccard index.