Normalized Loss Functions for Deep Learning with Noisy Labels

By Xingjun Ma, Hanxun Huang, Yisen Wang, Simone Romano, Sarah Monazam Erfani, James Bailey. International Conference on Machine Learning, 2020

563 citations recorded by the source corpus.

Robust loss functions are essential for training accurate deep neural networks (DNNs) in the presence of noisy (incorrect) labels. It has been shown that the commonly used Cross Entropy (CE) loss is not robust to noisy labels. Whilst new loss functions have been designed, they are only partially robust. In this paper, we theoretically show by applying a simple normalization that: any loss can be made robust to noisy labels. However, in practice, simply being robust is not sufficient for a loss function to train accurate DNNs. By investigating several robust loss functions, we find that they suffer from a problem of underfitting. To address this, we propose a framework to build robust loss functions called Active Passive Loss (APL). APL combines two robust loss functions that mutually boost each other. Experiments on benchmark datasets demonstrate that the family of new loss functions created by our APL framework can consistently outperform state-of-the-art methods by large margins, especially under large noise rates such as 60% or 80% incorrect labels.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Signal Processing

View Paper: https://www.semanticscholar.org/paper/07cca761749bfe21c2d096ff60f32b574d5c84c4

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. James Bailey
  3. Tongliang Liu
  4. Quoc V. Le
  5. Jeff Wu
  6. Bo Han
  7. Pavel Izmailov
  8. Gang Niu

In Computer Vision and Pattern Recognition

  1. Xiansheng Hua
  2. Tongliang Liu
  3. Xingjun Ma
  4. James Bailey
  5. Pavel Izmailov
  6. ZongYuan Ge
  7. Bo Han
  8. Xi Peng

In Signal Processing

  1. Tongliang Liu
  2. Xiansheng Hua
  3. Xi Peng
  4. Sarah Monazam Erfani
  5. James Bailey
  6. Yi Chang
  7. Ilya Sutskever
  8. Leo Gao

Citations in the corpus