SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidance

Shun Zou, Mingya Zhang, Bingjian Fan, Zhengyi Zhou, Xiuguo Zou
9/17/2024
eess.IVcs.CV

Official code: zs1314/SkinMamba (54 stars, Apache-2.0, last commit 2024-09-21).

Abstract

Skin lesion segmentation is a crucial method for identifying early skin cancer. In recent years, both convolutional neural network (CNN) and Transformer-based methods have been widely applied. Moreover, combining CNN and Transformer effectively integrates global and local relationships, but remains limited by the quadratic complexity of Transformer. To address this, we propose a hybrid architecture based on Mamba and CNN, called SkinMamba. It maintains linear complexity while offering powerful long-range dependency modeling and local feature extraction capabilities. Specifically, we introduce the Scale Residual State Space Block (SRSSB), which captures global contextual relationships and cross-scale information exchange at a macro level, enabling expert communication in a global state. This effectively addresses challenges in skin lesion segmentation related to varying lesion sizes and inconspicuous target areas. Additionally, to mitigate boundary blurring and information loss during model downsampling, we introduce the Frequency Boundary Guided Module (FBGM), providing sufficient boundary priors to guide precise boundary segmentation, while also using the retained information to assist the decoder in the decoding process. Finally, we conducted comparative and ablation experiments on two public lesion segmentation datasets (ISIC2017 and ISIC2018), and the results demonstrate the strong competitiveness of SkinMamba in skin lesion segmentation tasks. The code is available at https://github.com/zs1314/SkinMamba.

View on arXivView PDF

Code Implementations(1)

zs1314/SkinMambaOfficial100%
Apache-2.0

【ACCVW2025 Oral】Offical Pytorch Code for "SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidance"

548C, C++Sep 14, 20242 years agoApache-2.0
computer-visiondeep-learningmambamedical-imaging

Similar Papers

Cite this paper

@article{zou2024skinmamba,
  title  = {SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidance},
  author = {Shun Zou and Mingya Zhang and Bingjian Fan and Zhengyi Zhou and Xiuguo Zou},
  year   = {2024},
  eprint = {2409.10890},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2409.10890v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2409.10890.svg)](https://codeofpaper.com/paper/2409.10890)

Discussion