Understanding and Utilizing Deep Neural Networks Trained with Noisy Labels

Pengfei Chen, Benben Liao, Guangyong Chen, Shengyu Zhang
5/13/2019
cs.LGstat.ML

Official code: chenpf1025/noisy_label_understanding_utilizing (91 stars, last commit 2020-12-10).

Abstract

Noisy labels are ubiquitous in real-world datasets, which poses a challenge for robustly training deep neural networks (DNNs) as DNNs usually have the high capacity to memorize the noisy labels. In this paper, we find that the test accuracy can be quantitatively characterized in terms of the noise ratio in datasets. In particular, the test accuracy is a quadratic function of the noise ratio in the case of symmetric noise, which explains the experimental findings previously published. Based on our analysis, we apply cross-validation to randomly split noisy datasets, which identifies most samples that have correct labels. Then we adopt the Co-teaching strategy which takes full advantage of the identified samples to train DNNs robustly against noisy labels. Compared with extensive state-of-the-art methods, our strategy consistently improves the generalization performance of DNNs under both synthetic and real-world training noise.

View on arXivView PDF

Code Implementations(1)

ICML 2019: Understanding and Utilizing Deep Neural Networks Trained with Noisy Labels

9124PythonMay 4, 20195 years ago
icmlnoisy-labelsrobust

Similar Papers

Cite this paper

@article{chen2019understanding,
  title  = {Understanding and Utilizing Deep Neural Networks Trained with Noisy Labels},
  author = {Pengfei Chen and Benben Liao and Guangyong Chen and Shengyu Zhang},
  year   = {2019},
  eprint = {1905.05040},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1905.05040}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1905.05040.svg)](https://codeofpaper.com/paper/1905.05040)

Discussion