atlas

Research map / Detection and segmentation / Open-set and few-shot detection

Relationship Prompt Learning is Enough for Open-Vocabulary Semantic Segmentation

NeurIPS 2024 · Detection and segmentation › Open-set and few-shot detection · approach: Open-vocabulary semantic segmentation · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes Relationship Prompt Network (RPN), which uses a Relationship Prompt Module to make a frozen CLIP ViT-B/16 produce pixel-level OVSS predictions directly via prompt learning, with only about 3M trainable parameters and no segmentation-specific networks.

Problem

Open-vocabulary semantic segmentation typically needs extra segmentation-specific networks (mask proposal networks or semantic decoders) on top of VLMs, which makes training costly; prompt learning on VLMs has been limited to image-level granularity and lacks image-text relationships.

Approach

A Relationship Prompt Module generates pixel-level relationship prompts from image-text attention (image-to-pixel attention scheme) inserted layer-by-layer into the CLIP ViT encoder, combined with a two-linear-layer projection head, so the VLM itself outputs segmentation without segmentation-specific networks.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use RPN instead of the strongest listed baseline without self-trainingCLIP ViT-B/16 · VOC zero-shotmIoUu: 78.6 -> 84.6Helped
use RPN instead of the strongest listed baseline without self-trainingCLIP ViT-B/16 · COCO zero-shotmIoUu: 42.2 -> 42.8Helped
use RPN instead of the strongest listed baseline without self-trainingCLIP ViT-B/16 · Context zero-shotmIoUu: 54.6 -> 58.7Helped
add RPM to the CLIP with LPM baseline (ablation)CLIP with LPM · VOChIoU: 59.6 -> 87.9Helped
add LPM to RPM (ablation)RPN · VOChIoU: 86.3 -> 88.6Helped
use both image-level and pixel-level attention in ITP (ablation)RPM · VOChIoU: 33.2 -> 88.6Helped
use trainable nn.Parameter instead of nn.Linear in APG (ablation)RPM · VOChIoU: 86.0 -> 88.6Helped
increase M2oE dimension from r = 3 to r = 12 (ablation)RPM · VOChIoU: 88.6 -> 88.8Helped
use M2oE instead of direct Multi-Scale aggregation (ablation)RPM · VOChIoU: 88.2 -> 88.6Helped
use LPMc with Sigmoid loss instead of LPMc with Softmax loss (ablation)RPN · VOChIoU: 71.4 -> 88.6Helped
use LPMc instead of LPMa with Sigmoid loss (ablation)RPN · VOChIoU: 71.2 -> 88.6Helped
include the [CLS] token in the training-free projection module (TFPMc instead of TFPMa) (ablation)RPN without trainable LPM layers · VOChIoU: 68.9 -> 86.7Helped

Surprising findings

Limitations

Experimental setup

Datasets: ADE20K Pascal VOC 2012 COCO-Stuff164K Pascal Context

Benchmarks: Pascal VOC 2012 COCO-Stuff164K Pascal Context ADE20K A-847 PC-459 A-150 PC-59 PAS-20

Metrics: pAcc mIoU seen mIoU unseen hIoU

Baselines: ZegFormer ZegCLIP FreeSeg ZSSeg MaskCLIP OVSeg CAT-Seg SAN SED FC-CLIP BitFit Adapter

Builds on: CLIP VPT LoRA ZegCLIP MaskCLIP

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy