Psscl: A Progressive Sample Selection Framework with Contrastive Loss Designed for Noisy Labels

Authors:
Preprint
Publisher website
Google Scholar
Abstract
Large-scale image datasets frequently contain unavoidable noisy labels, resulting in overfitting in deep neural networks and declining performance. Most existing methods for learning from noisy labels operate as one-stage frameworks, where training data division and semi-supervised learning (SSL) are intertwined for optimization. Accordingly, their effectiveness is significantly influenced by the precision of the separated clean set, prior knowledge of noise, and the robustness of SSL. In this paper, we propose a progressive sample selection framework with contrastive loss for noisy labels named PSSCL. This framework operates in two stages, using robust and contrastive losses to augment the robustness of the model. The first stage focuses on identifying a small clean set through a long-term confidence detection strategy, while the second stage aims to enhance performance by expanding this clean set. PSSCL demonstrates significant improvement across various benchmarks when compared with state-of-the-art methods. The code will be available at https://github.com/LanXiaoPang613/PSSCL.

This publication has 21 references indexed in Scilit: