Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Estimating Instance-dependent Bayes-label Transition Matrix using a Deep Neural Network

About

In label-noise learning, estimating the transition matrix is a hot topic as the matrix plays an important role in building statistically consistent classifiers. Traditionally, the transition from clean labels to noisy labels (i.e., clean-label transition matrix (CLTM)) has been widely exploited to learn a clean label classifier by employing the noisy data. Motivated by that classifiers mostly output Bayes optimal labels for prediction, in this paper, we study to directly model the transition from Bayes optimal labels to noisy labels (i.e., Bayes-label transition matrix (BLTM)) and learn a classifier to predict Bayes optimal labels. Note that given only noisy data, it is ill-posed to estimate either the CLTM or the BLTM. But favorably, Bayes optimal labels have less uncertainty compared with the clean labels, i.e., the class posteriors of Bayes optimal labels are one-hot vectors while those of clean labels are not. This enables two advantages to estimate the BLTM, i.e., (a) a set of examples with theoretically guaranteed Bayes optimal labels can be collected out of noisy data; (b) the feasible solution space is much smaller. By exploiting the advantages, we estimate the BLTM parametrically by employing a deep neural network, leading to better generalization and superior classification performance.

Shuo Yang, Erkun Yang, Bo Han, Yang Liu, Min Xu, Gang Niu, Tongliang Liu• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy68.42
3518
Semantic segmentationS3DIS (Area 5)
mIOU62.4
1006
Image ClassificationCIFAR-100 (val)
Accuracy43.5
781
Image ClassificationSVHN (test)
Accuracy95.51
470
Semantic segmentationScanNet V2 (val)
mIoU64.8
380
Matrix EstimationMovieLens Symmetric Noise
L1 Distance Error0.7932
40
Matrix EstimationMovieLens Pairflip Noise
L1 Distance Error2.0672
40
RecommendationMovieLens
Recall@108.65
32
RecommendationYelp
Recall@100.0324
32
Matrix EstimationSymmetric Noise (test)
L1 Estimation Error1.4501
32
Showing 10 of 28 rows

Other info

Follow for update