Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Open Vocabulary Multi-Label Classification with Dual-Modal Decoder on Aligned Visual-Textual Features

About

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).

Shichao Xu, Yikang Li, Jenhao Hsiao, Chiuman Ho, Zhu Qi• 2022

Related benchmarks

TaskDatasetResultRank
Multi-Label ClassificationNUS-WIDE (test)
mAP36.56
112
Multi-Label ClassificationNUS-WIDE
mAP42.67
38
Multi-Label ClassificationMS-COCO (test)
mAP93.54
24
Multi-label Image ClassificationMS-COCO (test)
mAP59.18
24
Multi-Label ClassificationNUS-WIDE 81 unseen labels (test)
mAP0.39
17
Single-to-multi label classificationNUS-WIDE with overlapped classes (test)
mAP31.07
3
Single-to-multi label classificationNUS-WIDE without overlapped classes (test)
mAP30.66
3
Single-to-multi label classificationMS-COCO With overlapped classes (test)
mAP67.1
3
Single-to-multi label classificationMS-COCO Without overlapped classes (test)
mAP69.6
3
Showing 9 of 9 rows

Other info

Follow for update