Abstract:The practical successes of deep neural networks have not been matched by theoretical progress that satisfyingly explains their behavior. In this work, we study the information bottleneck (IB) theory of deep learning, which makes three specific claims: first, that deep networks undergo two distinct phases consisting of an initial fitting phase and a subsequent compression phase; second, that the compression phase is causally related to the excellent generalization performance of deep networks; and third, that the compression phase occurs due to the diffusion-like behavior of stochastic gradient descent. Here we show that none of these claims hold true in the general case, and instead reflect assumptions made to compute a finite mutual information metric in deterministic networks. When computed using simple binning, we demonstrate through a combination of analytical results and simulation that the information plane trajectory observed in prior work is predominantly a function of the neural nonlinearity employed: double-sided saturating nonlinearities like yield a compression phase as neural activations enter the saturation regime, but linear activation functions and single-sided saturating nonlinearities like the widely used ReLU in fact do not. Moreover, we find that there is no evident causal connection between compression and generalization: networks that do not compress are still capable of generalization, and vice versa. Next, we show that the compression phase, when it exists, does not arise from stochasticity in training by demonstrating that we can replicate the IB findings using full batch gradient descent rather than stochastic gradient descent. Finally, we show that when an input domain consists of a subset of task-relevant and task-irrelevant information, hidden representations do compress the task-irrelevant information, although the overall information about the input may monotonically increase with training time, and that this compression happens concurrently with the fitting process rather than during a subsequent compression period.

Information Bottleneck and Aggregated Learning

Aggregated Learning: A Vector-Quantization Approach to Learning Neural Network Classifiers

Aggregated Learning: A Deep Learning Framework Based on Information-Bottleneck Vector Quantization

Layer-wise Learning of Stochastic Neural Networks with Information Bottleneck

Elastic Information Bottleneck

Information Bottleneck Methods for Distributed Learning

Information Bottleneck Analysis of Deep Neural Networks via Lossy Compression

Generalized Information Bottleneck for Gaussian Variables

Information Bottleneck: Theory and Applications in Deep Learning

Tighter Bounds on the Information Bottleneck with Application to Deep Learning

Deep Learning and the Information Bottleneck Principle

Recognizable Information Bottleneck

Disentangled Representation Learning with Transmitted Information Bottleneck

Information Bottleneck Revisited: Posterior Probability Perspective with Optimal Transport

Class-Conditional Compression and Disentanglement: Bridging the Gap between Neural Networks and Naive Bayes Classifiers

IB-UQ: Information bottleneck based uncertainty quantification for neural function regression and neural operator learning

Opportunistic Information-Bottleneck for Goal-oriented Feature Extraction and Communication

An Achievable and Analytic Solution to Information Bottleneck for Gaussian Mixtures

On the Information Bottleneck Problems: Models, Connections, Applications and Information Theoretic Views

Applying the Information Bottleneck Principle to Prosodic Representation Learning

On the information bottleneck theory of deep learning