Prototype Refinement Network for Few-Shot Segmentation

2020-02-10 · Jinlu Liu, Yongqiang Qin

Few-shot segmentation targets to segment new classes with few annotated images provided. It is more challenging than traditional semantic segmentation tasks that segment known classes with abundant annotated images. In this paper, we propose a Prototype Refinement Network (PRNet) to attack the challenge of few-shot segmentation. It firstly learns to bidirectionally extract prototypes from both support and query images of the known classes. Furthermore, to extract representative prototypes of the new classes, we use adaptation and fusion for prototype refinement. The step of adaptation makes the model to learn new concepts which is directly implemented by retraining. Prototype fusion is firstly proposed which fuses support prototypes with query prototypes, incorporating the knowledge from both sides. It is effective in prototype refinement without importing extra learnable parameters. In this way, the prototypes become more discriminative in low-data regimes. Experiments on PASAL-$5^i$ and COCO-$20^i$ demonstrate the superiority of our method. Especially on COCO-$20^i$, PRNet significantly outperforms existing methods by a large margin of 13.1\% in 1-shot setting.

📄 PDF Abstract BibTeX arXiv:2002.03579

Code (0)

등록된 구현이 없습니다.

Tasks

Segmentation Semantic Segmentation

Uncertainty-aware Prototype Learning with Variational Inference for Few-shot Point Cloud Segmentation

2026-03-20 · Yifei Zhao, Fanyu Zhao, Yinsheng Li arxiv

Few-shot 3D semantic segmentation aims to generate accurate semantic masks for query point clouds with only a few annotated support examples. Existing prototype-based methods typically construct compact and deterministic…

Point Cloud Segmentation 3D Semantic Segmentation Point Clouds

RUFNet: Query-Guided Support Mask Refinement and Uncertainty Fusion based on Hybrid Mamba for Few-Shot Brain Tumor Segmentation

2026-07-06 · Dongyi He, Xiangkai Wang, Binbing Xu, Bin Jiang 외 arxiv

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…

Medical Image Segmentation Brain Tumor Segmentation

Symmetrical Joint Learning Support-query Prototypes for Few-shot Segmentation

2024-07-27 · Qun Li, Baoquan Sun, Fu Xiao, Yonggang Qi 외

We propose Sym-Net, a novel framework for Few-Shot Segmentation (FSS) that addresses the critical issue of intra-class variation by jointly learning both query and support prototypes in a symmetrical manner. Unlike previ…

Triplet

Uncertainty-Aware Semi-Supervised Few Shot Segmentation

2021-10-18 · Soopil Kim, Philip Chikontwe, Sang Hyun Park

Few shot segmentation (FSS) aims to learn pixel-level classification of a target object in a query image using only a few annotated support samples. This is challenging as it requires modeling appearance variations of ta…

Pseudo Label Segmentation

Attention-Based Prototype Calibration for Multi-Rater Few-Shot Medical Image Segmentation

2026-06-15 · Truong Vu, Minh Khoi Ho, Yutong Xie arxiv

Few-shot medical image segmentation methods typically assume a single ground-truth annotation, overlooking systematic variability across expert raters commonly observed in clinical datasets. We propose an attention-based…

Medical Image Segmentation Personalized Segmentation