Self-paced Resistance Learning against Overfitting on Noisy Labels

TL;DR

This paper introduces a self-paced resistance learning framework that leverages CNN memorization effects to effectively combat overfitting caused by noisy labels without requiring clean validation data.

Contribution

It proposes a novel curriculum and resistance loss integrated into a unified training framework to improve robustness against noisy labels in CNN training.

Findings

Outperforms recent state-of-the-art methods on noisy-label datasets.

Effectively resists overfitting on corrupted labels.

Demonstrates robustness without clean validation data.

Abstract

Noisy labels composed of correct and corrupted ones are pervasive in practice. They might significantly deteriorate the performance of convolutional neural networks (CNNs), because CNNs are easily overfitted on corrupted labels. To address this issue, inspired by an observation, deep neural networks might first memorize the probably correct-label data and then corrupt-label samples, we propose a novel yet simple self-paced resistance framework to resist corrupted labels, without using any clean validation data. The proposed framework first utilizes the memorization effect of CNNs to learn a curriculum, which contains confident samples and provides meaningful supervision for other training samples. Then it adopts selected confident samples and a proposed resistance loss to update model parameters; the resistance loss tends to smooth model parameters' update or attain equivalent prediction over each class, thereby resisting model overfitting on corrupted labels. Finally, we unify these two modules into a single loss function and optimize it in an alternative learning. Extensive experiments demonstrate the significantly superior performance of the proposed framework over recent state-of-the-art methods on noisy-label data. Source codes of the proposed method are available on https://github.com/xsshi2015/Self-paced-Resistance-Learning.