Guided Variational Autoencoder for Disentanglement Learning
Guided Variational Autoencoder for Disentanglement Learning
复制标题
DOI:
10.1109/cvpr42600.2020.00794
复制
发表时间:
2020-04
期刊:
影响因子:
--
通讯作者:
Zheng Ding;Yifan Xu;Weijian Xu;Gaurav Parmar;Yang Yang-Yang;M. Welling;Z. Tu
中科院分区:
文献类型:
--
作者:
Zheng Ding;Yifan Xu;Weijian Xu;Gaurav Parmar;Yang Yang-Yang;M. Welling;Z. Tu
We propose an algorithm, guided variational autoencoder (Guided-VAE), that is able to learn a controllable generative model by performing latent representation disentanglement learning. The learning objective is achieved by providing signal to the latent encoding/embedding in VAE without changing its main backbone architecture, hence retaining the desirable properties of the VAE. We design an unsupervised and a supervised strategy in Guided-VAE and observe enhanced modeling and controlling capability over the vanilla VAE. In the unsupervised strategy, we guide the VAE learning by introducing a lightweight decoder that learns latent geometric transformation and principal components; in the supervised strategy, we use an adversarial excitation and inhibition mechanism to encourage the disentanglement of the latent variables. Guided-VAE enjoys its transparency and simplicity for the general representation learning task, as well as disentanglement learning. On a number of experiments for representation learning, improved synthesis/sampling, better disentanglement for classification, and reduced classification errors in meta learning have been observed.