Guided Variational Autoencoder for Disentanglement Learning

By Zheng Ding, Yifan Xu, Weijian Xu, Gaurav Parmar, Yang Yang, Max Welling, Zhuowen Tu. 2020

126 citations recorded by the source corpus.

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.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Generative Models, Representation & Self-Supervised Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Kwonjoon Lee
  2. Wangmeng Zuo
  3. Changyou Chen
  4. Gaurav Parmar
  5. Zhuowen Tu
  6. Xiao Liu
  7. Dacheng Li
  8. Yuan Gao

In Computer Vision and Pattern Recognition

  1. Wangmeng Zuo
  2. Luc Van Gool
  3. Changyou Chen
  4. Ming-Hsuan Yang
  5. Kwonjoon Lee
  6. Zhuowen Tu
  7. Lei Zhang
  8. Dacheng Tao

In Generative Models

  1. Kwonjoon Lee
  2. Gaurav Parmar
  3. Zhuowen Tu
  4. Zhilong Ji
  5. Wangmeng Zuo
  6. Changyou Chen
  7. Yuxiang Wei
  8. Yang Zhao

In Representation & Self-Supervised Learning

  1. Zhuowen Tu
  2. Wangmeng Zuo
  3. Changyou Chen
  4. Gaurav Parmar
  5. Dacheng Li
  6. Yang Zhao
  7. Kwonjoon Lee

Citations in the corpus