GCVAE: Generalized-Controllable Variational AutoEncoder - Archive ouverte HAL Access content directly
Preprints, Working Papers, ... Year :

GCVAE: Generalized-Controllable Variational AutoEncoder

(1, 2, 3, 4) , (5) , (2, 3, 4, 1) , (2, 6, 4)
1
2
3
4
5
6

Abstract

Variational autoencoders (VAEs) have recently been used for unsupervised disentanglement learning of complex density distributions. Numerous variants exist to encourage disentanglement in latent space while improving reconstruction. However, none have simultaneously managed the trade-off between attaining extremely low reconstruction error and a high disentanglement score. We present a generalized framework to handle this challenge under constrained optimization and demonstrate that it outperforms state-of-the-art existing models as regards disentanglement while balancing reconstruction. We introduce three controllable Lagrangian hyperparameters to control reconstruction loss, KL divergence loss and correlation measure. We prove that maximizing information in the reconstruction network is equivalent to information maximization during amortized inference under reasonable assumptions and constraint relaxation.
Fichier principal
Vignette du fichier
2206.04225.pdf (1.1 Mo) Télécharger le fichier
Origin : Files produced by the author(s)

Dates and versions

emse-03712594 , version 1 (04-07-2022)

Identifiers

  • HAL Id : emse-03712594 , version 1

Cite

Kenneth Ezukwoke, Anis Hoayek, Mireille Batton-Hubert, Xavier Boucher. GCVAE: Generalized-Controllable Variational AutoEncoder. 2022. ⟨emse-03712594⟩
82 View
15 Download

Share

Gmail Facebook Twitter LinkedIn More