Posterior Collapse as Automatic Spectral Pruning

· ArXiv · AI/CL/LG ·

Posterior collapse in β-VAEs is shown to implement automatic spectral pruning; Landau stability analysis reveals latent modes collapse from least to most useful, with collapse spectrum coinciding with PCA spectrum.

Categories: Research

Excerpt

We show that posterior collapse in $β$-VAEs implements automatic spectral pruning. A latent mode collapses if its contribution to reconstruction is below the cutoff set by $β$. Equilibrium solutions with different $β$ thus reveal a cascade of collapses as latent modes decouple from least to most useful. We derive this as a consequence of the loss via a Landau stability analysis. We define a latent-rescaling-invariant order parameter that ranks active latent modes and whose collapse thresholds identify which effective variables to inspect first. In the linear Gaussian case, the collapse spectrum, utility spectrum, and normalized PCA spectrum coincide, and each collapse follows a mean-field law. We test these predictions on the WorldClim dataset.