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.

Prototype Comparison Convolutional Networks for One-Shot Segmentation

Dynamic Prototype Convolution Network for Few-Shot Semantic Segmentation

Semantically Meaningful Class Prototype Learning for One-Shot Image Segmentation

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

Multi-Similarity Enhancement Network for Few-Shot Segmentation.

Contrastive enhancement using latent prototype for few-shot segmentation

Semantically Meaningful Class Prototype Learning for One-Shot Image Semantic Segmentation

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

Intermediate Prototype Network for Few-Shot Segmentation

TPSN: Transformer-based Multi-Prototype Search Network for Few-Shot Semantic Segmentation

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

Prior-Enhanced Few-Shot Segmentation with Meta-Prototypes

Cycle association prototype network for few-shot semantic segmentation

Prototype Refinement Network for Few-Shot Segmentation

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

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

Adaptive Similarity-Guided Self-Merging Network for Few-Shot Semantic Segmentation

Pixel Matching Network for Cross-Domain Few-Shot Segmentation

SCNet: Enhancing Few-Shot Semantic Segmentation by Self-Contrastive Background Prototypes

Few-Shot Segmentation via Complementary Prototype Learning and Cascaded Refinement.

Compositional Prototype Network with Multi-view Comparision for Few-Shot Point Cloud Semantic Segmentation