atlas

Research map / Label-efficient and robust learning / Recognition and long-tail

TagCLIP: A Local-to-Global Framework to Enhance Open-Vocabulary Multi-Label Classification of CLIP without Training

AAAI 2024 · Label-efficient and robust learning › Recognition and long-tail · approach: Partial and multi-label learning · 28 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 TagCLIP, a training-free local-to-global framework that turns frozen CLIP ViT-B/16 into an open-vocabulary multi-label tagger (patch-level classification, dual-masking attention refinement, class-wise reidentification) and uses its tags as pseudo labels for annotation-free segmentation (CLS-SEG).

Problem

CLIP's class token, trained with a contrastive softmax loss for global features, performs poorly on open-vocabulary multi-label classification because the global embedding is dominated by the most prominent class and misses small or inconspicuous objects.

Approach

Use patch tokens from the penultimate ViT layer (skipping the last self-attention, which destroys spatial information) for patch-level classification, refine scores with a dual-masked multi-layer attention affinity, and re-check each candidate class with a class-cropped global CLIP pass, all without training.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add patch-level classification, DMAR, and CWR to frozen CLIPCLIP ViT-B/16 · PASCAL VOC 2007 testmAP: 85.8 -> 92.8Helped
Add patch-level classification, DMAR, and CWR to frozen CLIPCLIP ViT-B/16 · MS COCO 2014 valmAP: 63.3 -> 68.8Helped
Omit the last self-attention operation (ablation)CLIP ViT-B/16 · PASCAL VOC 2012 valmIoU: 16.2 -> 41.6Helped
Add DMAR to coarse scores (ablation)TagCLIP · PASCAL VOC 2012 valmAP: 85.4 -> 93.9Helped
Add CWR to coarse scores (ablation)TagCLIP · PASCAL VOC 2012 valmAP: 85.4 -> 88.0Helped
Add CWR after DMAR (ablation)TagCLIP · PASCAL VOC 2012 valmAP: 93.9 -> 94.1Helped
Fuse multiple attention layers rather than using a single layer for refinement (ablation)DMAR · Classification and segmentation benchmarksclassification and segmentation performanceHelped
Use dual masking in attention refinement (ablation)DMAR · Classification and segmentation benchmarksclassification and segmentation performanceHelped
Fuse local and global classification scores (ablation)TagCLIP CWR · PASCAL VOC 2007 test and MS COCO 2014 valmAPHelped
Add a background set (ablation)TagCLIP · PASCAL VOC 2007 testmAP: 92.4 -> 92.8Helped
Add a background set (ablation)TagCLIP · MS COCO 2014 valmAP: 67.9 -> 68.8Helped
Remove softmax from classification scores (ablation)CLIP classification · PASCAL VOC 2007 test and MS COCO 2014 valmAP: 85.8 -> 79.5; 63.3 -> 54.2Hurt

Surprising findings

Limitations

Experimental setup

Datasets: PASCAL VOC 2012 MS COCO 2014 COCO-Stuff

Benchmarks: PASCAL VOC 2007 MS COCO 2014 PASCAL VOC 2012 MS COCO 2017 COCO-Stuff

Metrics: mAP mIoU

Baselines: CLIP DPT CLIPSurgery SARB DualCoOp TAI-DPT MaskCLIP GroupViT SegCLIP ReCo NamedMask

Builds on: CLIP CLIP-ES CLIP-Surgery GradCAM

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