中文

面向图像分类的注意力感知噪声标签学习

计算机视觉与模式识别 2020-10-01 v1

摘要

在大规模标注样本上训练的深度卷积神经网络(CNN)已在计算机视觉(如图像/视频分类)中取得显著进展。获取大量标注视觉数据最廉价的方式是从带有用户标注的网站(如 Flickr)抓取。然而,这些样本往往包含错误标签(即噪声标签),会显著降低网络性能。本文提出注意力感知噪声标签学习方法(A2NLA^2NL),以提升在含潜在标签噪声数据集上训练的网络判别能力。具体而言,设计了一个包含多个噪声专用单元的噪声注意力模型,以更好地捕捉噪声信息。每个单元期望为图像子集学习特定的噪声分布,从而更精确地建模不同干扰。此外,引入递归学习过程,利用已学习的高层知识增强注意力网络的学习能力。为充分评估所提方法,我们从两方面开展实验:在 CIFAR-10、SVHN 等大规模图像分类数据集上的手动翻转标签噪声,以及在线抓取的带多属性服装数据集上的真实世界标签噪声。优于 SOTA 方法的结果验证了我们所提方法的有效性。

关键词

引用

@article{arxiv.2009.14757,
  title  = {Attention-Aware Noisy Label Learning for Image Classification},
  author = {Zhenzhen Wang and Chunyan Xu and Yap-Peng Tan and Junsong Yuan},
  journal= {arXiv preprint arXiv:2009.14757},
  year   = {2020}
}

备注

10 pages, 8 figures