Label Noise Cleaning for Supervised Classification via Bernoulli Random Sampling
Authors: Yuxin Liu, Xiong Jin, Yang Han
Organizations: Department of Mathematics, The University of Manchester, UK · Department of Mathematics, The University of Manchester, Oxford Road, Manchester, M13 9PL, UK
Label noise - incorrect labels assigned to observations - can substantially degrade the performance of supervised classifiers. This paper proposes a label noise cleaning method based on Bernoulli random sampling. We show that the mean label noise levels of subsets generated by Bernoulli random sampling containing a given observation are identically distributed for all clean observations, and identically distributed, with a different distribution, for all noisy observations. Although the mean label noise levels are not independent across observations, by introducing an independent coupling we further prove that they converge to a mixture of two well-separated distributions corresponding to clean and noisy observations. By establishing a linear model between cross-validated classification errors and label noise levels, we are able to approximate this mixture distribution and thereby separate clean and noisy observations without any prior label information. The proposed method is classifier-agnostic, theoretically justified, and demonstrates strong performance on both simulated and real datasets.
Figures & tables
Figure 1: Empirical distribution of {l~z}z∈D obtained from a dataset with N=1000 observations ( N1=800 clean and N2=200 noisy). The values of l~z are computed under the Bernoulli random sampling scheme with M=2×106 subsets and inclusion probability q=0.4 . The histogram shows two well-separated laws corresponding to L1 and L2 , with a mean gap E[(N−1)q+11−q]≈0.0015 .
Figure 2: Empirical distribution of the i.i.d. sequence {l~z′}z∈D generated under the same settings as Figure 1 . The mixture shows two components, approximately corresponding to L1 and L2 , with a mean gap of (N−1)q+11−q≈0.0015 , consistent with Lemma 2.3 .
Figure 3: Linear model between l~z and e~z under Setting 1 with 20% label noise. Results are shown for classifiers trained with RBF SVM and 1-NN, as described in Section 4 .
Figure 4: Comparison between the empirical and model-estimated distributions of l~z . The estimated distribution l^z closely matches the empirical l~z , and the estimated cut point l^∗ aligns well with the true l∗ .
Figure 5: Three-iteration BRSLC performance of RBF SVM and 1-NN. The legend <e∗ denotes #{z∈D∣e~z<e∗}#{z∈D1∣e~z<e∗} , ≥e∗ denotes #{z∈D∣e~z≥e∗}#{z∈D2∣e~z≥e∗} .
Setting
ℓ(D) (%)
ℓcleaned (%)
rnoise (%)
rclean (%)
1
10
0.54
94.6
96.7
20
1.18
94.1
93.2
30
2.62
91.3
92.0
40
3.40
91.5
88.6
2
10
1.06
89.4
95.7
20
2.20
89.0
92.6
Table 1: Summary of label noise cleaning performance across four simulated data settings. All values are expressed as percentages. ℓ(D) : original label noise level; ℓcleaned : residual label noise level after cleaning; rnoise=100×[1−ℓcleaned/ℓ(D)] : proportion of label noise removed; rclean=100×(ncleaned/nclean) : proportion of clean observations retained after cleaning.
Figure 6: Test error rates of noisy and cleaned datasets across the four simulated data settings. Each panel corresponds to one setting, showing results for three classifiers: RBF SVM, CART, and 1-NN. Solid lines represent performance on after BRSLC cleaning, while dashed lines represent performance of original noisy data.
Figure 7: Examples of letters B, H, R in dataset UCI letter recognition, and digits 4, 7, 9 in dataset MNIST digit recognition.
Experiment
Label 0
Label 1
ntrain
ntest
WDBC
Malignant
Benign
400
100
Mushroom
Edible
Poisonous
1000
500
Letter-B
B
H and R
1000
200
Letter-H
H
B and R
1000
200
Letter-R
R
B and H
1000
200
MNIST-4
4
7 and 9
2000
1000
Table 2: Summary of the eight experiments in our study. Each row defines a binary classification task derived from an underlying dataset. For the Letter and MNIST subsets, the total sample sizes for Label 0 and Label 1 follow a 50:50 split. Within Label 1, the two constituent classes are balanced in a 50:50 ratio as well.
ℓ(D) (%)
Method
Classifier
ℓcleaned (%)
rclean (%)
10
ALNR
Linear SVM
1.1
89.4
RBF SVM
1.0
86.0
BRSLC
Linear SVM
1.0
90.1
RBF SVM
0.5
89.6
20
ALNR
Linear SVM
2.9
85.9
RBF SVM
2.4
83.0
Table 3: Average results of ALNR and BRSLC across eight datasets under different noise levels ℓ(D) . ℓcleaned : residual label noise level after cleaning; rclean : proportion of clean observations retained after cleaning, expressed as a percentage. All values are averaged over datasets and classifiers.
Figure 8: Average test error rates of Linear SVM trained on noisy data, and data cleaned by ALNR and BRSLC.
Figure 9: Examples of CIFAR-10H soft labels. Top: noisy soft label distributions. Bottom: cleaned soft label distributions obtained after applying the proposed method.
n
ncleaned
lcleaned
500
484.37
0.0085
3500
3285.89
0.0089
6500
6211.01
0.0086
9500
8852.58
0.0083
Table 4: Averaged ResNet-18 convolutional neural network label cleaning results over 20 random subsets from CIFAR-10H of sizes 500, 3500, 6500, and 9500. n is the size of the noisy subset, ncleaned is the size after cleaning, lcleaned is the purified noise rate.
Figure 10: Comparison between the noisy data and the adaptive method proposed by M. Sesia in terms of coverage and size, computed over 10 CIFAR-10H random calibration sets of size 500–9500.
Department of Electrical Eng. National Taiwan Ocean University, Keelung City 20224, Taiwan · AI Research Center, National Taiwan Ocean University, Keelung City 20224, Taiwan
Department of Computer Science University of Toronto The Hospital for Sick Children UHN KITE Research Institute T-CAIREM Vector Institute · Department of Computer Science Institute of Biomedical Engineering University of Toronto Rehabilitation Sciences Institute UHN KITE Research Institute Vector Institute · Department of Computer Science Department of Laboratory Medicine and Pathobiology University of Toronto The Hospital for Sick Children T-CAIREM Vector Institute