Abstract:With the advent of large-scale datasets, significant advancements have been made in image semantic segmentation. However, the annotation of these datasets necessitates substantial human and financial resources. Therefore, the focus of research has shifted towards few-shot semantic segmentation, which leverages a small number of labeled samples to effectively segment unknown categories. The current mainstream methods are to use the meta-learning framework to achieve model generalization, and the main challenges are as follows. (1) The trained model will be biased towards the seen class, so the model will misactivate the seen class when segmenting the unseen class, which makes it difficult to achieve the idealized class agnostic effect. (2) When the sample size is limited, there exists an intra-class gap between the provided support images and the query images, significantly impacting the model’s generalization capability. To solve the above two problems, we propose a network with prototype complementarity characteristics (PCNet). Specifically, we first generate a self-support query prototype based on the query image. Through the self-distillation, the query prototype and the support prototype perform feature complementary learning, which effectively reduces the influence of the intra-class gap on the model generalization. A standard semantic segmentation model is introduced to segment the seen classes during the training process to achieve accurate irrelevant class shielding. After that, we use the rough prediction map to extract its background prototype and shield the background in the query image by the background prototype. In this way, we obtain more accurate fine-grained segmentation results. The proposed method exhibits superiority in extensive experiments conducted on the PASCAL-5i and COCO-20i datasets. We achieve new state-of-the-art results in the few-shot semantic segmentation task, with an mIoU of 71.27% and 51.71% in the 5-shot setting, respectively. Comprehensive ablation experiments and visualization studies show that the proposed method has a significant effect on small-sample semantic segmentation.

Uncertainty Guided Semi-Supervised Few-Shot Segmentation with Prototype Level Fusion

APANet: Adaptive Prototypes Alignment Network for Few-Shot Semantic Segmentation

Learning Prototype from Unlabeled Regions for Few-shot Segmentation

Prototype Queue Learning for Multi-Class Few-Shot Semantic Segmentation

Adaptive Prototype Learning and Allocation for Few-Shot Segmentation

Adaptive Self-Supporting Prototype Learning for Remote Sensing Few-Shot Semantic Segmentation

Dual-Guided Frequency Prototype Network for Few-Shot Semantic Segmentation

Adaptive FSS: A Novel Few-Shot Segmentation Framework Via Prototype Enhancement

Learning from the Target: Dual Prototype Network for Few Shot Semantic Segmentation

Prior-Enhanced Few-Shot Segmentation with Meta-Prototypes

PCNet: Leveraging Prototype Complementarity to Improve Prototype Affinity for Few-Shot Segmentation

Intermediate Prototype Network for Few-Shot Segmentation

CLIP-Driven Prototype Network for Few-Shot Semantic Segmentation

Spatial Correlation Fusion Network for Few-Shot Segmentation

Query-guided Prototype Evolution Network for Few-Shot Segmentation

Memory-guided Network with Uncertainty-based Feature Augmentation for Few-shot Semantic Segmentation

Cycle association prototype network for few-shot semantic segmentation

Prototype Refinement Network for Few-Shot Segmentation

PANet: Few-Shot Image Semantic Segmentation With Prototype Alignment

Support-Query Prototype Fusion Network for Few-shot Medical Image Segmentation

Prototype Comparison Convolutional Networks for One-Shot Segmentation