arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.13279cs.CV

UQGAN: A Unified Model for Uncertainty Quantification of Deep Classifiers trained via Conditional GANs

TU Dortmund University(多特蒙德工业大学);EPFL(洛桑联邦理工学院);University of Wuppertal(伍珀塔尔大学)

Philipp Oberdiek, Gernot A. Fink, Matthias Rottmann

英文摘要:

We present an approach to quantifying both aleatoric and epistemic uncertainty for deep neural networks in image classification, based on generative adversarial networks (GANs). While most works in the literature that use GANs to generate out-of-distribution (OoD) examples only focus on the evaluation of OoD detection, we present a GAN based approach to learn a classifier that produces proper uncertainties for OoD examples as well as for false positives (FPs). Instead of shielding the entire in-distribution data with GAN generated OoD examples which is state-of-the-art, we shield each class separately with out-of-class examples generated by a conditional GAN and complement this with a one-vs-all image classifier. In our experiments, in particular on CIFAR10, CIFAR100 and Tiny ImageNet, we improve over the OoD detection and FP detection performance of state-of-the-art GAN-training based classifiers. Furthermore, we also find that the generated GAN examples do not significantly affect the calibration error of our classifier and result in a significant gain in model accuracy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑