optimizer and numerical stability
[D] Optimizers and numerical stability for Bayesian deep nets
I've been experimenting with flipout layers with tensorflow and tensorflow probability, but I keep having numerical stability issues in late epochs using negative log likelihood losses with adam. It also seems like the network has a lot of difficulty with learning scale parameters, (e.g. the standard deviation if the probability model is normal) particularly when the n is smallest, and returns a lot of samples far, far outside the range of the training data.