English

Variational Encoder-based Reliable Classification

Machine Learning 2020-10-20 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Machine learning models provide statistically impressive results which might be individually unreliable. To provide reliability, we propose an Epistemic Classifier (EC) that can provide justification of its belief using support from the training dataset as well as quality of reconstruction. Our approach is based on modified variational auto-encoders that can identify a semantically meaningful low-dimensional space where perceptually similar instances are close in 2\ell_2-distance too. Our results demonstrate improved reliability of predictions and robust identification of samples with adversarial attacks as compared to baseline of softmax-based thresholding.

Keywords

Cite

@article{arxiv.2002.08289,
  title  = {Variational Encoder-based Reliable Classification},
  author = {Chitresh Bhushan and Zhaoyuan Yang and Nurali Virani and Naresh Iyer},
  journal= {arXiv preprint arXiv:2002.08289},
  year   = {2020}
}

Comments

Published in ICIP 2020. Typos fixed in revision