Toward Architecture-Agnostic Local Control of Posterior Collapse in VAEs

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Song, Hyunsoo, Kim, Seungwhan, Lee, Seungkyu
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909740064309248
author Song, Hyunsoo
Kim, Seungwhan
Lee, Seungkyu
author_facet Song, Hyunsoo
Kim, Seungwhan
Lee, Seungkyu
contents Variational autoencoders (VAEs), one of the most widely used generative models, are known to suffer from posterior collapse, a phenomenon that reduces the diversity of generated samples. To avoid posterior collapse, many prior works have tried to control the influence of regularization loss. However, the trade-off between reconstruction and regularization is not satisfactory. For this reason, several methods have been proposed to guarantee latent identifiability, which is the key to avoiding posterior collapse. However, they require structural constraints on the network architecture. For further clarification, we define local posterior collapse to reflect the importance of individual sample points in the data space and to relax the network constraint. Then, we propose Latent Reconstruction(LR) loss, which is inspired by mathematical properties of injective and composite functions, to control posterior collapse without restriction to a specific architecture. We experimentally evaluate our approach, which controls posterior collapse on varied datasets such as MNIST, fashionMNIST, Omniglot, CelebA, and FFHQ.
format Preprint
id arxiv_https___arxiv_org_abs_2508_12530
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Toward Architecture-Agnostic Local Control of Posterior Collapse in VAEs
Song, Hyunsoo
Kim, Seungwhan
Lee, Seungkyu
Machine Learning
Computer Vision and Pattern Recognition
I.2.6
Variational autoencoders (VAEs), one of the most widely used generative models, are known to suffer from posterior collapse, a phenomenon that reduces the diversity of generated samples. To avoid posterior collapse, many prior works have tried to control the influence of regularization loss. However, the trade-off between reconstruction and regularization is not satisfactory. For this reason, several methods have been proposed to guarantee latent identifiability, which is the key to avoiding posterior collapse. However, they require structural constraints on the network architecture. For further clarification, we define local posterior collapse to reflect the importance of individual sample points in the data space and to relax the network constraint. Then, we propose Latent Reconstruction(LR) loss, which is inspired by mathematical properties of injective and composite functions, to control posterior collapse without restriction to a specific architecture. We experimentally evaluate our approach, which controls posterior collapse on varied datasets such as MNIST, fashionMNIST, Omniglot, CelebA, and FFHQ.
title Toward Architecture-Agnostic Local Control of Posterior Collapse in VAEs
topic Machine Learning
Computer Vision and Pattern Recognition
I.2.6
url https://arxiv.org/abs/2508.12530