Beyond Categorical Label Representations for Image Classification

Abstract

We find that the way we choose to represent data labels can have a profound effect on the quality of trained models. For example, training an image classifier to regress audio labels rather than traditional categorical probabilities produces a more reliable classification. This result is surprising, considering that audio labels are more complex than simpler numerical probabilities or text. We hypothesize that high dimensional, high entropy label representations are generally more useful because they provide a stronger error signal. We support this hypothesis with evidence from various label representations including constant matrices, spectrograms, shuffled spectrograms, Gaussian mixtures, and uniform random matrices of various dimensionalities. Our experiments reveal that high dimensional, high entropy labels achieve comparable accuracy to text (categorical) labels on the standard image classification task, but features learned through our label representations exhibit more robustness under various adversarial attacks and better effectiveness with a limited amount of training data. These results suggest that label representation may play a more important role than previously thought. The project website is at \url{https://www.creativemachineslab.com/label-representation.html}.

arXiv commentInternational Conference on Learning Representations (ICLR 2021). Project page is at \url{https://www.creativemachineslab.com/label-representation.html}

References in corpus (17)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Distilling the Knowledge in a Neural Network
• Explaining and Harnessing Adversarial Examples
• Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
• Intriguing properties of neural networks
• Deep Residual Learning for Image Recognition
• Theoretical Models of Learning to Learn
• Meta-SGD: Learning to Learn Quickly for Few-Shot Learning
• The Open Images Dataset V4: Unified image classification, object detection, and visual relationship detection at scale
• Understanding Adversarial Training: Increasing Local Stability of Neural Nets through Robust Optimization
• One-shot Learning with Memory-Augmented Neural Networks
• PixelDefend: Leveraging Generative Models to Understand and Defend against Adversarial Examples
• Meta-Learning: A Survey
• Blocking Transferability of Adversarial Examples in Black-Box Learning Systems
• Adversarial Robustness via Label-Smoothing
• CNN depth analysis with different channel inputs for Acoustic Scene Classification