preprints.org > computer science and mathematics > artificial intelligence and machine learning > doi: 10.20944/preprints202008.0051.v1

Preprint Article Version 1 Preserved in Portico This version is not peer-reviewed

Version 1 : Received: 1 August 2020 / Approved: 2 August 2020 / Online: 2 August 2020 (18:11:27 CEST)

We present a method to improve the reconstruction and generation performance of variational autoencoder (VAE) by injecting an adversarial learning. On the other hand, instead of comparing the reconstructed with the original data to calculate the reconstruction loss, we use a consistency principle for deep features. The training process of the VAE is then divided into two steps, training the encoder and then training the decoder. By using this two-step learning process, our method can be more widely used in applications other than image processing. While training the encoder, the label information is integrated to better structure the latent space in a supervised way. The adversarial constraints allow the decoder to generate data with better authenticity and more realistic than the conventional VAE. We present experimental results to show that our method gives better performance than the original VAE.

Variational autoencoder; Adversarial learning; Deep feature consistent; Data generation

Computer Science and Mathematics, Artificial Intelligence and Machine Learning

* All users must log in before leaving a comment