Long-Tailed Recognition via Information-Preservable Two-Stage Learning
About
The imbalance (or long-tail) is the nature of many real-world data distributions, which often induces the undesirable bias of deep classification models toward frequent classes, resulting in poor performance for tail classes. In this paper, we propose a novel two-stage learning approach to mitigate such a majority-biased tendency while preserving valuable information within datasets. Specifically, the first stage proposes a new representation learning technique from the information theory perspective. This approach is theoretically equivalent to minimizing intra-class distance, yielding an effective and well-separated feature space. The second stage develops a novel sampling strategy that selects mathematically informative instances, able to rectify majority-biased decision boundaries without compromising a model's overall performance. As a result, our approach achieves the state-of-the-art performance across various long-tailed benchmark datasets, validated via extensive experiments. Our code is available at https://github.com/fudong03/BNS_IPDPP.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Long-Tailed Image Classification | ImageNet-LT (test) | Top-1 Acc (Overall)51.7 | 246 | |
| Image Classification | CIFAR-10-LT IF 100 | Top-1 Accuracy76.4 | 65 | |
| Image Classification | CIFAR-100 LT (IF=100) | Top-1 Acc52.4 | 42 | |
| Long-tailed recognition | iNaturalist 2018 | Top-1 Accuracy74 | 31 |