Efficient Variational Inference for Sparse Deep Learning with Theoretical Guarantee

arXiv2011.07439MaRDI QIDQ6353756

Author name not available (Why is that?)

Publication date: 14 November 2020

Abstract: Sparse deep learning aims to address the challenge of huge storage consumption by deep neural networks, and to recover the sparse structure of target functions. Although tremendous empirical successes have been achieved, most sparse deep learning algorithms are lacking of theoretical support. On the other hand, another line of works have proposed theoretical frameworks that are computationally infeasible. In this paper, we train sparse deep neural networks with a fully Bayesian treatment under spike-and-slab priors, and develop a set of computationally efficient variational inferences via continuous relaxation of Bernoulli distribution. The variational posterior contraction rate is provided, which justifies the consistency of the proposed variational Bayes method. Notably, our empirical results demonstrate that this variational procedure provides uncertainty quantification in terms of Bayesian predictive distribution and is also capable to accomplish consistent variable selection by training a sparse multi-layer neural network.

Has companion code repository: https://github.com/JinchengBai/sparse-variational-bnn

This page was built for publication: Efficient Variational Inference for Sparse Deep Learning with Theoretical Guarantee

Report a bug (only for logged in users!)Click here to report a bug for this page (MaRDI item Q6353756)