Out-of-Distribution Detection from Small Training Sets using Bayesian Neural Network Classifiers
By: Kevin Raina, Tanya Schmah
Potential Business Impact:
Helps AI know when it's unsure.
Out-of-Distribution (OOD) detection is critical to AI reliability and safety, yet in many practical settings, only a limited amount of training data is available. Bayesian Neural Networks (BNNs) are a promising class of model on which to base OOD detection, because they explicitly represent epistemic (i.e. model) uncertainty. In the small training data regime, BNNs are especially valuable because they can incorporate prior model information. We introduce a new family of Bayesian posthoc OOD scores based on expected logit vectors, and compare 5 Bayesian and 4 deterministic posthoc OOD scores. Experiments on MNIST and CIFAR-10 In-Distributions, with 5000 training samples or less, show that the Bayesian methods outperform corresponding deterministic methods.
Similar Papers
EOOD: Entropy-based Out-of-distribution Detection
CV and Pattern Recognition
Helps computers know when they see something new.
Improving Out-of-Distribution Detection with Markov Logic Networks
Machine Learning (CS)
Helps computers spot fake or wrong information.
Revisiting Likelihood-Based Out-of-Distribution Detection by Modeling Representations
Machine Learning (CS)
Makes AI better at spotting fake or wrong information.