[edit]
Learning with Confident Examples: Rank Pruning for Robust Classification with Noisy Labels
Proceedings of the 33rd Conference on Uncertainty in Artificial Intelligence, PMLR R15:531-540, 2017.
Abstract
$ ilde{P}$ $ ilde{N}$ learning is the problem of binary classification when training examples may be mislabeled (flipped) uniformly with noise rate $\rho$1 for positive examples and $
ho$0 for negative examples. We propose Rank Pruning (RP) to solve $ ilde{P}$ $ ilde{N}$ learning and the open problem of estimating the noise rates. Unlike prior solutions, RP is efficient and general, requiring $O(T)$ for any unrestricted choice of probabilistic classifier with $T$ fitting time. We prove RP achieves consistent noise estimation and equivalent expected risk as learning with uncorrupted labels in ideal conditions, and derive closed-form solutions when conditions are non-ideal. RP achieves state-of-the-art noise estimation and F1, error, and AUC-PR for both MNIST and CIFAR datasets, regardless of the amount of noise. To highlight, RP with a CNN classifier can predict if an MNIST digit is a one or not with only 0.25% error, and 0.46% error across all digits, even when 50% of positive examples are mislabeled and 50% of observed positive labels are mislabeled negative examples.