MUCM-Net: A Mamba Powered UCM-Net for Skin Lesion Segmentation
arXiv:2405.15925 · doi:10.37349/emed.2024.00250
Abstract
Skin lesion segmentation is key for early skin cancer detection. Challenges in automatic segmentation from dermoscopic images include variations in color, texture, and artifacts of indistinct lesion boundaries. Deep learning methods like CNNs and U-Net have shown promise in addressing these issues. To further aid early diagnosis, especially on mobile devices with limited computing power, we present MUCM-Net. This efficient model combines Mamba State-Space Models with our UCM-Net architecture for improved feature learning and segmentation. MUCM-Net's Mamba-UCM Layer is optimized for mobile deployment, offering high accuracy with low computational needs. Tested on ISIC datasets, it outperforms other methods in accuracy and computational efficiency, making it a scalable tool for early detection in settings with limited resources. Our MUCM-Net source code is available for research and collaboration, supporting advances in mobile health diagnostics and the fight against skin cancer. In order to facilitate accessibility and further research in the field, the MUCM-Net source code is https://github.com/chunyuyuan/MUCM-Net
11 pages, 8 figures, journal paper is accepted by Exploration of Medicine
References in corpus (11)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Decoupled Weight Decay Regularization
- Attention U-Net: Learning Where to Look for the Pancreas
- TransUNet: Transformers Make Strong Encoders for Medical Image Segmentation
- The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions
- Mamba: Linear-Time Sequence Modeling with Selective State Spaces
- Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model
- The Computational Limits of Deep Learning
- VM-UNet: Vision Mamba UNet for Medical Image Segmentation
- A comprehensive review of Binary Neural Network
- State Space Model for New-Generation Network Alternative to Transformers: A Survey