arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.07514cs.NEcs.LG

Balancing reconstruction error and Kullback-Leibler divergence in Variational Autoencoders

University of Bologna(博洛尼亚大学);Department of Informatics: Science and Engineering (DISI)(信息学系:科学与工程(DISI))

Andrea Asperti, Matteo Trentin

英文摘要:

In the loss function of Variational Autoencoders there is a well known tension between two components: the reconstruction loss, improving the quality of the resulting images, and the Kullback-Leibler divergence, acting as a regularizer of the latent space. Correctly balancing these two components is a delicate issue, easily resulting in poor generative behaviours. In a recent work, Dai and Wipf obtained a sensible improvement by allowing the network to learn the balancing factor during training, according to a suitable loss function. In this article, we show that learning can be replaced by a simple deterministic computation, helping to understand the underlying mechanism, and resulting in a faster and more accurate behaviour. On typical datasets such as Cifar and Celeba, our technique sensibly outperforms all previous VAE architectures.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑