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

HCMA-UNet:一种用于高效乳腺癌分割的混合CNN-Mamba UNet与轴向自注意力机制

HCMA-UNet: A Hybrid CNN-Mamba UNet with Axial Self-Attention for Efficient Breast Cancer Segmentation

The Fifth Affiliated Hospital of Wenzhou Medical University(温州医科大学附属第五医院)

Haoxuan Li, Wei song, Peiwu Qin, Xi Yuan, Zhenglin Chen

AI总结:

本研究提出混合分割网络HCMA-UNet,结合轻量级CNN主干与多视图轴向自注意力Mamba模块,并引入特征引导区域感知损失函数,在DCE-MRI乳腺癌病灶分割任务中实现了计算高效的最先进性能与良好的跨架构泛化能力。

摘要:

由于肿瘤形态的异质性和边界模糊,DCE-MRI中的乳腺癌病灶分割仍然具有挑战性。为应对这些挑战,本研究提出了一种新型混合分割网络HCMA-UNet,用于乳腺癌病灶分割。该网络由轻量级CNN主干和多视图轴向自注意力Mamba(MISM)模块组成。MISM模块集成了视觉状态空间块(VSSB)和轴向自注意力(ASA)机制,通过非对称拆分通道(ASC)策略有效减少参数,实现高效的三向特征提取。该轻量级模型以2.87M参数和126.44 GFLOPs实现了卓越性能。此外,提出了特征引导区域感知损失函数(FRLoss)以提高分割精度。在一个私有和两个公开DCE-MRI乳腺癌数据集上的大量实验表明,该方法在保持计算效率的同时实现了最先进的性能。FRLoss也展现出良好的跨架构泛化能力。源代码可在https://github.com/Haoxuanli-Thu/HCMA-UNet获取。

英文摘要:

Breast cancer lesion segmentation in DCE-MRI remains challenging due to heterogeneous tumor morphology and indistinct boundaries. To address these challenges, this study proposes a novel hybrid segmentation network, HCMA-UNet, for lesion segmentation of breast cancer. Our network consists of a lightweight CNN backbone and a Multi-view Axial Self-Attention Mamba (MISM) module. The MISM module integrates Visual State Space Block (VSSB) and Axial Self-Attention (ASA) mechanism, effectively reducing parameters through Asymmetric Split Channel (ASC) strategy to achieve efficient tri-directional feature extraction. Our lightweight model achieves superior performance with 2.87M parameters and 126.44 GFLOPs. A Feature-guided Region-aware loss function (FRLoss) is proposed to enhance segmentation accuracy. Extensive experiments on one private and two public DCE-MRI breast cancer datasets demonstrate that our approach achieves state-of-the-art performance while maintaining computational efficiency. FRLoss also exhibits good cross-architecture generalization capabilities. The source code is available at https://github.com/Haoxuanli-Thu/HCMA-UNet.

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

↑