3 个月前

通过自监督对抗性噪声掩码进行带噪声标签的学习

通过自监督对抗性噪声掩码进行带噪声标签的学习

摘要

大规模数据集的收集对于深度模型的训练至关重要,然而数据标注过程不可避免地会产生噪声标签,这给深度学习算法带来了严峻挑战。以往的研究通常通过识别并剔除噪声样本,或根据训练样本间的统计特性(如损失值)对标签进行修正来缓解该问题。本文从一个全新的视角出发,深入分析深度特征图,我们通过实证发现:使用干净样本与错误标注样本训练的模型,其激活特征分布具有明显可区分性。基于这一观察,本文提出一种新颖的鲁棒训练方法——对抗性噪声掩码(adversarial noisy masking)。该方法通过一种由标签质量引导的掩码机制对深度特征进行正则化,能够自适应地同时调节输入数据与标签,从而防止模型对噪声样本过拟合。此外,我们设计了一个辅助任务以重建输入数据,该任务自然地提供了无噪声的自监督信号,有效增强了深度模型的泛化能力。所提方法结构简洁、灵活,已在合成数据集和真实世界噪声数据集上进行了验证,相较于现有最先进方法取得了显著的性能提升。

代码仓库

yuanpengtu/SANM
官方
pytorch

基准测试

基准方法指标
image-classification-on-clothing1mSANM (DivideMix)
Accuracy: 75.63%

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过自监督对抗性噪声掩码进行带噪声标签的学习 | 论文 | HyperAI超神经