C2AE: Class Conditioned Auto-Encoder for Open-Set Recognition

By Poojan Oza, Vishal M. Patel. Computer Vision and Pattern Recognition, 2019

412 citations recorded by the source corpus.

Models trained for classification often assume that all testing classes are known while training. As a result, when presented with an unknown class during testing, such closed-set assumption forces the model to classify it as one of the known classes. However, in a real world scenario, classification models are likely to encounter such examples. Hence, identifying those examples as unknown becomes critical to model performance. A potential solution to overcome this problem lies in a class of learning problems known as open-set recognition. It refers to the problem of identifying the unknown classes during testing, while maintaining performance on the known classes. In this paper, we propose an open-set recognition algorithm using class conditioned auto-encoders with novel training and testing methodologies. In this method, training procedure is divided in two sub-tasks, 1. closed-set classification and, 2. open-set identification (i.e. identifying a class as known or unknown). Encoder learns the first task following the closed-set classification training pipeline, whereas decoder learns the second task by reconstructing conditioned on class identity. Furthermore, we model reconstruction errors using the Extreme Value Theory of statistical modeling to find the threshold for identifying known/unknown class samples. Experiments performed on multiple image classification datasets show that the proposed method performs significantly better than the state of the art methods. The source code is available at: github.com/otkupjnoz/c2ae.

Subfields: Artificial Intelligence, Media Technology

View Paper: https://www.semanticscholar.org/paper/967d532a66dab7edcb818b0f9dc59fe8da7dc171

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. Han-Jia Ye
  2. De-chuan Zhan
  3. Ziwei Liu
  4. Da-Wei Zhou
  5. Andrew Zisserman
  6. Mohit Bansal
  7. Andrea Vedaldi
  8. Bing Liu

In Media Technology

  1. Andrea Vedaldi
  2. Deva Ramanan
  3. Vishal M. Patel
  4. Andrew Zisserman
  5. Ziwei Liu
  6. Shu Kong
  7. Kai Han
  8. Yonghong Tian

Citations in the corpus