轻量级MRI脑肿瘤分割模型,适配低资源医疗环境。
MMRINet: Efficient Mamba-Based Segmentation with Dual-Path Refinement for Low-Resource MRI Analysis
- 用Mamba替代注意力机制,实现高效长程上下文建模
- 双路径细化+渐进特征融合,提升边界精度与特征多样性
- 仅250万参数,在非洲数据集上达0.752 Dice,适合低算力场景
多参数MRI中自动化脑肿瘤分割在资源匮乏的临床环境中仍具挑战性,尤其在撒哈拉以南非洲(SSA),低场磁共振、患者群体异质性及严重数据稀缺使标准深度学习方法难以应用。本文提出MMRINet,一种专为该场景设计的轻量级分割架构。其核心采用线性复杂度的Mamba状态空间模型替代二次复杂度的自注意力机制,实现无高计算开销的长程体积上下文建模。结合两种轻量化优化组件:双路径特征细化(DPFR)提取互补细节与上下文表示,增强有限数据下的特征多样性;渐进特征聚合(PFA)分层融合多尺度解码器输出,提升分割边界清晰度。在包含尼日利亚临床数据的BraTS-Lighthouse SSA 2025挑战数据集上,MMRINet平均Dice达0.752,平均HD95为12.23 mm,参数量仅约250万,优于包括UNETR、Swin-UNETR、SegMamba和SegResNet3D在内的所有基线模型。结果表明,极低计算成本下仍可实现强验证集性能,为低资源环境下神经肿瘤学人工智能应用提供了可行路径。代码已开源:BioMedIA-MBZUAI/MMRINet。
原文摘要 · Abstract (English)
Automated brain tumor segmentation in multi-parametric MRI remains a critical yet underserved challenge in resource-constrained clinical settings, where deep 3D networks requiring high-end GPUs are not viable. This is particularly acute across sub-Saharan Africa (SSA), where low-field scanners, heterogeneous patient demographics, and severe data scarcity compound the difficulty of applying standard deep learning pipelines. We present MMRINet, a lightweight segmentation architecture purpose-built for these constraints. At its core, MMRINet replaces quadratic-complexity self-attention with linear-complexity Mamba state-space models, enabling efficient long-range volumetric context modeling without the computational overhead of Transformer-based approaches. We combine two lightweight refinement components:Dual-Path Feature Refinement (DPFR), which extracts complementary detail and contextual representations to improve feature diversity under limited data, and Progressive Feature Aggregation (PFA), which hierarchically fuses multi-scale decoder outputs for sharper segmentation boundaries. Evaluated on the BraTS-Lighthouse SSA 2025 challenge dataset, comprising 3D MRI scans from Nigerian clinical sites, MMRINet achieves an average Dice score of 0.752 and an average HD95 of 12.23 mm with only ~2.5M parameters, outperforming all evaluated baselines, including UNETR, Swin-UNETR, SegMamba, and SegResNet3D. These results indicate that strong validation-set segmentation performance can be achieved with substantially reduced computation, offering a practical step toward AI-assisted neuro-oncology in low-resource clinical environments. Our GitHub repository can be accessed here: BioMedIA-MBZUAI/MMRINet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。