Noise Label Image Classification Based on Semi-supervised Multi-dimensional Contrastive Learning
Author:
  • Article
  • | |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • | |
  • Comments
    Abstract:

    Deep learning-based artificial intelligence diagnostic models rely heavily on high-quality and exhaustively annotated data for algorithm training, but they are affected by noise label information. To enhance the robustness of the model and prevent memorization of noise labels, this study proposes a noise label image classification method based on multi-dimensional contrastive learning. This method can effectively integrate multi-dimensional contrastive learning and semi-supervised learning to combat label noise. Specifically, the proposed method consists of three carefully designed components. A mixed feature embedding module with a momentum update mechanism is designed to extract abstract distributed feature representations using mixed augmented images as input. Simultaneously, the study adjusts the features in the feature space from different dimensions by employing a multi-dimensional contrastive learning module, which combines instance contrastive learning and inter-class contrastive learning. Additionally, a noise-robust loss function is utilized to ensure that samples with correct labels dominate the learning process. Experiments conducted on CIFAR-10 and CIFAR-100 datasets demonstrate that the proposed method achieves better results than existing methods.

    Reference
    Related
    Cited by
Get Citation

朱定局,叶展昊,何珂仪.基于半监督多维度对比学习的噪声标签图像分类.计算机系统应用,,():1-8

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:October 17,2024
  • Revised:October 30,2024
  • Online: February 26,2025
Article QR Code
You are the firstVisitors
Copyright: Institute of Software, Chinese Academy of Sciences Beijing ICP No. 05046678-3
Address:4# South Fourth Street, Zhongguancun,Haidian, Beijing,Postal Code:100190
Phone:010-62661041 Fax: Email:csa (a) iscas.ac.cn
Technical Support:Beijing Qinyun Technology Development Co., Ltd.

Beijing Public Network Security No. 11040202500063