Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

UltraLBM-UNet: Ultralight Bidirectional Mamba-based Model for Skin Lesion Segmentation

About

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.

Linxuan Fan, Juntao Jiang, Weixuan Liu, Zhucun Xue, Jiajun Lv, Jiangning Zhang, Yong Liu (2)__INSTITUTION_7__ Data Science Institute, Vanderbilt University, Nashville, USA (2) College of Control Science, Engineering, Zhejiang University, Hangzhou, China (3) School of Computer Science, Technology, East China Normal University, Shanghai, China)• 2025

Related benchmarks

TaskDatasetResultRank
Skin Lesion SegmentationISIC 2017 (test)
Dice Score88.78
100
Skin Lesion SegmentationPH2
DIC0.9185
58
Skin Lesion SegmentationISIC 2018
Dice Coefficient88.85
42
Showing 3 of 3 rows

Other info

Follow for update