UltraLBM-UNet: Ultralight Bidirectional Mamba-based Model for Skin Lesion Segmentation
On the map: State-space sequence models (S4, Mamba) → Mamba for medical image analysis
Abstract (arXiv)
Skin lesion segmentation is a crucial step in dermatology for guiding clinical decision-making. However, existing methods for accurate, robust, and resource-efficient lesion analysis have limitations, including low performance and high computational complexity. To address these limitations, we propose UltraLBM-UNet, a lightweight U-Net variant that integrates a bidirectional Mamba-based global modeling mechanism with multi-branch local feature perception. The proposed architecture integrates efficient local feature injection with bidirectional state-space modeling, enabling richer contextual interaction across spatial dimensions while maintaining computational compactness suitable for point-of-care deployment. Extensive experiments on the ISIC 2017, ISIC 2018, and PH2 datasets demonstrate that our model consistently achieves state-of-the-art segmentation accuracy, outperforming existing lightweight and Mamba counterparts with only 0.034M parameters and 0.060 GFLOPs. In addition, we introduce a hybrid knowledge distillation strategy to train an ultra-compact student model, where the distilled variant UltraLBM-UNet-T, with only 0.011M parameters and 0.019 GFLOPs, achieves competitive segmentation performance. These results highlight the suitability of UltraLBM-UNet for point-of-care deployment, where accurate and robust lesion analyses are essential. The source code is publicly available at https://github.com/LinLinLin-X/UltraLBM-UNet.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Mamba for medical image analysis, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 29 of its 46 references that are in the map. First posted on arXiv 25 Dec 2025.
Builds on
29 of 46 references.
Closest ancestors · 6
Same direction, newest first
- 2024MambaU-Lite: A Lightweight Model based on Mamba and Integrated Channel-Spatial Attention for Skin Lesion Segmentation2 Dec
- 2024UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation29 Mar
- 2024LightM-UNet: Mamba Assists in Lightweight UNet for Medical Image Segmentation8 Mar
- 2024Mamba-UNet: UNet-Like Pure Visual Mamba for Medical Image Segmentation7 Feb
- 2024Swin-UMamba: Mamba-based UNet with ImageNet-based pretraining5 Feb
- 2024VM-UNet: Vision Mamba UNet for Medical Image Segmentation4 Feb
Other references · 23
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2021TransUNet: Transformers Make Strong Encoders for Medical Image Segmentationcited by 1,558
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2023RWKV: Reinventing RNNs for the Transformer Eracited by 827
- 2019Skin Lesion Analysis Toward Melanoma Detection 2018: A Challenge Hosted by the International Skin Imaging Collaboration (ISIC)cited by 669
- 2017Skin Lesion Analysis Toward Melanoma Detection: A Challenge at the 2017 International Symposium on Biomedical Imaging (ISBI), Hosted by the International Skin Imaging Collaboration (ISIC)cited by 549
- 2019GhostNet: More Features from Cheap Operationscited by 410
11 more
- 2022Decoupled Knowledge Distillationcited by 388
- 2022UNeXt: MLP-based Rapid Medical Image Segmentation Networkcited by 185
- 2021Mixed Transformer U-Net For Medical Image Segmentationcited by 95
- 2022MALUNet: A Multi-Attention and Light-weight UNet for Skin Lesion Segmentationcited by 59
- 2023EGE-UNet: an Efficient Group Enhanced UNet for skin lesion segmentationcited by 53
- 2023CMUNeXt: An Efficient Medical Image Segmentation Network based on Large Kernel and Skip Fusioncited by 31
- 2018The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesionscited by 26
- 2022A Survey on Deep Learning for Skin Lesion Segmentationcited by 22
- 20231M parameters are enough? A lightweight CNN-based model for medical image segmentationcited by 15
- 2025MK-UNet: Multi-kernel Lightweight CNN for Medical Image Segmentationcited by 12
- 2024LV-UNet: A Lightweight and Vanilla Model for Medical Image Segmentationcited by 6
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026UtVAA: Ultra-tiny Vision Transformer with Affix Attention for Mobile Image Classificationcited by 0
- 2026APRIL-MedSeg: A Modular Medical Image Segmentation Toolbox Embracing Modern Paradigmscited by 0
- 2026Lightweight Polyp Segmentation via a Gain-Aware Prediction-Space Recursive Controllercited by 0
- 2026Lightweight Polyp Segmentation via a Gain-Aware Prediction-Space Recursive Controller3 Jul 2026
- 2026APRIL-MedSeg: A Modular Medical Image Segmentation Toolbox Embracing Modern Paradigms29 Jun 2026
- 2026UtVAA: Ultra-tiny Vision Transformer with Affix Attention for Mobile Image Classification2 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2024MUCM-Net: A Mamba Powered UCM-Net for Skin Lesion Segmentation
- 2026MambaLiteUNet: Cross-Gated Adaptive Feature Fusion for Robust Skin Lesion Segmentation
- 2024SkinMamba: A Precision Skin Lesion Segmentation Architecture with Cross-Scale Global State Modeling and Frequency Boundary Guidance
- 2025ASP-VMUNet: Atrous Shifted Parallel Vision Mamba U-Net for Skin Lesion Segmentation
- 2024When Mamba Meets xLSTM: An Efficient and Precise Method with the xLSTM-VMUNet Model for Skin lesion Segmentation
- 2024AC-MAMBASEG: An adaptive convolution and Mamba-based architecture for enhanced skin lesion segmentation
- 2024Optimizing Universal Lesion Segmentation: State Space Model-Guided Hierarchical Networks with Feature Importance Adjustment
- 2025Confidence-Weighted Semi-Supervised Learning for Skin Lesion Segmentation Using Hybrid CNN-Transformer Networks