Uncertainty Estimation with a VAE-Classifier Hybrid Model

Shuyu Lin; Ronald Clark; Niki Trigoni; Stephen J. Roberts

ICASSP 2022accepted0 citations

Uncertainty Estimation with a VAE-Classifier Hybrid Model

Shuyu Lin, Ronald Clark, Niki Trigoni, Stephen J. Roberts

Abstract

We propose a hybrid model that combines a generative unit and a discriminative classifier to quantify uncertainty in a classification task. The representation learning capability in the VAE module allows our method to learn more useful and generalizable features and outperform other purely discriminative classifiers when training labels are limited. With proper statistical treatment, the probabilistic encoder in our VAE module offers a convenient mechanism to express uncertainty for out-of-distribution (OOD) data. As a result, our method gives better calibrated uncertainty prediction. We demonstrate the effectiveness of our method on MNIST and a challenging medical image dataset for skin lesion diagnosis.

BibTeX

@inproceedings{icassp2022_uncertaintyestim,
  title = {Uncertainty Estimation with a VAE-Classifier Hybrid Model},
  author = {Shuyu Lin and Ronald Clark and Niki Trigoni and Stephen J. Roberts},
  booktitle = {ICASSP 2022},
  year = {2022}
}