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

Self-Supervised Aggregation of Diverse Experts for Test-Agnostic Long-Tailed Recognition

About

Existing long-tailed recognition methods, aiming to train class-balanced models from long-tailed data, generally assume the models would be evaluated on the uniform test class distribution. However, practical test class distributions often violate this assumption (e.g., being either long-tailed or even inversely long-tailed), which may lead existing methods to fail in real applications. In this paper, we study a more practical yet challenging task, called test-agnostic long-tailed recognition, where the training class distribution is long-tailed while the test class distribution is agnostic and not necessarily uniform. In addition to the issue of class imbalance, this task poses another challenge: the class distribution shift between the training and test data is unknown. To tackle this task, we propose a novel approach, called Self-supervised Aggregation of Diverse Experts, which consists of two strategies: (i) a new skill-diverse expert learning strategy that trains multiple experts from a single and stationary long-tailed dataset to separately handle different class distributions; (ii) a novel test-time expert aggregation strategy that leverages self-supervision to aggregate the learned multiple experts for handling unknown test class distributions. We theoretically show that our self-supervised strategy has a provable ability to simulate test-agnostic class distributions. Promising empirical results demonstrate the effectiveness of our method on both vanilla and test-agnostic long-tailed recognition. Code is available at \url{https://github.com/Vanint/SADE-AgnosticLT}.

Yifan Zhang, Bryan Hooi, Lanqing Hong, Jiashi Feng• 2021

Related benchmarks

TaskDatasetResultRank
Long-Tailed Image ClassificationImageNet-LT (test)
Top-1 Acc (Overall)61.4
220
Image ClassificationCIFAR-10 long-tailed (test)--
201
Image ClassificationiNaturalist 2018 (test)--
192
Image ClassificationCIFAR-10-LT (test)--
185
Image ClassificationImageNet-LT (test)
Top-1 Acc (All)58.8
159
Image ClassificationCIFAR-100 Long-Tailed (test)--
149
Image ClassificationPlaces-LT (test)
Accuracy (Medium)43.2
128
Long-tailed Visual RecognitionImageNet LT
Overall Accuracy58.8
89
Long-Tailed Image ClassificationiNaturalist 2018
Accuracy72.9
82
Image ClassificationCIFAR-100-LT IF 100 (test)
Top-1 Acc52.2
77
Showing 10 of 67 rows

Other info

Code

Follow for update