DualCoOp: Fast Adaptation to Multi-Label Recognition with Limited Annotations
About
Solving multi-label recognition (MLR) for images in the low-label regime is a challenging task with many real-world applications. Recent work learns an alignment between textual and visual spaces to compensate for insufficient image labels, but loses accuracy because of the limited amount of available MLR annotations. In this work, we utilize the strong alignment of textual and visual features pretrained with millions of auxiliary image-text pairs and propose Dual Context Optimization (DualCoOp) as a unified framework for partial-label MLR and zero-shot MLR. DualCoOp encodes positive and negative contexts with class names as part of the linguistic input (i.e. prompts). Since DualCoOp only introduces a very light learnable overhead upon the pretrained vision-language framework, it can quickly adapt to multi-label recognition tasks that have limited annotations and even unseen classes. Experiments on standard multi-label recognition benchmarks across two challenging low-label settings demonstrate the advantages of our approach over state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Multi-Label Classification | NUS-WIDE (test) | mAP43.6 | 112 | |
| Multi-label recognition | MS-COCO | -- | 66 | |
| Multi-Label Classification | VOC 07 | mAP93.2 | 61 | |
| Multi-Label Classification | NUS-WIDE 925/81 (unseen) | mAP (Mean Average Precision)43.6 | 43 | |
| Multi-Label Classification | VOC-MLT (test) | Total mAP81.03 | 34 | |
| Multi-label recognition | PASCAL VOC 2007 (test) | Avg. mAP93.2 | 25 | |
| Multi-label Image Classification | PASCAL VOC 2007 | mAP90.3 | 25 | |
| Long-Tailed Multi-Label Visual Recognition | COCO Long-Tailed (test) | mAP Total53.11 | 21 | |
| Multi-label image recognition | PASCAL VOC 2007 | mAP @ 10% Threshold91.4 | 18 | |
| Multi-Label Classification | MS COCO 48 seen / 17 unseen classes v1 | Precision58.4 | 18 |