Combating Noisy Labels through Fostering Self- and Neighbor-Consistency
About
Label noise is pervasive in various real-world scenarios, posing challenges in supervised deep learning. Deep networks are vulnerable to such label-corrupted samples due to the memorization effect. One major stream of previous methods concentrates on identifying clean data for training. However, these methods often neglect imbalances in label noise across different mini-batches and devote insufficient attention to out-of-distribution noisy data. To this end, we propose a noise-robust method named Jo-SNC (\textbf{Jo}int sample selection and model regularization based on \textbf{S}elf- and \textbf{N}eighbor-\textbf{C}onsistency). Specifically, we propose to employ the Jensen-Shannon divergence to measure the ``likelihood'' of a sample being clean or out-of-distribution. This process factors in the nearest neighbors of each sample to reinforce the reliability of clean sample identification. We design a self-adaptive, data-driven thresholding scheme to adjust per-class selection thresholds. While clean samples undergo conventional training, detected in-distribution and out-of-distribution noisy samples are trained following partial label learning and negative learning, respectively. Finally, we advance the model performance further by proposing a triplet consistency regularization that promotes self-prediction consistency, neighbor-prediction consistency, and feature consistency. Extensive experiments on various benchmark datasets and comprehensive ablation studies demonstrate the effectiveness and superiority of our approach over existing state-of-the-art methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | ILSVRC 2012 (val) | -- | 156 | |
| Image Classification | ILSVRC 2012 (test) | Top-1 Acc80.16 | 117 | |
| Image Classification | ANIMAL-10N | Accuracy0.8617 | 53 | |
| Image Classification | CIFAR-100 Sym-20% (test) | Accuracy65.49 | 33 | |
| Image Classification | CompCars Web (test) | Top-1 Acc90.56 | 33 | |
| Image Classification | CIFAR-100 Sym-50% (test) | Accuracy59.88 | 32 | |
| Image Classification | Web-Aircraft (test) | Test Accuracy87.97 | 26 | |
| Image Classification | Web-Bird (test) | Accuracy82.76 | 26 | |
| Image Classification | WebVision mini (test) | Top-1 Acc82.6 | 21 | |
| Classification | CIFAR-100 Symmetry-80% noise (test) | Test Accuracy43.45 | 17 |