English

Mutual Information Collapse Explains Disentanglement Failure in $\beta$-VAEs

Machine Learning 2026-02-11 v1 Machine Learning

Abstract

The β\beta-VAE is a foundational framework for unsupervised disentanglement, using β\beta 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 β\beta 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\beta > 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 λβ\lambda\beta-VAE, which decouples regularization pressure from informational collapse via an auxiliary L2L_2 reconstruction penalty λ\lambda. Extensive experiments on dSprites, Shapes3D, and MPI3D-real confirm that λ>0\lambda > 0 stabilizes disentanglement and restores latent informativeness over a significantly broader range of β\beta, providing a principled theoretical justification for dual-parameter regularization in variational inference backbones.

Cite

@article{arxiv.2602.09277,
  title  = {Mutual Information Collapse Explains Disentanglement Failure in $\beta$-VAEs},
  author = {Minh Vu and Xiaoliang Wan and Shuangqing Wei},
  journal= {arXiv preprint arXiv:2602.09277},
  year   = {2026}
}
R2 v1 2026-07-01T10:28:56.775Z