基于对比学习的标签带噪图像分类
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(62201282);江苏省自然科学基金(BK20231456)


Label Noisy Image Classification Based on Contrastive Learning
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    标签噪声会极大地降低深度网络模型的性能. 针对这一问题, 本文提出了一种基于对比学习的标签带噪图像分类方法. 该方法包括自适应阈值、对比学习模块和基于类原型的标签去噪模块. 首先采用对比学习最大化一幅图像的两个增强视图的相似度来提取图像鲁棒特征; 接下来通过一种新颖的自适应阈值过滤训练样本, 在模型训练过程中根据各个类别的学习情况动态调整阈值; 然后创新性地引入基于类原型的标签去噪模块, 通过计算样本特征向量与原型向量的相似度更新伪标签, 从而避免标签中噪声的影响; 在公开数据集CIFAR-10、CIFAR-100和真实数据集ANIMAL10上进行对比实验, 实验结果表明, 在人工合成噪声的条件下, 本文方法实验结果均高于常规方法, 通过计算图像鲁棒的特征向量与各个原型向量的相似度更新伪标签的方式, 降低了噪声标签的负面影响, 在一定程度上提高模型的抗噪声能力, 验证了该模型的有效性.

    Abstract:

    Label noise can greatly reduce the performance of deep network models. To address this problem, this study proposes a contrastive learning-based label noisy image classification method. The method includes an adaptive threshold, contrastive learning module, and class prototype-based label denoising module. Firstly, the robust features of the image are extracted by maximizing the similarity between two augmented views of the same image using contrastive learning. Then, a novel adaptive threshold filtering training sample is used to dynamically adjust the threshold based on the learning status of each class during model training. Finally, a class prototype-based label denoising module is introduced to update pseudo-labels by calculating the similarity between sample feature vectors and prototype vectors, thus avoiding the influence of label noise. Comparative experiments are conducted on the publicly available datasets CIFAR-10 and CIFAR-100 and the real dataset ANIMAL10. The experimental results show that under the condition of artificially synthesized noise, the proposed method outperforms conventional methods. By updating pseudo-labels based on the similarity between the robust feature vector of the image and various prototype vectors, the negative impact of noisy labels is reduced, and the anti-noise ability of the model is improved to certain extent, verifying the effectiveness of the proposed model.

    参考文献
    相似文献
    引证文献
引用本文

李俊哲,曹国.基于对比学习的标签带噪图像分类.计算机系统应用,2023,32(12):104-111

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2023-06-21
  • 最后修改日期:2023-07-19
  • 录用日期:
  • 在线发布日期: 2023-10-20
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京海淀区中关村南四街4号 中科院软件园区 7号楼305房间,邮政编码:100190
电话:010-62661041 传真: Email:csa (a) iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号