Journal
NEURAL NETWORKS
Volume 135, Issue -, Pages 105-114Publisher
PERGAMON-ELSEVIER SCIENCE LTD
DOI: 10.1016/j.neunet.2020.12.011
Keywords
Predictive uncertainty; Neural networks; Deep learning; Uncertainty quantification; Dirichlet
Funding
- U.S. Air Force [FA8702-15-D-0001]
Ask authors/readers for more resources
This paper introduces the Information Aware Dirichlet networks method to improve uncertainty estimation in neural network predictions by learning a Dirichlet prior distribution. Experimental results demonstrate that this method significantly outperforms existing neural network techniques in accuracy and detecting adversarial examples.
Precise estimation of uncertainty in predictions for AI systems is a critical factor in ensuring trust and safety. Deep neural networks trained with a conventional method are prone to over-confident predictions. In contrast to Bayesian neural networks that learn approximate distributions on weights to infer prediction confidence, we propose a novel method, Information Aware Dirichlet networks, that learn an explicit Dirichlet prior distribution on predictive distributions by minimizing a bound on the expected max norm of the prediction error and penalizing information associated with incorrect outcomes. Properties of the new cost function are derived to indicate how improved uncertainty estimation is achieved. Experiments using real datasets show that our technique outperforms, by a large margin, state-of-the-art neural networks for estimating within-distribution and out-of-distribution uncertainty, and detecting adversarial examples. (c) 2020 Elsevier Ltd. All rights reserved.
Authors
I am an author on this paper
Click your name to claim this paper and add it to your profile.
Reviews
Recommended
No Data Available