arXiv:2502.18232eess.IVcs.AI2025-02被引 6

RMA-Mamba提升肝病影像分割精度,兼顾长程依赖与局部细节。

A Reverse Mamba Attention Network for Pathological Liver Segmentation

  • 设计反向Mamba注意力模块,分层处理特征以平衡全局与局部信息。
  • 在肝硬化MRI数据集上达Dice 92.08%,在肿瘤CT数据集上Dice达92.9%。
  • 适用于复杂肝脏病变分割,适合医学影像算法研究者参考。

我们提出RMA-Mamba,一种基于改进的反向Mamba注意力模块(RMA)的新架构,通过分层处理流程增强视觉状态空间模型对长程依赖的捕捉能力,同时保持精细的局部特征表示。该架构融合了视觉Mamba(VMamba)的高效序列建模与RMA的针对性特征优化,在多尺度下实现更优的特征学习。双机制设计使模型能有效应对复杂的形态变化,且计算效率高。我们在新提出的肝硬化肝脏数据集(CirrMRI600+,T2加权MRI)上验证其有效性,获得Dice系数92.08%、平均IoU 87.36%、召回率92.96%。在肝脏肿瘤分割数据集(LiTS,CT扫描)上,达到Dice 92.9%和mIoU 88.99%。代码已开源:https://github.com/JunZengz/RMAMamba。

原文摘要 · Abstract (English)

We present RMA-Mamba, a novel architecture that advances the capabilities of vision state space models through a specialized reverse mamba attention module (RMA). The key innovation lies in RMA-Mamba's ability to capture long-range dependencies while maintaining precise local feature representation through its hierarchical processing pipeline. By integrating Vision Mamba (VMamba)'s efficient sequence modeling with RMA's targeted feature refinement, our architecture achieves superior feature learning across multiple scales. This dual-mechanism approach enables robust handling of complex morphological patterns while maintaining computational efficiency. We demonstrate RMA-Mamba's effectiveness in the challenging domain of pathological liver segmentation (from both CT and MRI), where traditional segmentation approaches often fail due to tissue variations. When evaluated on a newly introduced cirrhotic liver dataset (CirrMRI600+) of T2-weighted MRI scans, RMA-Mamba achieves the state-of-the-art performance with a Dice coefficient of 92.08%, mean IoU of 87.36%, and recall of 92.96%. The architecture's generalizability is further validated on the cancerous liver segmentation from CT scans (LiTS: Liver Tumor Segmentation dataset), yielding a Dice score of 92.9% and mIoU of 88.99%. Our code is available for public: https://github.com/JunZengz/RMAMamba.

医学图像分割Mamba肝病

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