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

Decoupling Representation and Classifier for Long-Tailed Recognition

About

The long-tail distribution of the visual world poses great challenges for deep learning based classification models on how to handle the class imbalance problem. Existing solutions usually involve class-balancing strategies, e.g., by loss re-weighting, data re-sampling, or transfer learning from head- to tail-classes, but most of them adhere to the scheme of jointly learning representations and classifiers. In this work, we decouple the learning procedure into representation learning and classification, and systematically explore how different balancing strategies affect them for long-tailed recognition. The findings are surprising: (1) data imbalance might not be an issue in learning high-quality representations; (2) with representations learned with the simplest instance-balanced (natural) sampling, it is also possible to achieve strong long-tailed recognition ability by adjusting only the classifier. We conduct extensive experiments and set new state-of-the-art performance on common long-tailed benchmarks like ImageNet-LT, Places-LT and iNaturalist, showing that it is possible to outperform carefully designed losses, sampling strategies, even complex modules with memory, by using a straightforward approach that decouples representation and classification. Our code is available at https://github.com/facebookresearch/classifier-balancing.

Bingyi Kang, Saining Xie, Marcus Rohrbach, Zhicheng Yan, Albert Gordo, Jiashi Feng, Yannis Kalantidis• 2019

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy50.76
3518
Image ClassificationCIFAR-10 (test)
Accuracy82.33
3381
Fine-grained Image ClassificationCUB200 2011 (test)
Accuracy87.7
536
Object DetectionLVIS v1.0 (val)
APbbox30.7
518
Fine-grained Image ClassificationStanford Cars (test)
Accuracy94.3
348
Fine-grained visual classificationFGVC-Aircraft (test)
Top-1 Acc92.1
287
Image ClassificationiNaturalist 2018
Top-1 Accuracy72.5
287
Image ClassificationImageNet LT
Top-1 Accuracy54
251
Long-Tailed Image ClassificationImageNet-LT (test)
Top-1 Acc (Overall)56
220
Image ClassificationCIFAR-10 long-tailed (test)
Top-1 Acc91.1
201
Showing 10 of 151 rows
...

Other info

Code

Follow for update