Confident Learning: Estimating Uncertainty in Dataset Labels
Curtis G. Northcutt Affiliation: Cleanlab Affiliation: Massachusetts Institute of Technology, Affiliation: Department of EECS, Cambridge, MA, USA Lu Jiang Affiliation: Google Research, Mountain View, CA, USA Isaac L. Chuang Affiliation: Massachusetts Institute of Technology, Affiliation: Department of EECS, Department of Physics, Cambridge, MA, USA
Abstract
Learning exists in the context of data, yet notions of confidence typically focus on model predictions, not label quality. Confident learning (CL) is a data-centric approach which focuses instead on label quality by characterizing and identifying label errors in datasets, based on the principles of pruning noisy data, counting with probabilistic thresholds to estimate noise, and ranking examples to train with confidence. Whereas numerous studies have developed these principles independently, here, we combine them, building on the assumption of a class-conditional noise process to directly estimate the joint distribution between noisy (given) labels and uncorrupted (unknown) labels. This results in a generalized CL which is provably consistent and experimentally performant. We present sufficient conditions where CL exactly finds label errors, and show CL performance exceeding seven recent competitive approaches for learning with noisy labels on the CIFAR dataset. Uniquely, the CL framework is not coupled to a specific data modality or model (e.g., we use CL to find several label errors in the presumed error-free MNIST dataset and improve sentiment classification on text data in Amaz
原文 arXiv:1911.00068;中英对照 + 大白话阅读 https://aha.fim.ai/paper/1911.00068v6