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

MetaSAug: Meta Semantic Augmentation for Long-Tailed Visual Recognition

About

Real-world training data usually exhibits long-tailed distribution, where several majority classes have a significantly larger number of samples than the remaining minority classes. This imbalance degrades the performance of typical supervised learning algorithms designed for balanced training sets. In this paper, we address this issue by augmenting minority classes with a recently proposed implicit semantic data augmentation (ISDA) algorithm, which produces diversified augmented samples by translating deep features along many semantically meaningful directions. Importantly, given that ISDA estimates the class-conditional statistics to obtain semantic directions, we find it ineffective to do this on minority classes due to the insufficient training data. To this end, we propose a novel approach to learn transformed semantic directions with meta-learning automatically. In specific, the augmentation strategy during training is dynamically optimized, aiming to minimize the loss on a small balanced validation set, which is approximated via a meta update step. Extensive empirical results on CIFAR-LT-10/100, ImageNet-LT, and iNaturalist 2017/2018 validate the effectiveness of our method.

Shuang Li, Kaixiong Gong, Chi Harold Liu, Yulin Wang, Feng Qiao, Xinjing Cheng• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationiNaturalist 2018
Top-1 Accuracy68.8
287
Image ClassificationImageNet LT
Top-1 Accuracy50.5
251
Long-Tailed Image ClassificationImageNet-LT (test)
Top-1 Acc (Overall)47.4
220
Image ClassificationCIFAR-10 long-tailed (test)
Top-1 Acc80.5
201
Image ClassificationiNaturalist 2018 (test)
Top-1 Accuracy68.8
192
Image ClassificationCIFAR-10-LT (test)
Top-1 Error0.1032
185
Image ClassificationImageNet-LT (test)--
159
Image ClassificationPlaces-LT (test)--
128
Image ClassificationCIFAR-100-LT IF 100 (test)
Top-1 Acc48
77
Image ClassificationCIFAR-LT-100 (test)
Top-1 Error42.15
74
Showing 10 of 43 rows

Other info

Code

Follow for update