English

InfoVAEGAN : learning joint interpretable representations by information maximization and maximum likelihood

Machine Learning 2021-07-13 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Learning disentangled and interpretable representations is an important step towards accomplishing comprehensive data representations on the manifold. In this paper, we propose a novel representation learning algorithm which combines the inference abilities of Variational Autoencoders (VAE) with the generalization capability of Generative Adversarial Networks (GAN). The proposed model, called InfoVAEGAN, consists of three networks~: Encoder, Generator and Discriminator. InfoVAEGAN aims to jointly learn discrete and continuous interpretable representations in an unsupervised manner by using two different data-free log-likelihood functions onto the variables sampled from the generator's distribution. We propose a two-stage algorithm for optimizing the inference network separately from the generator training. Moreover, we enforce the learning of interpretable representations through the maximization of the mutual information between the existing latent variables and those created through generative and inference processes.

Keywords

Cite

@article{arxiv.2107.04705,
  title  = {InfoVAEGAN : learning joint interpretable representations by information maximization and maximum likelihood},
  author = {Fei Ye and Adrian G. Bors},
  journal= {arXiv preprint arXiv:2107.04705},
  year   = {2021}
}

Comments

Accepted at International Conference on Image Processing (ICIP 2021)

R2 v1 2026-06-24T04:03:35.684Z