arXiv · 2212.12667
Visualizing Information Bottleneck through Variational Inference
Abstract
The Information Bottleneck theory provides a theoretical and computational framework for finding approximate minimum sufficient statistics. Analysis of the Stochastic Gradient Descent (SGD) training of a neural network on a toy problem has shown the existence of two phases, fitting and compression. In this work, we analyze the SGD training process of a Deep Neural Network on MNIST classification and confirm the existence of two phases of SGD training. We also propose a setup for estimating the mutual information for a Deep Neural Network through Variational Inference.
Explore related subjects
Keep this discovery
Cipta Herwana, Abhishek Kadian. 2022-12-24. Visualizing Information Bottleneck through Variational Inference. https://arxiv.org/abs/2212.12667
Cite the original work for its findings. Save a collection to share your selection of sources.