
摘要
带有噪声标签的深度学习是弱监督学习中一个具有实际挑战性的问题。当前最先进的方法——“解耦(Decoupling)”与“Co-teaching+”均认为,“分歧(disagreement)”策略对于缓解噪声标签学习问题至关重要。本文从一个不同的视角出发,提出了一种名为JoCoR的鲁棒学习范式,旨在训练过程中降低两个网络之间的差异性。具体而言,我们首先利用两个网络对同一小批量数据进行预测,并为每个训练样本计算一个联合损失(joint loss),该损失基于协同正则化(Co-Regularization)机制。随后,我们选择损失值较小的样本,同时更新两个网络的参数。由于协同正则化的作用,这两个网络在联合损失的驱动下,训练过程中会逐渐趋于相似。在MNIST、CIFAR-10、CIFAR-100以及Clothing1M等基准数据集上经过标签污染的实验结果表明,JoCoR在噪声标签学习任务中显著优于多种当前最先进的方法。
代码仓库
hongxin001/JoCoR
官方
pytorch
hongxin001/ODNL
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| image-classification-on-clothing1m | JoCoR | Accuracy: 70.3% |
| learning-with-noisy-labels-on-cifar-100n | JoCoR | Accuracy (mean): 59.97 |
| learning-with-noisy-labels-on-cifar-10n | JoCoR | Accuracy (mean): 91.44 |
| learning-with-noisy-labels-on-cifar-10n-1 | JoCoR | Accuracy (mean): 90.30 |
| learning-with-noisy-labels-on-cifar-10n-2 | JoCoR | Accuracy (mean): 90.21 |
| learning-with-noisy-labels-on-cifar-10n-3 | JoCoR | Accuracy (mean): 90.11 |
| learning-with-noisy-labels-on-cifar-10n-worst | JoCoR | Accuracy (mean): 83.37 |