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.

CLIP-Driven Prototype Network for Few-Shot Semantic Segmentation

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

Iterative Few-shot Semantic Segmentation from Image Label Text

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

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

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

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

Cycle association prototype network for few-shot semantic segmentation

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

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

Semantically Meaningful Class Prototype Learning for One-Shot Image Segmentation

Prototype Refinement Network for Few-Shot Segmentation

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

Found Missing Semantics: Supplemental Prototype Network for Few-Shot Semantic Segmentation

Self-Regularized Prototypical Network for Few-Shot Semantic Segmentation

Multiscale Attention-Based Prototypical Network for Few-Shot Semantic Segmentation.

Intermediate Prototype Network for Few-Shot Segmentation

PANet: Few-Shot Image Semantic Segmentation With Prototype Alignment

POEM: A prototype cross and emphasis network for few-shot semantic segmentation

Learning self-target knowledge for few-shot segmentation

Prototype Adaption and Projection for Few- and Zero-Shot 3D Point Cloud Semantic Segmentation