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.

The Role of Information Complexity and Randomization in Representation Learning

A Measure of the Complexity of Neural Representations based on Partial Information Decomposition

Representation Based Complexity Measures for Predicting Generalization in Deep Learning

Do Compressed Representations Generalize Better?

Understanding the Generalization Ability of Deep Learning Algorithms: A Kernelized Rényi’s Entropy Perspective

Understanding Encoder-Decoder Structures in Machine Learning Using Information Measures

Elastic Information Bottleneck

Understanding the Generalization Ability of Deep Learning Algorithms: A Kernelized Renyi's Entropy Perspective

Going Deeper, Generalizing Better: an Information-Theoretic View for Deep Learning.

Learned feature representations are biased by complexity, learning order, position, and more

Deep Representation Learning for Domain Generalization with Information Bottleneck Principle

On the information bottleneck theory of deep learning

Information Plane Analysis for Dropout Neural Networks

Dropout Training, Data-dependent Regularization, and Generalization Bounds.

Information-Theoretic Generalization Bounds for Deep Neural Networks

Entropy and mutual information in models of deep neural networks

Learning Representations in Reinforcement Learning:An Information Bottleneck Approach

Information bottleneck theory of high-dimensional regression: relevancy, efficiency and optimality

The Geometric Occam's Razor Implicit in Deep Learning

Information Bottleneck Theory Based Exploration of Cascade Learning

Understanding Dynamics of Nonlinear Representation Learning and Its Application