LCM-UNet: Scale-Selective Mamba with Reparameterizable Multi-Basis Local Modeling for Lightweight Skin Lesion Segmentation
Organizations: NIMISHES Lab, Dhaka, Bangladesh · Stamford University Bangladesh, Dhaka, Bangladesh · Elite Research Lab, New York, NY, USA
Abstract
Accurate lesion segmentation underpins automated dermoscopic analysis: masks enable measurements of asymmetry, border and size. Deployment in primary care or on mobile devices requires a compact network. Existing ultra-light Mamba U-Nets offer global context, but global-modeling cost depends on each Mamba block's resolution, while their local branches rely on a single square kernel. We introduce LCM-UNet, a 33,474-parameter U-shaped network that fuses every skip at its native resolution, restricts bidirectional Mamba to stages at one-sixteenth of the input resolution or coarser, and uses local convolutions at finer scales. A multi-basis reparameterizable adapter augments each local branch during training with dilated, horizontal and vertical depthwise kernels, then folds them into the host kernel before inference. Each skip connection also learns a stage-specific gain. Nine networks were retrained on the official ISIC 2017 and ISIC 2018 partitions under a harmonized protocol, with PH reserved for external validation and ablations run on a development split. LCM-UNet achieved Dice scores of 88.62%, 89.73% and 90.46% on ISIC 2017, ISIC 2018 and PH, respectively, ranking first on both ISIC test sets and second on PH among the compared models. In the native-resolution host, removing Mamba from the one-eighth-resolution decoder stage eliminated 63% of Mamba operations, corresponding to 5.3% of all operations, with a Dice change of percentage points. The adapter improved Dice by 0.48 percentage points over a single host kernel without adding deployed parameters or operations. With latency so far measured only on a GPU, LCM-UNet is a candidate for resource-constrained deployment.
Figures & tables
| Dataset | Role | Train | Val. | Test | Total |
|---|---|---|---|---|---|
| ISIC 2017 [ 36 ] | Benchmark | 2,000 | 150 | 600 | 2,750 |
| ISIC 2018 [ 37 , 38 ] | Benchmark | 2,594 | 100 | 1,000 | 3,694 |
| ISIC 2018 development | Ablation | 2,075 | 100 | 519 | – |
| PH 2 [ 39 ] | External validation | – | – | 200 | 200 |
| ISIC 2017 (600 test images) | ISIC 2018 (1,000 test images) | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | Year | Params (M) | GMACs | DSC (%) | IoU (%) | SE (%) | SP (%) | DSC (%) | IoU (%) | SE (%) | SP (%) |
| U-Net [ 4 ] | 2015 | 17.263 | 39.980 | 86.34 0.41 | 75.96 0.63 | 85.12 | 95.85 | 88.21 0.35 | 78.91 0.56 | 87.93 | 96.37 |
| UNeXt [ 13 ] | 2022 | 1.472 | 0.562 | 86.92 0.33 | 76.87 0.52 | 86.05 | 95.89 | 88.54 0.29 | 79.44 0.47 | 88.36 | 96.43 |
| MALUNet [ 8 ] | 2022 | 0.178 | 0.083 | 87.35 0.36 | 77.54 0.57 | 86.41 | 96.06 | 88.73 0.31 | 79.74 0.50 | 88.18 | 96.64 |
| EGE-UNet [ 9 ] | 2023 | 0.053 | 0.072 | 87.52 0.31 | 77.81 0.49 | 86.93 | 95.97 | 88.86 0.26 | 79.95 0.42 | 88.91 | 96.44 |
| VM-UNet [ 7 ] | 2025 | 27.428 | 5.049 | 87.81 0.29 | 78.27 0.46 | 88.47 | 95.52 | 89.37 0.22 | 80.78 0.36 | 89.64 | 96.52 |
| Method | DSC (%) | IoU (%) | SE (%) | SP (%) |
|---|---|---|---|---|
| U-Net [ 4 ] | 88.93 0.52 | 80.07 0.84 | 91.84 | 95.07 |
| UNeXt [ 13 ] | 89.47 0.47 | 80.95 0.77 | 92.26 | 95.32 |
| MALUNet [ 8 ] | 89.62 0.49 | 81.19 0.80 | 91.95 | 95.56 |
| EGE-UNet [ 9 ] | 89.81 0.41 | 81.50 0.68 | 92.40 | 95.52 |
| VM-UNet [ 7 ] | 90.71 0.33 | 83.00 0.55 | 93.12 | 95.91 |
| UltraLight VM-UNet [ 10 ] | 89.74 0.44 | 81.39 0.72 | 92.18 | 95.56 |
| (a) Component ablation | ||||||||
|---|---|---|---|---|---|---|---|---|
| ID | Adapter | D2 Mamba | Params (M) | GMACs | MMACs | DSC (%) | IoU (%) | |
| A 1 | shared | none | ✓ | 0.0342 | 0.062 | 4.4 | 88.15 0.29 | 78.81 0.46 |
| A0 | shared | none | ✓ | 0.0347 | 0.208 | 18.9 | 88.35 0.24 | 79.13 0.39 |
| A1 | stage | none | ✓ | 0.0347 | 0.208 | 18.9 | 88.44 0.22 | 79.28 0.35 |
| A2 | stage | RDLA | ✓ | 0.0347 | 0.208 | 18.9 | 88.61 0.20 | 79.55 0.32 |
| A3 | stage | RDLA | ✗ | 0.0335 | 0.197 | 6.9 | 88.58 0.23 | 79.50 0.37 |
| Model | Params | GMACs | ms | FPS | DSC | IoU | |
|---|---|---|---|---|---|---|---|
| Host (A0) | 34,720 | 0.208 | 7.21 | 138.7 | – | – | – |
| LCM-UNet (unfused) | 34,014 | 0.197 | 6.69 | 149.5 | 89.73 | 81.37 | – |
| LCM-UNet (deployed) | 33,474 | 0.197 | 6.43 | 155.5 | 89.73 | 81.37 |