Search papers, labs, and topics across Lattice.
This study introduces RUFNet, a novel few-shot framework for brain tumor segmentation that integrates support mask refinement and uncertainty-aware posterior fusion using a Hybrid Mamba backbone. By employing an Attention-Guided Mask Refinement module, RUFNet effectively reduces noise in support masks and enhances prototype consistency, while the Uncertainty-Aware Posterior Fusion module addresses ambiguous predictions through pixel-wise variance estimation. RUFNet achieves impressive Dice coefficients of 84.3% and 86.1% in 1-way 1-shot and 1-way 5-shot settings, respectively, surpassing state-of-the-art methods on the BraTS 2020 dataset.
RUFNet's innovative approach to combining mask refinement and uncertainty modeling leads to significant performance gains in few-shot brain tumor segmentation.
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.
