Information-theoretical label embeddings for large-scale image classification

By François Chollet. 2016

17 citations recorded by the source corpus.

We present a method for training multi-label, massively multi-class image classification models, that is faster and more accurate than supervision via a sigmoid cross-entropy loss (logistic regression). Our method consists in embedding high-dimensional sparse labels onto a lower-dimensional dense sphere of unit-normed vectors, and treating the classification problem as a cosine proximity regression problem on this sphere. We test our method on a dataset of 300 million high-resolution images with 17,000 labels, where it yields considerably faster convergence, as well as a 7% higher mean average precision compared to logistic regression.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Information Theory

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Jianwei Yang
  3. Lu Yuan
  4. Pengchuan Zhang
  5. Bin Xiao
  6. Ce Liu
  7. Serge J. Belongie
  8. Mohit Bansal

In Computer Vision and Pattern Recognition

  1. Jianfeng Gao
  2. Luc Van Gool
  3. Dacheng Tao
  4. Lu Yuan
  5. Jianwei Yang
  6. Trevor Darrell
  7. Xiaogang Wang
  8. Ming-Hsuan Yang

In Information Theory

  1. Jianwei Yang
  2. Lu Yuan
  3. Ce Liu
  4. Maximilian Nickel
  5. Jianfeng Gao
  6. L. V. D. van der Maaten
  7. Bin Xiao
  8. Andreas Veit

Citations in the corpus