Uncertainty Quantification in CNN Through the Bootstrap of Convex Neural Networks

Abstract

Despite the popularity of Convolutional Neural Networks (CNN), the problem of uncertainty quantification (UQ) of CNN has been largely overlooked. Lack of efficient UQ tools severely limits the application of CNN in certain areas, such as medicine, where prediction uncertainty is critically important. Among the few existing UQ approaches that have been proposed for deep learning, none of them has theoretical consistency that can guarantee the uncertainty quality. To address this issue, we propose a novel bootstrap based framework for the estimation of prediction uncertainty. The inference procedure we use relies on convexified neural networks to establish the theoretical consistency of bootstrap. Our approach has a significantly less computational load than its competitors, as it relies on warm-starts at each bootstrap that avoids refitting the model from scratch. We further explore a novel transfer learning method so our framework can work on arbitrary neural networks. We experimentally demonstrate our approach has a much better performance compared to other baseline CNNs and state-of-the-art methods on various image datasets.

9 pages, 1 figure. Accepted at AAAI 2021

References in corpus (10)

• Very Deep Convolutional Networks for Large-Scale Image Recognition
• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning
• Weight Uncertainty in Neural Networks
• Understanding deep learning requires rethinking generalization
• Simple and Scalable Predictive Uncertainty Estimation using Deep Ensembles
• High-Quality Prediction Intervals for Deep Learning: A Distribution-Free, Ensembled Approach
• Randomized Prior Functions for Deep Reinforcement Learning
• Generalized bootstrap for estimating equations
• Convexified Convolutional Neural Networks