arXiv · 1902.02603
Radial and Directional Posteriors for Bayesian Neural Networks
Abstract
We propose a new variational family for Bayesian neural networks. We decompose the variational posterior into two components, where the radial component captures the strength of each neuron in terms of its magnitude; while the directional component captures the statistical dependencies among the weight parameters. The dependencies learned via the directional density provide better modeling performance compared to the widely-used Gaussian mean-field-type variational family. In addition, the strength of input and output neurons learned via the radial density provides a structured way to compress neural networks. Indeed, experiments show that our variational family improves predictive performance and yields compressed networks simultaneously.
Explore related subjects
Keep this discovery
Changyong Oh, Kamil Adamczewski, Mijung Park. 2019-02-07. Radial and Directional Posteriors for Bayesian Neural Networks. https://arxiv.org/abs/1902.02603
Cite the original work for its findings. Save a collection to share your selection of sources.