Share your thoughts, 1 month free Claude Pro on usSee more
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
543
Object DetectionLVIS v1.0 (val)
APbbox30.7
529
Fine-grained Image ClassificationStanford Cars (test)
Accuracy94.3
348
Fine-grained visual classificationFGVC-Aircraft (test)
Top-1 Acc92.1
312
Image ClassificationiNaturalist 2018
Top-1 Accuracy72.5
291
Image ClassificationImageNet LT
Top-1 Accuracy54
264
Image ClassificationCIFAR-100 Long-Tailed (test)
Top-1 Accuracy63.4
234
Long-Tailed Image ClassificationImageNet-LT (test)
Top-1 Acc (Overall)56
220
Showing 10 of 164 rows
...

Other info

Code

Follow for update