arXiv:2502.02024eess.IVcs.CV2025-02被引 9

用不确定性驱动扫描,提升医学图像分割精度

UD-Mamba: A pixel-level uncertainty-driven Mamba model for medical image segmentation

  • 根据像素不确定性动态调整扫描顺序,优化局部特征捕捉
  • 在三种医学影像数据集上均超越基线模型,边界分割更准
  • 适合需要精细分割的临床场景,如病灶检测与器官建模

近期研究表明,Mamba 框架作为状态空间模型,在以线性复杂度捕捉长程依赖方面表现优异。尽管其在医学图像分割中已展现竞争力,但传统基于位置的扫描方式难以有效建模局部特征,且医学图像常存在复杂模糊的边界。为此,我们提出不确定性驱动的 Mamba(UD-Mamba),通过将通道不确定性融入扫描机制,重新设计像素级扫描流程。该方法引入两种关键扫描策略:1)顺序扫描,按行逐行扫描,优先处理高不确定性区域;2)跳过扫描,垂直列方向以固定间隔从高到低或从低到高扫描不确定性区域。顺序扫描能有效聚类边界和前景等高不确定性区域,提升分割精度;跳过扫描增强背景与前景间的交互,及时融合背景信息以支持更准确的前景推断。为平衡不同扫描方式提取特征的重要性,引入四个可学习参数。同时,采用余弦一致性损失缓解扫描过程中不确定与确定区域切换带来的负面影响。实验验证了该方法在病理、皮肤病变和心脏任务三个医学影像数据集上的鲁棒性能。

原文摘要 · Abstract (English)

Recent advancements have highlighted the Mamba framework, a state-space model known for its efficiency in capturing long-range dependencies with linear computational complexity. While Mamba has shown competitive performance in medical image segmentation, it encounters difficulties in modeling local features due to the sporadic nature of traditional location-based scanning methods and the complex, ambiguous boundaries often present in medical images. To overcome these challenges, we propose Uncertainty-Driven Mamba (UD-Mamba), which redefines the pixel-order scanning process by incorporating channel uncertainty into the scanning mechanism. UD-Mamba introduces two key scanning techniques: 1) sequential scanning, which prioritizes regions with high uncertainty by scanning in a row-by-row fashion, and 2) skip scanning, which processes columns vertically, moving from high-to-low or low-to-high uncertainty at fixed intervals. Sequential scanning efficiently clusters high-uncertainty regions, such as boundaries and foreground objects, to improve segmentation precision, while skip scanning enhances the interaction between background and foreground regions, allowing for timely integration of background information to support more accurate foreground inference. Recognizing the advantages of scanning from certain to uncertain areas, we introduce four learnable parameters to balance the importance of features extracted from different scanning methods. Additionally, a cosine consistency loss is employed to mitigate the drawbacks of transitioning between uncertain and certain regions during the scanning process. Our method demonstrates robust segmentation performance, validated across three distinct medical imaging datasets involving pathology, dermatological lesions, and cardiac tasks.

医学图像分割Mamba不确定性建模

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