Uncertainty in the Variational Information Bottleneck
Adam: A Method for Stochastic Optimization