Alert button

Out of Distribution Data Detection Using Dropout Bayesian Neural Networks

Feb 18, 2022
Andre T. Nguyen, Fred Lu, Gary Lopez Munoz, Edward Raff, Charles Nicholas, James Holt

Figure 1 for Out of Distribution Data Detection Using Dropout Bayesian Neural Networks
Figure 2 for Out of Distribution Data Detection Using Dropout Bayesian Neural Networks
Figure 3 for Out of Distribution Data Detection Using Dropout Bayesian Neural Networks
Figure 4 for Out of Distribution Data Detection Using Dropout Bayesian Neural Networks

Share this with someone who'll enjoy it:

We explore the utility of information contained within a dropout based Bayesian neural network (BNN) for the task of detecting out of distribution (OOD) data. We first show how previous attempts to leverage the randomized embeddings induced by the intermediate layers of a dropout BNN can fail due to the distance metric used. We introduce an alternative approach to measuring embedding uncertainty, justify its use theoretically, and demonstrate how incorporating embedding uncertainty improves OOD data identification across three tasks: image classification, language classification, and malware detection.

View paper onarxiv icon

Share this with someone who'll enjoy it: