Semi-Supervised Adversarial Variational Autoencoder - Archive ouverte HAL Access content directly
Journal Articles Machine Learning and Knowledge Extraction Year : 2020

Semi-Supervised Adversarial Variational Autoencoder

(1)
1

Abstract

We present a method to improve the reconstruction and generation performance of a variational autoencoder (VAE) by injecting an adversarial learning. Instead of comparing the reconstructed with the original data to calculate the reconstruction loss, we use a consistency principle for deep features. The main contributions are threefold. Firstly, our approach perfectly combines the two models, i.e., GAN and VAE, and thus improves the generation and reconstruction performance of the VAE. Secondly, the VAE training is done in two steps, which allows to dissociate the constraints used for the construction of the latent space on the one hand, and those used for the training of 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 third contribution is to use the trained encoder for the consistency principle for deep features extracted from the hidden layers. We present experimental results to show that our method gives better performance than the original VAE. The results demonstrate that the adversarial constraints allow the decoder to generate images that are more authentic and realistic than the conventional VAE.
Fichier principal
Vignette du fichier
make-02-00020-v2.pdf (6.42 Mo) Télécharger le fichier
Origin : Publisher files allowed on an open archive
Loading...

Dates and versions

hal-02931571 , version 1 (21-09-2020)

Licence

Attribution - CC BY 4.0

Identifiers

Cite

Ryad Zemouri. Semi-Supervised Adversarial Variational Autoencoder. Machine Learning and Knowledge Extraction, 2020, 2 (3), pp.361-378. ⟨10.3390/make2030020⟩. ⟨hal-02931571⟩
97 View
376 Download

Altmetric

Share

Gmail Facebook Twitter LinkedIn More