arXiv:2512.21584cs.CV2025-12被引 3

轻量级皮肤病变分割模型,兼顾高精度与低计算开销。

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

  • 用双向Mamba机制建模全局上下文,结合多分支局部感知。
  • 参数仅0.034M,算力0.060 GFLOPs,性能超越现有轻量模型。
  • 适合移动端部署,临床点检场景下实用性强。

皮肤病变分割在皮肤病学中对辅助临床决策至关重要。然而,现有方法在准确率、鲁棒性与资源效率之间存在权衡,普遍存在性能不足或计算复杂度高的问题。为此,我们提出UltraLBM-UNet,一种轻量级U-Net变体,融合双向Mamba-based全局建模机制与多分支局部特征感知。该架构通过高效局部特征注入与双向状态空间建模,在保持空间维度丰富上下文交互的同时,实现计算紧凑,适用于即时诊疗部署。在ISIC 2017、ISIC 2018与PH2数据集上的大量实验表明,本模型持续达到当前最优分割精度,仅需0.034M参数与0.060 GFLOPs,优于现有轻量及Mamba类模型。此外,我们引入混合知识蒸馏策略,训练出超紧凑学生模型UltraLBM-UNet-T,参数仅0.011M,算力0.019 GFLOPs,仍具竞争力。结果表明,UltraLBM-UNet适用于对精确与鲁棒分析要求高的点检场景。源代码已公开于https://github.com/LinLinLin-X/UltraLBM-UNet。

原文摘要 · Abstract (English)

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.

皮肤分割轻量模型Mamba点检部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。