Classification with imperfect training labels

Timothy Cannings, Yingying Fan, Richard J. Samworth

Research output: Contribution to journalArticlepeer-review

Abstract

We study the effect of imperfect training data labels on the performance of classification methods. In a general setting, where the probability that an observation in the training dataset is mislabelled may depend on both the feature vector and the true label, we bound the excess risk of an arbitrary classifier trained with imperfect labels in terms of its excess risk for predicting a noisy label. This reveals conditions under which a classifier trained with imperfect labels remains consistent for classifying uncorrupted test data points. Furthermore, under stronger conditions, we derive detailed asymptotic properties for the popular k-nearest neighbour (knn), support vector machine (SVM) and linear discriminant analysis (LDA) classifiers. One consequence of these results is that the knn and SVM classifiers are robust to imperfect training labels, in the sense that the rate of convergence of the excess risks of these classifiers remains unchanged; in fact, our theoretical and empirical results even show that in some cases, imperfect labels may improve the performance of these methods. On the other hand, the LDA classifier is shown to be typically inconsistent in the presence of label noise unless the prior probabilities of each class are equal. Our theoretical results are supported by a simulation study.
Original languageEnglish
Pages (from-to)311-330
Number of pages29
JournalBiometrika
Volume107
Issue number2
Early online date22 Apr 2020
DOIs
Publication statusPublished - 30 Jun 2020

Fingerprint

Dive into the research topics of 'Classification with imperfect training labels'. Together they form a unique fingerprint.

Cite this