Learning Conditionally Untangled Latent Spaces using Fixed Point Iteration - Archive ouverte HAL
Communication Dans Un Congrès Année : 2024

Learning Conditionally Untangled Latent Spaces using Fixed Point Iteration

Victor Enescu
  • Fonction : Auteur
  • PersonId : 1436429
Hichem Sahbi

Résumé

Normalizing flows (NFs) are powerful generative models that map arbitrary complex (ambient) distributions to simple (latent) ones such as the monomodal gaussian. Despite their ability in modeling and sampling highly nonlinear manifolds, NFs are less effective in assigning labels to the generated data. This stems from the insufficient expressivity of monomodal gaussians, and also the difficulty in learning multimodal distributions in the latent spaces. In this paper, we devise a multimodal NF-based approach suitable both for image generation and classification. The particularity of our method resides in its ability to design multimodal gaussian distributions as a part of NF training using an objective function that mixes a likelihood term and a Kullback-Leibler Divergence (KLD) criterion. The parameters of the trained gaussians (namely means and covariance matrices) are obtained as an interpretable fixed-point solution of this objective function. Besides, our proposed method avoids the overwhelming and sensitive process of tuning the learning rates as required by gradient descent. Extensive experiments conducted on different datasets, including CIFAR10, CIFAR100 and ImageNet, show competitive performances of our method against different baselines as well as the related work.
Fichier principal
Vignette du fichier
paper.pdf (2.52 Mo) Télécharger le fichier
Origine Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-04767310 , version 1 (05-11-2024)

Identifiants

  • HAL Id : hal-04767310 , version 1

Citer

Victor Enescu, Hichem Sahbi. Learning Conditionally Untangled Latent Spaces using Fixed Point Iteration. The British Machine Vision Conference (BMVC), Nov 2024, Glasgow, United Kingdom. ⟨hal-04767310⟩
0 Consultations
0 Téléchargements

Partager

More