Regularizing Neural Network Training via Identity-wise Discriminative Feature Suppression
arXiv:2209.14553 · doi:10.1109/DICTA56598.2022
Abstract
It is well-known that a deep neural network has a strong fitting capability and can easily achieve a low training error even with randomly assigned class labels. When the number of training samples is small, or the class labels are noisy, networks tend to memorize patterns specific to individual instances to minimize the training error. This leads to the issue of overfitting and poor generalisation performance. This paper explores a remedy by suppressing the network's tendency to rely on instance-specific patterns for empirical error minimisation. The proposed method is based on an adversarial training framework. It suppresses features that can be utilized to identify individual instances among samples within each class. This leads to classifiers only using features that are both discriminative across classes and common within each class. We call our method Adversarial Suppression of Identity Features (ASIF), and demonstrate the usefulness of this technique in boosting generalisation accuracy when faced with small datasets or noisy labels. Our source code is available.
arXiv commentDICTA 2022
References in corpus (4)
- Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
- Unsupervised Domain Adaptation by Backpropagation
- Generalized Cross Entropy Loss for Training Deep Neural Networks with Noisy Labels
- MentorNet: Learning Data-Driven Curriculum for Very Deep Neural Networks on Corrupted Labels