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

Long-tailed Visual Recognition via Gaussian Clouded Logit Adjustment

About

Long-tailed data is still a big challenge for deep neural networks, even though they have achieved great success on balanced data. We observe that vanilla training on long-tailed data with cross-entropy loss makes the instance-rich head classes severely squeeze the spatial distribution of the tail classes, which leads to difficulty in classifying tail class samples. Furthermore, the original cross-entropy loss can only propagate gradient short-lively because the gradient in softmax form rapidly approaches zero as the logit difference increases. This phenomenon is called softmax saturation. It is unfavorable for training on balanced data, but can be utilized to adjust the validity of the samples in long-tailed data, thereby solving the distorted embedding space of long-tailed problems. To this end, this paper proposes the Gaussian clouded logit adjustment by Gaussian perturbation of different class logits with varied amplitude. We define the amplitude of perturbation as cloud size and set relatively large cloud sizes to tail classes. The large cloud size can reduce the softmax saturation and thereby making tail class samples more active as well as enlarging the embedding space. To alleviate the bias in a classifier, we therefore propose the class-based effective number sampling strategy with classifier re-training. Extensive experiments on benchmark datasets validate the superior performance of the proposed method. Source code is available at https://github.com/Keke921/GCLLoss.

Mengke Li, Yiu-ming Cheung, Yang Lu• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationiNaturalist 2018
Top-1 Accuracy72
287
Image ClassificationImageNet LT
Top-1 Accuracy54.9
251
Image ClassificationCIFAR-10 long-tailed (test)--
201
Image ClassificationiNaturalist 2018 (test)
Top-1 Accuracy72.01
192
Image ClassificationImageNet-LT (test)
Top-1 Acc (All)54.5
159
Image ClassificationPlaces-LT (test)--
128
Image ClassificationCIFAR-100-LT Imbalance Ratio 100
Top-1 Acc0.486
88
Long-Tailed Image ClassificationiNaturalist 2018
Accuracy71
82
Image ClassificationCIFAR-100-LT (Imbalance Ratio 50)
Top-1 Accuracy53.6
61
Long-Tailed Image ClassificationPlaces-LT (test)
Accuracy40.3
61
Showing 10 of 30 rows

Other info

Code

Follow for update