arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.04173cs.CV

LongReMix: Robust Learning with High Confidence Samples in a Noisy Label Environment

Universidade Federal Rural de Pernambuco(伯南布哥联邦农村大学);University of Oxford(牛津大学);Otto-von-Guericke-Universität Magdeburg(马格德堡奥托·冯·格里克大学);Australian Institute for Machine Learning(澳大利亚机器学习研究所);University of Surrey(萨里大学)

Filipe R. Cordeiro, Ragav Sachdeva, Vasileios Belagiannis, Ian Reid, Gustavo Carneiro

英文摘要:

Deep neural network models are robust to a limited amount of label noise, but their ability to memorise noisy labels in high noise rate problems is still an open issue. The most competitive noisy-label learning algorithms rely on a 2-stage process comprising an unsupervised learning to classify training samples as clean or noisy, followed by a semi-supervised learning that minimises the empirical vicinal risk (EVR) using a labelled set formed by samples classified as clean, and an unlabelled set with samples classified as noisy. In this paper, we hypothesise that the generalisation of such 2-stage noisy-label learning methods depends on the precision of the unsupervised classifier and the size of the training set to minimise the EVR. We empirically validate these two hypotheses and propose the new 2-stage noisy-label training algorithm LongReMix. We test LongReMix on the noisy-label benchmarks CIFAR-10, CIFAR-100, WebVision, Clothing1M, and Food101-N. The results show that our LongReMix generalises better than competing approaches, particularly in high label noise problems. Furthermore, our approach achieves state-of-the-art performance in most datasets. The code is available at https://github.com/filipe-research/LongReMix.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑