arXiv · 1802.04942
Isolating Sources of Disentanglement in Variational Autoencoders
Abstract
We decompose the evidence lower bound to show the existence of a term measuring the total correlation between latent variables. We use this to motivate our $β$-TCVAE (Total Correlation Variational Autoencoder), a refinement of the state-of-the-art $β$-VAE objective for learning disentangled representations, requiring no additional hyperparameters during training. We further propose a principled classifier-free measure of disentanglement called the mutual information gap (MIG). We perform extensive quantitative and qualitative experiments, in both restricted and non-restricted settings, and show a strong relation between total correlation and disentanglement, when the latent variables model is trained using our framework.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Ricky T. Q. Chen, Xuechen Li, Roger Grosse, David Duvenaud. 2019-04-23. Isolating Sources of Disentanglement in Variational Autoencoders. https://arxiv.org/abs/1802.04942
Cite the original work for its findings. Save a collection to share your selection of sources.