混合CNN与Mamba的3D模型,提升脑部分割速度与精度
MedSegMamba: 3D CNN-Mamba Hybrid Architecture for Brain Segmentation
- 用Mamba的选通扫描机制处理3D MRI数据,提升效率
- 在1784张脑MRI上测试,Dice系数达0.884,优于多数现有模型
- 参数量少20%却更准,适合医疗影像高效分析场景
传统皮层下脑区分割流程效率低,尤其面对高分辨率MRI和多类别结构时。本文提出一种基于3D局部块的CNN-Mamba混合模型,利用Mamba的选通扫描算法,提升3D输入的分割准确率与效率。研究使用来自多中心、健康人群的1784例T1加权MRI扫描(1.5T与3T设备采集),按1076/345/363比例划分训练、验证与测试集。以FreeSurfer生成的金标准为基准,采用Dice相似系数(DSC)、体积相似性(VS)与平均对称表面距离(ASSD)评估性能。统计显著性通过威尔科克斯符号秩检验(P < 0.05)判断。该模型在所有指标上均表现最优(DSC 0.88383;VS 0.97076;ASSD 0.33604),显著优于非Mamba基线模型(P < 0.001)。虽与另一Mamba模型相比在DSC与VS上无显著差异(P=0.114, 0.425),但在ASSD上显著更优(P < 0.001),且参数减少约20%。结果表明,该混合架构在3D脑区分割中兼具高效与精准,具有实际应用潜力。代码已开源:https://github.com/aaroncao06/MedSegMamba。
原文摘要 · Abstract (English)
Widely used traditional pipelines for subcortical brain segmentation are often inefficient and slow, particularly when processing large datasets. Furthermore, deep learning models face challenges due to the high resolution of MRI images and the large number of anatomical classes involved. To address these limitations, we developed a 3D patch-based hybrid CNN-Mamba model that leverages Mamba's selective scan algorithm, thereby enhancing segmentation accuracy and efficiency for 3D inputs. This retrospective study utilized 1784 T1-weighted MRI scans from a diverse, multi-site dataset of healthy individuals. The dataset was divided into training, validation, and testing sets with a 1076/345/363 split. The scans were obtained from 1.5T and 3T MRI machines. Our model's performance was validated against several benchmarks, including other CNN-Mamba, CNN-Transformer, and pure CNN networks, using FreeSurfer-generated ground truths. We employed the Dice Similarity Coefficient (DSC), Volume Similarity (VS), and Average Symmetric Surface Distance (ASSD) as evaluation metrics. Statistical significance was determined using the Wilcoxon signed-rank test with a threshold of P < 0.05. The proposed model achieved the highest overall performance across all metrics (DSC 0.88383; VS 0.97076; ASSD 0.33604), significantly outperforming all non-Mamba-based models (P < 0.001). While the model did not show significant improvement in DSC or VS compared to another Mamba-based model (P-values of 0.114 and 0.425), it demonstrated a significant enhancement in ASSD (P < 0.001) with approximately 20% fewer parameters. In conclusion, our proposed hybrid CNN-Mamba architecture offers an efficient and accurate approach for 3D subcortical brain segmentation, demonstrating potential advantages over existing methods. Code is available at: https://github.com/aaroncao06/MedSegMamba.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。