HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

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

Shichao Xu; Yikang Li; Jenhao Hsiao; Chiuman Ho; Zhu Qi

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

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

BenchmarkMethodologyMetrics
multi-label-classification-on-ms-cocoADDS(ViT-L-336, resolution 640)
mAP: 93.41
multi-label-classification-on-ms-cocoADDS(ViT-L-336, resolution 1344)
mAP: 93.54
multi-label-classification-on-ms-cocoADDS(ViT-L-336, resolution 336)
mAP: 91.76
multi-label-zero-shot-learning-on-imagenet-1kADDS
mAP: 67.10
multi-label-zero-shot-learning-on-nus-wideADDS (ViT-L-336, resolution 336)
mAP: 39.01
multi-label-zero-shot-learning-on-nus-wideADDS (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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp