RUFNet通过优化支持图和不确定性融合,提升少样本脑肿瘤分割精度。
RUFNet: Query-Guided Support Mask Refinement and Uncertainty Fusion based on Hybrid Mamba for Few-Shot Brain Tumor Segmentation

- 用查询特征引导支持图精炼,改善原型一致性
- 1样本设置下Dice达84.3%,5样本时达86.1%,优于现有方法
- 适合医疗图像少样本分割任务,尤其关注预测可靠性
少样本脑肿瘤分割因支持图噪声、患者间差异大以及缺乏像素级置信估计而困难。本文提出RUFNet,一种基于混合Mamba的少样本框架,结合支持图精炼与不确定性感知后验融合。为在低计算成本下保持支持-查询依赖,RUFNet采用线性复杂度的混合Mamba交互骨干网络。为降低支持图噪声,引入注意力引导的掩码精炼模块(AGMR),利用查询特征重校支持图。为处理模糊预测,设计不确定性感知后验融合模块(UAPF),估计像素级方差并自适应平衡少样本预测与查询对齐先验。在BraTS 2020数据集上,RUFNet在1类1样本和1类5样本设置下分别取得84.3%和86.1%的Dice系数,优于对比的最先进方法。结果表明,混合Mamba交互、掩码精炼与不确定性建模可显著提升少样本医学图像分割的鲁棒性。官方代码已开源:https://github.com/hdy6438/RUFNet。
原文摘要 · Abstract (English)
Few-shot brain tumor segmentation remains challenging due to noisy support masks, inter-patient variations between support and query images, and the lack of pixel-wise confidence estimation. This study proposes RUFNet, a Hybrid Mamba-based few-shot framework that combines support mask refinement with uncertainty-aware posterior fusion. To preserve support-query dependencies with manageable cost, RUFNet adopts a Hybrid Mamba interaction backbone with linear complexity. To reduce support-mask noise, an Attention-Guided Mask Refinement module (AGMR) uses query features to recalibrate support masks and improve prototype consistency. To handle ambiguous predictions, an Uncertainty-Aware Posterior Fusion module (UAPF) estimates pixel-wise variance and adaptively balances few-shot predictions with query-aligned priors. On the Brain Tumor Segmentation Challenge (BraTS) 2020 dataset, RUFNet achieves Dice coefficients of 84.3% and 86.1% in the 1-way 1-shot and 1-way 5-shot settings, respectively, outperforming the compared state-of-the-art methods. These results suggest that Hybrid Mamba interaction, mask refinement and uncertainty modelling can improve the robustness of few-shot medical image segmentation. The official implementation code is available at https://github.com/hdy6438/RUFNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。