Long-tailed Recognition by Learning from Latent Categories
About
In this work, we address the challenging task of long-tailed image recognition. Previous long-tailed recognition methods commonly focus on the data augmentation or re-balancing strategy of the tail classes to give more attention to tail classes during the model training. However, due to the limited training images for tail classes, the diversity of tail class images is still restricted, which results in poor feature representations. In this work, we hypothesize that common latent features among the head and tail classes can be used to give better feature representation. Motivated by this, we introduce a Latent Categories based long-tail Recognition (LCReg) method. Specifically, we propose to learn a set of class-agnostic latent features shared among the head and tail classes. Then, we implicitly enrich the training sample diversity via applying semantic data augmentation to the latent features. Extensive experiments on five long-tailed image recognition datasets demonstrate that our proposed LCReg is able to significantly outperform previous methods and achieve state-of-the-art results.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Long-Tailed Image Classification | ImageNet-LT (test) | Top-1 Acc (Overall)55.3 | 220 | |
| Image Classification | CIFAR-10 long-tailed (test) | Top-1 Acc83.1 | 201 | |
| Image Classification | iNaturalist 2018 (test) | Top-1 Accuracy72.6 | 192 | |
| Image Classification | Places-LT (test) | -- | 128 | |
| Image Classification | CIFAR-100-LT IF 100 (test) | Top-1 Acc47.6 | 77 | |
| Long-tail Image Classification | iNaturalist 2018 (test) | Accuracy (Few)71.5 | 59 | |
| Image Classification | CIFAR-10-LT IF 50 (test) | Top-1 Acc86.5 | 26 | |
| Long-tailed Image Recognition | CIFAR100 LT IF 100 (test) | Acc (Many)64.2 | 21 | |
| Image Classification | CIFAR-100 IF=10 long-tail (test) | Top-1 Accuracy64.2 | 19 | |
| Image Classification | CIFAR-100 (IF=50) long-tail (test) | Top-1 Acc53.1 | 17 |