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

A Simple Long-Tailed Recognition Baseline via Vision-Language Model

About

The visual world naturally exhibits a long-tailed distribution of open classes, which poses great challenges to modern visual systems. Existing approaches either perform class re-balancing strategies or directly improve network modules to address the problem. However, they still train models with a finite set of predefined labels, limiting their supervision information and restricting their transferability to novel instances. Recent advances in large-scale contrastive visual-language pretraining shed light on a new pathway for visual recognition. With open-vocabulary supervisions, pretrained contrastive vision-language models learn powerful multimodal representations that are promising to handle data deficiency and unseen concepts. By calculating the semantic similarity between visual and text inputs, visual recognition is converted to a vision-language matching problem. Inspired by this, we propose BALLAD to leverage contrastive vision-language models for long-tailed recognition. We first continue pretraining the vision-language backbone through contrastive learning on a specific long-tailed target dataset. Afterward, we freeze the backbone and further employ an additional adapter layer to enhance the representations of tail classes on balanced training samples built with re-sampling strategies. Extensive experiments have been conducted on three popular long-tailed recognition benchmarks. As a result, our simple and effective approach sets the new state-of-the-art performances and outperforms competitive baselines with a large margin. Code is released at https://github.com/gaopengcuhk/BALLAD.

Teli Ma, Shijie Geng, Mengmeng Wang, Jing Shao, Jiasen Lu, Hongsheng Li, Peng Gao, Yu Qiao• 2021

Related benchmarks

TaskDatasetResultRank
Long-Tailed Image ClassificationImageNet-LT (test)
Top-1 Acc (Overall)76.5
220
Image ClassificationImageNet-LT (test)
Top-1 Acc (All)75.7
159
Image ClassificationPlaces-LT (test)
Accuracy (Medium)50.2
128
Long-tailed recognitionPlaces-LT (test)
Accuracy (Overall)49.5
71
Image ClassificationCIFAR-100-LT Imbalance Ratio 100 (test)
Accuracy77.8
62
Long-Tailed Image ClassificationPlaces-LT (test)
Accuracy49.5
61
Long-Tailed Image ClassificationCIFAR100-LT imbalance factor 100 (test)
Top-1 Acc77.8
23
Long-Tailed Image ClassificationCIFAR-100-LT λ=100 (test)
Accuracy (top-1)77.8
21
Long-tailed recognitionImageNet-LT (test)
Acc (Many)71
15
Long-tail Image ClassificationPlaces-LT
Overall Accuracy49.5
15
Showing 10 of 10 rows

Other info

Code

Follow for update