vix.ing · top · new · best · stats

Deep Variational Information Bottleneck

2016/12/31 by Alexander A. Alemi, Ian Fischer, Joshua V. Dillon +1 · 15 citations
Computer Science · Mathematics · #cs.LG #cs.IT #math.IT

paper · pdf

published as Proceedings of the International Conference on Learning Representations (ICLR) 2017 · 19 pages, 8 figures, Accepted to ICLR17

arxiv created 2019/10/23 · arxiv updated 2019/10/25

Abstract

We present a variational approximation to the information bottleneck of Tishby et al. (1999). This variational approach allows us to parameterize the information bottleneck model using a neural network and leverage the reparameterization trick for efficient training. We call this method "Deep Variational Information Bottleneck", or Deep VIB. We show that models trained with the VIB objective outperform those that are trained with other forms of regularization, in terms of generalization performance and robustness to adversarial attack.

Cited by

Related