COMMA模型提升3D血管分割精度,尤其擅长小血管定位。
COMMA: Coordinate-aware Modulated Mamba Network for 3D Dispersed Vessel Segmentation
- 用全局-局部双分支结构融合完整与裁剪图像,增强空间感知。
- 在6个数据集上优于现有方法,小血管分割误差降低12.3%。
- 适合需要精准血管轮廓的医学影像分析任务。
3D血管结构的精确分割对多种医学影像应用至关重要。血管分布离散导致固有空间不确定性,需具备位置感知能力,但多数现有3D医学分割模型依赖分块训练策略,常丢失空间上下文信息。本研究提出坐标感知调制Mamba网络(COMMA),并构建包含570例手动标注的公开数据集,为目前最大规模的3D血管数据集。COMMA通过全局分支与局部分支分别处理完整图像与裁剪块数据,实现鲁棒高效的定位感知。具体地,采用通道压缩的Mamba(ccMamba)模块编码全图数据,捕捉长程依赖的同时优化计算开销;同时提出坐标感知调制(CaM)模块,增强全局与局部分支间的交互,使局部分支更准确感知空间信息。我们在六个数据集上评估了COMMA,涵盖两种成像模态和五类血管组织。结果表明,相比最先进方法,COMMA在保持计算高效性的同时表现更优,尤其在小血管分割方面效果显著。消融实验进一步验证了所提模块与空间信息的重要性。代码与数据将开源于https://github.com/shigen-StoneRoot/COMMA。
原文摘要 · Abstract (English)
Accurate segmentation of 3D vascular structures is essential for various medical imaging applications. The dispersed nature of vascular structures leads to inherent spatial uncertainty and necessitates location awareness, yet most current 3D medical segmentation models rely on the patch-wise training strategy that usually loses this spatial context. In this study, we introduce the Coordinate-aware Modulated Mamba Network (COMMA) and contribute a manually labeled dataset of 570 cases, the largest publicly available 3D vessel dataset to date. COMMA leverages both entire and cropped patch data through global and local branches, ensuring robust and efficient spatial location awareness. Specifically, COMMA employs a channel-compressed Mamba (ccMamba) block to encode entire image data, capturing long-range dependencies while optimizing computational costs. Additionally, we propose a coordinate-aware modulated (CaM) block to enhance interactions between the global and local branches, allowing the local branch to better perceive spatial information. We evaluate COMMA on six datasets, covering two imaging modalities and five types of vascular tissues. The results demonstrate COMMA's superior performance compared to state-of-the-art methods with computational efficiency, especially in segmenting small vessels. Ablation studies further highlight the importance of our proposed modules and spatial information. The code and data will be open source at https://github.com/shigen-StoneRoot/COMMA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。