3 个月前

通过一致性来对抗噪声标签:一种具有协同正则化的联合训练方法

通过一致性来对抗噪声标签:一种具有协同正则化的联合训练方法

摘要

带有噪声标签的深度学习是弱监督学习中一个具有实际挑战性的问题。当前最先进的方法——“解耦(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-clothing1mJoCoR
Accuracy: 70.3%
learning-with-noisy-labels-on-cifar-100nJoCoR
Accuracy (mean): 59.97
learning-with-noisy-labels-on-cifar-10nJoCoR
Accuracy (mean): 91.44
learning-with-noisy-labels-on-cifar-10n-1JoCoR
Accuracy (mean): 90.30
learning-with-noisy-labels-on-cifar-10n-2JoCoR
Accuracy (mean): 90.21
learning-with-noisy-labels-on-cifar-10n-3JoCoR
Accuracy (mean): 90.11
learning-with-noisy-labels-on-cifar-10n-worstJoCoR
Accuracy (mean): 83.37

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过一致性来对抗噪声标签:一种具有协同正则化的联合训练方法 | 论文 | HyperAI超神经