mlsanityData to 7 Oct 2026

Mutual Information Collapse Explains Disentanglement Failure in β-VAEs

arXiv 2602.09277 ↗First posted 9 Feb 2026stat.ML cs.LGcited by 0

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

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  3. 2018Challenging Common Assumptions in the Unsupervised Learning of Disentangled Representationscited by 929
  4. 2018Disentangling by Factorisingcited by 614
  5. 2017Variational Inference of Disentangled Latent Concepts from Unlabeled Observationscited by 240
  6. 2021Towards Causal Representation Learningcited by 207
  7. 2019On the Transfer of Inductive Bias from Simulation to the Real World: a New Disentanglement Datasetcited by 100
  8. 2020Measuring Disentanglement: A Review of Metricscited by 69
  9. 2022SQ-VAE: Variational Bayes on Discrete Representation with Self-annealed Stochastic Quantizationcited by 68
  10. 2021Visual Representation Learning Does Not Generalize Strongly Within the Same Domaincited by 60
  11. 2022Posterior Collapse of a Linear Latent Variable Modelcited by 18
  12. 2019Scalarizing Functions in Bayesian Multiobjective Optimizationcited by 15
7 more
  1. 2021Preventing Oversmoothing in VAE via Generalized Variance Parameterizationcited by 15
  2. 2022Hyperbolic VAE via Latent Gaussian Distributionscited by 14
  3. 2023Learning Dynamics in Linear VAE: Posterior Collapse Threshold, Superfluous Latent Space Pitfalls, and Speedup with KL Annealingcited by 12
  4. 2021Beta-VAE Reproducibility: Challenges and Extensionscited by 10
  5. 2023High-dimensional Asymptotics of VAEs: Threshold of Posterior Collapse and Dataset-Size Dependence of Rate-Distortion Curvecited by 5
  6. 2019A Preliminary Study of Disentanglement With Insights on the Inadequacy of Metricscited by 4
  7. 2025Compressing Biology: Evaluating the Stable Diffusion VAE for Phenotypic Drug Discoverycited by 1

Built on by

Not cited yet.