Symmetric Cross Entropy for Robust Learning With Noisy Labels

By Yisen Wang, Xingjun Ma, Zaiyi Chen, Yuan Luo, Jinfeng Yi, James Bailey. 2019

1,167 citations recorded by the source corpus.

Training accurate deep neural networks (DNNs) in the presence of noisy labels is an important and challenging task. Though a number of approaches have been proposed for learning with noisy labels, many open issues remain. In this paper, we show that DNN learning with Cross Entropy (CE) exhibits overfitting to noisy labels on some classes (``easy" classes), but more surprisingly, it also suffers from significant under learning on some other classes (``hard" classes). Intuitively, CE requires an extra term to facilitate learning of hard classes, and more importantly, this term should be noise tolerant, so as to avoid overfitting to noisy labels. Inspired by the symmetric KL-divergence, we propose the approach of Symmetric cross entropy Learning (SL), boosting CE symmetrically with a noise robust counterpart Reverse Cross Entropy (RCE). Our proposed SL approach simultaneously addresses both the under learning and overfitting problem of CE in the presence of noisy labels. We provide a theoretical analysis of SL and also empirically show, on a range of benchmark and real-world datasets, that SL outperforms state-of-the-art methods. We also show that SL can be easily incorporated into existing methods in order to further enhance their performance.

Subfields: Artificial Intelligence, Civil and Structural Engineering

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Xingjun Ma
  2. Yisen Wang
  3. Wen Li
  4. James Bailey
  5. Bo Han
  6. Görkem Algan
  7. Ilkay Ulusoy
  8. Yahao Liu

In Civil and Structural Engineering

  1. Tongliang Liu
  2. Alexander J. Ratner
  3. Ran He
  4. Xiansheng Hua
  5. Tien-Ping Tan
  6. James Bailey
  7. Görkem Algan
  8. Ilkay Ulusoy

Citations in the corpus