Command Palette
Search for a command to run...
Open Vocabulary Multi-Label Classification with Dual-Modal Decoder on Aligned Visual-Textual Features
Shichao Xu; Yikang Li; Jenhao Hsiao; Chiuman Ho; Zhu Qi

Abstract
In computer vision, multi-label recognition are important tasks with many real-world applications, but classifying previously unseen labels remains a significant challenge. In this paper, we propose a novel algorithm, Aligned Dual moDality ClaSsifier (ADDS), which includes a Dual-Modal decoder (DM-decoder) with alignment between visual and textual features, for open-vocabulary multi-label classification tasks. Then we design a simple and yet effective method called Pyramid-Forwarding to enhance the performance for inputs with high resolutions. Moreover, the Selective Language Supervision is applied to further enhance the model performance. Extensive experiments conducted on several standard benchmarks, NUS-WIDE, ImageNet-1k, ImageNet-21k, and MS-COCO, demonstrate that our approach significantly outperforms previous methods and provides state-of-the-art performance for open-vocabulary multi-label classification, conventional multi-label classification and an extreme case called single-to-multi label classification where models trained on single-label datasets (ImageNet-1k, ImageNet-21k) are tested on multi-label ones (MS-COCO and NUS-WIDE).
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| multi-label-classification-on-ms-coco | ADDS(ViT-L-336, resolution 640) | mAP: 93.41 |
| multi-label-classification-on-ms-coco | ADDS(ViT-L-336, resolution 1344) | mAP: 93.54 |
| multi-label-classification-on-ms-coco | ADDS(ViT-L-336, resolution 336) | mAP: 91.76 |
| multi-label-zero-shot-learning-on-imagenet-1k | ADDS | mAP: 67.10 |
| multi-label-zero-shot-learning-on-nus-wide | ADDS (ViT-L-336, resolution 336) | mAP: 39.01 |
| multi-label-zero-shot-learning-on-nus-wide | ADDS (ViT-B-32, resolution 224) | mAP: 36.56 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.