Variational Auto-Encoder: not all failures are equal - Inria - Institut national de recherche en sciences et technologies du numérique
Preprints, Working Papers, ... Year : 2020

Variational Auto-Encoder: not all failures are equal

Victor Berger
Michèle Sebag
  • Function : Author
  • PersonId : 836537

Abstract

We claim that a source of severe failures for Variational Auto-Encoders is the choice of the distribution class used for the observation model. A first theoretical and experimental contribution of the paper is to establish that even in the large sample limit with arbitrarily powerful neural architectures and latent space, the VAE fails if the sharpness of the distribution class does not match the scale of the data. Our second claim is that the distribution sharpness must preferably be learned by the VAE (as opposed to, fixed and optimized offline): Autonomously adjusting this sharpness allows the VAE to dynamically control the trade-off between the optimization of the reconstruction loss and the latent compression. A second empirical contribution is to show how the control of this trade-off is instrumental in escaping poor local optima, akin a simulated annealing schedule. Both claims are backed upon experiments on artificial data, MNIST and CelebA, showing how sharpness learning addresses the notorious VAE blurriness issue.
Fichier principal
Vignette du fichier
main.pdf (3.64 Mo) Télécharger le fichier
Origin Files produced by the author(s)
Loading...

Dates and versions

hal-02497248 , version 1 (03-03-2020)

Identifiers

Cite

Victor Berger, Michèle Sebag. Variational Auto-Encoder: not all failures are equal. 2020. ⟨hal-02497248⟩
193 View
2287 Download

Altmetric

Share

More