SCNet: semi-supervised and contrastive learning against noisy labels with two selection strategy approach
Dingju Zhu,Zhanhao Ye,Keyi He,KaiLeung Yung,Andrew W. H. Ip
DOI: https://doi.org/10.1080/03081079.2024.2430349
2024-11-28
International Journal of General Systems
Abstract:The deep learning-driven artificial intelligence diagnostic model heavily relies on high-quality and detailed annotated data during algorithm training. However, this process is susceptible to interference from noisy label information. To enhance the model's robustness and prevent the adverse effects of noisy labels, we propose a contrastive learning-based approach for noisy label image classification. Specifically, we designed three components: a mixed feature embedding module that takes mixed augmented images as input, a momentum update mechanism to explore abstract distributed feature representations, and two selection strategies. We incorporated a flexible pseudo-label boosting strategy, refining supervised information for noisy data with pseudo-labels based on initial classification predictions. By measuring the similarity between classification distributions, we effectively select more reliable pairs of confidence, thereby reducing the impact of noisy labels on contrastive learning. Furthermore, we employed a noise-robust loss function to ensure that samples with correct labels dominate the learning process. To validate the model's effectiveness, we conducted experiments on our collected data SAID (Sensitive Audio Information Detection), the CIFAR-10 and CIFAR-100 datasets, simulating scenarios with noisy label conditions.
computer science, theory & methods