Mutual Information Collapse Explains Disentanglement Failure in β-VAEs
Not on the map. Why?
Abstract (arXiv)
The β-VAE is a foundational framework for unsupervised disentanglement, using β to regulate the trade-off between latent factorization and reconstruction fidelity. Empirically, however, disentanglement performance exhibits a pervasive non-monotonic trend: benchmarks such as MIG and SAP typically peak at intermediate β and collapse as regularization increases. We demonstrate that this collapse is a fundamental information-theoretic failure, where strong Kullback-Leibler pressure promotes marginal independence at the expense of the latent channel's semantic informativeness. By formalizing this mechanism in a linear-Gaussian setting, we prove that for β > 1, stationarity-induced dynamics trigger a spectral contraction of the encoder gain, driving latent-factor mutual information to zero. To resolve this, we introduce the λβ-VAE, which decouples regularization pressure from informational collapse via an auxiliary L2 reconstruction penalty λ. Extensive experiments on dSprites, Shapes3D, and MPI3D-real confirm that λ > 0 stabilizes disentanglement and restores latent informativeness over a significantly broader range of β, providing a principled theoretical justification for dual-parameter regularization in variational inference backbones.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 35 references that are in the map. First posted on arXiv 9 Feb 2026.
Builds on
19 of 35 references.
References · 19
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 929
- 2018Disentangling by Factorisingcited by 614
- 2017Variational Inference of Disentangled Latent Concepts from Unlabeled Observationscited by 240
- 2021Towards Causal Representation Learningcited by 207
- 2019On the Transfer of Inductive Bias from Simulation to the Real World: a New Disentanglement Datasetcited by 100
- 2020Measuring Disentanglement: A Review of Metricscited by 69
- 2022SQ-VAE: Variational Bayes on Discrete Representation with Self-annealed Stochastic Quantizationcited by 68
- 2021Visual Representation Learning Does Not Generalize Strongly Within the Same Domaincited by 60
- 2022Posterior Collapse of a Linear Latent Variable Modelcited by 18
- 2019Scalarizing Functions in Bayesian Multiobjective Optimizationcited by 15
7 more
- 2021Preventing Oversmoothing in VAE via Generalized Variance Parameterizationcited by 15
- 2022Hyperbolic VAE via Latent Gaussian Distributionscited by 14
- 2023Learning Dynamics in Linear VAE: Posterior Collapse Threshold, Superfluous Latent Space Pitfalls, and Speedup with KL Annealingcited by 12
- 2021Beta-VAE Reproducibility: Challenges and Extensionscited by 10
- 2023High-dimensional Asymptotics of VAEs: Threshold of Posterior Collapse and Dataset-Size Dependence of Rate-Distortion Curvecited by 5
- 2019A Preliminary Study of Disentanglement With Insights on the Inadequacy of Metricscited by 4
- 2025Compressing Biology: Evaluating the Stable Diffusion VAE for Phenotypic Drug Discoverycited by 1
Built on by
Not cited yet.