MHMamba提升3D脑肿瘤分割精度,兼顾速度与边界清晰度
MHMamba: Multi-Head Mamba for 3D Brain Tumor Segmentation

- 用多头状态空间模型替代CNN和Transformer,建模长距离依赖
- 在BraTS2021/2023上实现更高准确率、更平滑边界和小病灶敏感度
- 适合需要高精度、快速分割的医学影像研究者使用
脑肿瘤形态异质性强且多模态对比明显,手动逐切片勾画耗时且依赖经验,亟需高效稳定的自动化分割方法。为克服CNN建模长程依赖能力弱,以及3D MRI中Transformer计算量大、内存开销高且块间上下文不一致的问题,本文提出多头状态空间模型MHMamba。该方法结合U型结构与多头状态空间模型(Mamba),将通道维度拆分为并行的SSM头,通过残差聚合增强长程表征,并保持线性复杂度。为对齐统计特性并强化病灶响应,设计通道-空间校准模块处理多头输出,并在跳跃连接引入自适应融合机制,动态连接全局语义与局部细节,提升边界一致性与小体积病灶检测能力。在BraTS2021和BraTS2023数据集上的实验与消融表明,MHMamba在整体精度、边界平滑性和肿瘤核心及小体积增强区域敏感度上均取得稳定显著提升,同时保留Mamba的线性复杂度优势,验证了方法的有效性与通用性。
原文摘要 · Abstract (English)
Brain tumors exhibit high heterogeneity in morphology and multimodal contrast, making manual slice-by-slice de lineation time-consuming and experience-dependent, thus necessitating efficient and stable automated segmentation methods. To address the limitations of CNNs in modeling long-range dependencies, and the heavy computational and memory overhead and inter-block contextual in coherence of Transformers in 3D MRI, this paper proposes Multi-Head Mamba (MHMamba). This method combines a U-shaped architecture with a multi-head state-space model (Mamba), splitting the channel dimension into parallel SSM heads and aggregating them with residuals. This enhances long-range representation and improves the stability of multimodal training while maintaining linear complexity. To further align statistics and enhance lesion response, we designed a channel-space calibration module for multi-head outputs and introduced an adaptive fusion mechanism at skip connections to dynamically connect global semantics with local details, thereby improving boundary consistency and the detection of small-volume lesions. We conducted experiments and ablations on BraTS2021 and BraTS2023. The results showed that MHMamba achieved stable and significant improvements in overall accuracy, boundary smoothness, and sensitivity to tumor core and small-volume enhancement areas, while preserving the linear-complexity advantage of Mamba-based modeling, thus verifying the effectiveness and versatility of the method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。