Abstract:Few-shot segmentation (FSS) aims to segment the novel class with a few annotated images. Due to CLIP's advantages of aligning visual and textual information, the integration of CLIP can enhance the generalization ability of FSS model. However, even with the CLIP model, the existing CLIP-based FSS methods are still subject to the biased prediction towards base class, which is caused by the class-specific feature level interactions. To solve this issue, we propose a visual and textual Prior Guided Mask Assemble Network (PGMA-Net). It employs a class-agnostic mask assembly process to alleviate the bias, and formulates diverse tasks into a unified manner by assembling the prior through affinity. Specifically, the class-relevant textual and visual features are first transformed to class-agnostic prior in the form of probability map. Then, a Prior-Guided Mask Assemble Module (PGMAM) including multiple General Assemble Units (GAUs) is introduced. It considers diverse and plug-and-play interactions, such as visual-textual, inter- and intra-image, training-free, and high-order ones. Lastly, to ensure the class-agnostic ability, a Hierarchical Decoder with Channel-Drop Mechanism (HDCDM) is proposed to flexibly exploit the assembled masks and low-level features, without relying on any class-specific information. It achieves new state-of-the-art results in the FSS task, with mIoU of 77.6 on $\rm{PASCAL-}5^{i}$ and 59.4 on $\rm{COCO-}20^{i}$ in 1-shot scenario. Beyond this, we show that without extra re-training, the proposed PGMA-Net can solve bbox-level and cross-domain FSS, co-segmentation, zero-shot segmentation (ZSS) tasks, leading an any-shot segmentation framework capable of accommodating diverse weak or pixel annotations.

Context-FPN and Memory Contrastive Learning for Partially Supervised Instance Segmentation.

When Masked Image Modeling Meets Source-free Unsupervised Domain Adaptation: Dual-Level Masked Network for Semantic Segmentation

ContrastMask: Contrastive Learning to Segment Every Thing

Rethinking mask heads for partially supervised instance segmentation

Visual and Textual Prior Guided Mask Assemble for Few-Shot Segmentation and Beyond

Few-Shot Semantic Segmentation with Cyclic Memory Network

SegMind: Semisupervised Remote Sensing Image Semantic Segmentation With Masked Image Modeling and Contrastive Learning Method

Multi-Level Contrastive Learning for Dense Prediction Task

MFCP-Net: Multiscale Feature Correlation Perception Network for Image Segmentation Based on Background Prior

Contextual-wise Discriminative Feature Extraction and Robust Network Learning for Subcortical Structure Segmentation

Multi-stage Context Refinement Network for Semantic Segmentation

Region-aware Contrastive Learning for Semantic Segmentation

Commonality-Parsing Network Across Shape and Appearance for Partially Supervised Instance Segmentation

Multi-Task Learning Via SA-FPN and EJ-Head

DMF-CL: Dense Multi-scale Feature Contrastive Learning for Semantic Segmentation of Remote-Sensing Images

Early Hierarchical Contexts Learned by Convolutional Networks for Image Segmentation

Saliency Based Multiple Object Cosegmentation by Ensemble MIML Learning

Augmented FCN: Rethinking Context Modeling for Semantic Segmentation

HCNet: Hierarchical Context Network for Semantic Segmentation

CRF-Based Figure/Ground Segmentation with Pixel-Level Sparse Coding and Neighborhood Interactions.

Masked Supervised Learning for Semantic Segmentation