Hey guys, is it possible to make a Variational AutoEncoder where the Encoder, instead of generating as output the mean and the log variance of a distribution(which will be the latent vector), instead generates the standard deviation directly?
I've been studying VAEs and I'm being quite troubled by the fact that I can't manage to make a model that works properly, so I'm trying to review some things such as the KL-Divergence for the Encoder loss, the Gaussian Likelihood for Decoder(not MSE), and my Encoder output...
For now, I've been making the Encoder output the standard deviation(or, at least, I guess I'm making it output the standard deviation) besides the mean, I've been using KL-Divergence loss for the Encoder - though I've seen that KL Divergence also has a "closed form" which is useful for multivariate dimensions - and Gaussian Likelihood for the Decoder instead of MSE(because I find the idea around Likelihood more mathmatically correct and interesting)



but thanks




