arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2502.02024eess.IVcs.CV

UD-Mamba:一种用于医学图像分割的像素级不确定性驱动 Mamba 模型

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

University of California, Santa Cruz(加州大学圣克鲁兹分校);Johns Hopkins University(约翰霍普金斯大学);Stanford University(斯坦福大学);Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学);University of Adelaide(阿德莱德大学)

Weiren Zhao, Feng Wang, Yanran Wang, Yutong Xie, Qi Wu, Yuyin Zhou

AI总结:

本文提出 UD-Mamba,通过将通道不确定性引入像素扫描顺序,结合顺序扫描与跳跃扫描并辅以可学习参数和余弦一致性损失,提升医学图像分割精度。

摘要:

近期进展凸显了 Mamba 框架的优势,该状态空间模型以线性计算复杂度高效捕获长程依赖而著称。尽管 Mamba 在医学图像分割中已展现出有竞争力的性能,但由于传统基于位置的扫描方法具有零散性,且医学图像中常存在复杂、模糊的边界,它在建模局部特征方面仍面临困难。为克服这些挑战,我们提出 Uncertainty-Driven Mamba(UD-Mamba),通过将通道不确定性纳入扫描机制,重新定义了像素顺序扫描过程。UD-Mamba 引入两种关键扫描技术:1)顺序扫描,以逐行方式优先扫描高不确定性区域;2)跳跃扫描,按固定间隔垂直处理列,从高到低或从低到高不确定性移动。顺序扫描高效聚集高不确定性区域(如边界和前景目标)以提高分割精度,而跳跃扫描增强背景与前景区域之间的交互,允许及时整合背景信息以支持更准确的前景推断。认识到从确定区域扫描到不确定区域的优势,我们引入四个可学习参数来平衡不同扫描方法所提取特征的重要性。此外,采用余弦一致性损失以缓解扫描过程中不确定区域与确定区域之间转换带来的不利影响。我们的方法在涉及病理、皮肤病变和心脏任务的三个不同医学影像数据集上得到验证,展现出稳健的分割性能。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑