Deep Deterministic Information Bottleneck with Matrix-based Entropy Functional

Abstract

We introduce the matrix-based Renyi's $\alpha$-order entropy functional to parameterize Tishby et al. information bottleneck (IB) principle with a neural network. We term our methodology Deep Deterministic Information Bottleneck (DIB), as it avoids variational inference and distribution assumption. We show that deep neural networks trained with DIB outperform the variational objective counterpart and those that are trained with other forms of regularization, in terms of generalization performance and robustness to adversarial attack.Code available at https://github.com/yuxi120407/DIB

arXiv commentAccepted at ICASSP-21. Code available at https://github.com/yuxi120407/DIB. Extended version of the suppelementary material in "Measuring the Dependence with Matrix-based Entropy Functional", AAAI-21, arXiv:2101.10160

References in corpus (8)

• PyTorch: An Imperative Style, High-Performance Deep Learning Library
• Auto-Encoding Variational Bayes
• Learning deep representations by mutual information estimation and maximization
• Opening the Black Box of Deep Neural Networks via Information
• Regularizing Neural Networks by Penalizing Confident Output Distributions
• Understanding disentangling in $\beta$-VAE
• Learning Representations for Neural Network-Based Classification Using the Information Bottleneck Principle
• Interpretable Fault Detection using Projections of Mutual Information Matrix

Cited by in corpus (1)

• Understanding Neural Networks with Logarithm Determinant Entropy Estimator