提出高效鲁棒的脑肿瘤分割模型,显著提升精度与速度。
DRBD-Mamba for Robust and Efficient Brain Tumor Segmentation with Analytical Insights
- 设计双分辨率双向Mamba结构,减少多轴计算开销。
- 在BraTS2023上实现肿瘤核心提升1.75%、增强部分提升1.68%的精度。
- 15倍效率提升,适合临床部署与复杂数据场景。
精准脑肿瘤分割对临床诊疗至关重要,但受肿瘤异质性影响仍具挑战。基于状态空间模型的Mamba虽具计算效率优势,但在该任务中因跨多空间轴顺序计算仍存在显著开销。此外,其在不同BraTS数据集上的鲁棒性尚未充分评估。为此,本文提出双分辨率双向Mamba(DRBD-Mamba),一种高效3D分割模型,通过空间填充曲线将3D特征映射为1D,保留空间局部性,降低多轴扫描开销。引入门控融合模块自适应整合正反向上下文,并加入量化模块提升鲁棒性。构建五组系统性分层测试集用于严格评估,分析常见失败模式。在近期方法使用的20%测试集上,模型实现全肿瘤Dice提升0.10%,肿瘤核心提升1.75%,增强肿瘤提升0.93%。在系统性分层评估中,平均取得肿瘤核心1.16%、增强肿瘤1.68%的提升。同时,相较现有方法实现15倍效率提升,兼具高精度与强鲁棒性。
原文摘要 · Abstract (English)
Accurate brain tumor segmentation is significant for clinical diagnosis and treatment but remains challenging due to tumor heterogeneity. Mamba-based State Space Models have demonstrated promising performance. However, despite their computational efficiency over other neural architectures, they incur considerable overhead for this task due to their sequential feature computation across multiple spatial axes. Moreover, their robustness across diverse BraTS data partitions remains largely unexplored, leaving a critical gap in reliable evaluation. To address this, we first propose a dual-resolution bi-directional Mamba (DRBD-Mamba), an efficient 3D segmentation model that captures multi-scale long-range dependencies with minimal computational overhead. We leverage a space-filling curve to preserve spatial locality during 3D-to-1D feature mapping, thereby reducing reliance on computationally expensive multi-axial feature scans. To enrich feature representation, we propose a gated fusion module that adaptively integrates forward and reverse contexts, along with a quantization block that improves robustness. We further propose five systematic folds on BraTS2023 for rigorous evaluation of segmentation techniques under diverse conditions and present analysis of common failure scenarios. On the 20% test set used by recent methods, our model achieves Dice improvements of 0.10% for whole tumor, 1.75% for tumor core, and 0.93% for enhancing tumor. Evaluations on the proposed systematic folds demonstrate that our model maintains competitive whole tumor accuracy while achieving clear average Dice gains of 1.16% for tumor core and 1.68% for enhancing tumor over existing state-of-the-art. Furthermore, our model achieves a 15x efficiency improvement while maintaining high segmentation accuracy, highlighting its robustness and computational advantage over existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。