Solving the Reconstruction-Generation Trade-Off: Generative Model with Implicit Embedding Learning

Cong Geng,Jia Wang,Li Chen,Zhiyong Gao
DOI: https://doi.org/10.1016/j.neucom.2023.126428
IF: 6
2023-01-01
Neurocomputing
Abstract:Variational Autoencoder (VAE) and Generative adversarial network (GAN) are two classic generative models that generate realistic data from a predefined prior distribution, such as a Gaussian distribution. One advantage of VAE over GAN is its ability to simultaneously generate high-dimensional data and learn latent representations that are useful for data manipulation. However, it has been observed that a trade-off exists between reconstruction and generation in VAE, as matching the prior distribution for the latent representations may destroy the geometric structure of the data manifold. To address this issue, we propose an autoencoder-based generative model that allows the prior to learn the embedding distribution, rather than imposing the latent variables to fit the prior. To preserve the geometric structure of the data manifold to the maximum, the embedding distribution is trained using a simple regularized autoencoder architecture. Then an adversarial strategy is employed to achieve a latent mapping. We provide both theoretical and experimental support for the effectiveness of our method, which eliminates the contradiction between preserving the geometric structure of the data manifold and matching the distribution in latent space. Once this paper is accepted, the code will be released on https://github.com/gengcong940126/GMIEL .
What problem does this paper attempt to address?