Semi-Supervised Adversarial Variational Autoencoder - Cnam - Conservatoire national des arts et métiers
Article Dans Une Revue Machine Learning and Knowledge Extraction Année : 2020

Semi-Supervised Adversarial Variational Autoencoder

Résumé

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
Origine Fichiers éditeurs autorisés sur une archive ouverte
Loading...

Dates et versions

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

Licence

Identifiants

Citer

Ryad Zemouri. Semi-Supervised Adversarial Variational Autoencoder. Machine Learning and Knowledge Extraction, 2020, 2 (3), pp.361-378. ⟨10.3390/make2030020⟩. ⟨hal-02931571⟩
140 Consultations
580 Téléchargements

Altmetric

Partager

More