I know this paper from the references of Machine Theory of Mind, which uses a Deep Variational Information Bottleneck to compress the mental expression of an agent in order to get an expressive representation and a better visualization. Deep Variational Information Bottleneck can be categorized into a line of work where the information theory is used to regularize the embedding. VAE is one of the most classical examples in this field.
Suppose we embed the input source to an encoding , usually defined by a parameteric model/encoder . In the supervision task, we usually want to maximize the mutual information between the encoding and the label so that we can use that informative encoding to classify. The target can be express as
Asking the encoding to be informative is not general enough in many cases. We can add some restrictions to limit its complexity. This paper uses the mutual information between the encoding and the original data as a regularization term. The final optimization (maximization) objective is
To optimize this objective, few tricks are applied:
-  Since and are intractable, and are introduced to approximate them respectively. And therefore, the final optimization goal is a lower bound of the original one.
-  The encoder might take a stochastic form like the Gaussian distribution . They used “the reparameterization trick” in Auto-encoding variational Bayes to replace with where .
Find a lower bound with :
The empirical estimation:
Apply  to (16) to allow gradient to back-propagate through a single sample:
[DR023] Take away:
- Embedding an input as a distribution is more robust to adversarial attacks than deterministic models.
- Once some probabilities are intractable, use other functions to approximate and use the property of KL-divergence (greater than 0) to find the bound.
- Reparameterization trick allows us to use Monte Carlo sampling to get an unbiased estimate of the gradient.