Affiliation:
1. University of Florence, IT
Abstract
Deep neural networks (DNNs)
for social image classification are prone to performance reduction and overfitting when trained on datasets plagued by noisy or imbalanced labels. Weight loss methods tend to ignore the influence of noisy or frequent category examples during the training, resulting in a reduction of final accuracy and, in the presence of extreme noise, even a failure of the learning process. A new advisor network is introduced to address both imbalance and noise problems, and is able to pilot learning of a main network by adjusting the visual features and the gradient with a meta-learning strategy. In a curriculum learning fashion, the impact of redundant data is reduced while recognizable noisy label images are downplayed or redirected.
Meta Feature Re-Weighting (MFRW)
and
Meta Equalization Softmax (MES)
methods are introduced to let the main network focus only on the information in an image deemed relevant by the advisor network and to adjust the training gradient to reduce the adverse effects of frequent or noisy categories. The proposed method is first tested on synthetic versions of CIFAR10 and CIFAR100, and then on the more realistic ImageNet-LT, Places-LT, and Clothing1M datasets, reporting state-of-the-art results.
Funder
European Commission under European Horizon 2020 Programme
Publisher
Association for Computing Machinery (ACM)
Subject
Computer Networks and Communications,Hardware and Architecture
Reference59 articles.
1. Image classification with deep learning in the presence of noisy labels: A survey
2. Eric Arazo, Diego Ortego, Paul Albert, Noel O’Connor, and Kevin McGuinness. 2019. Unsupervised label noise modeling and loss correction. In International Conference on Machine Learning. PMLR, 312–321.
3. Learning imbalanced datasets with label-distribution-aware margin loss;Cao Kaidi;arXiv preprint arXiv:1906.07413,2019
4. SMOTE: Synthetic Minority Over-sampling Technique
5. Generalized DataWeighting via class-level gradient manipulation;Chen Can;Advances in Neural Information Processing Systems,2021
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献