Autonomous Learning Conference Paper 2021

Demystifying Inductive Biases for (Beta-)VAE Based Architectures

Preview 2846

The performance of Beta-Variational-Autoencoders and their variants on learning semantically meaningful, disentangled representations is unparalleled. On the other hand, there are theoretical arguments suggesting the impossibility of unsupervised disentanglement. In this work, we shed light on the inductive bias responsible for the success of VAE-based architectures. We show that in classical datasets the structure of variance, induced by the generating factors, is conveniently aligned with the latent directions fostered by the VAE objective. This builds the pivotal bias on which the disentangling abilities of VAEs rely. By small, elaborate perturbations of existing datasets, we hide the convenient correlation structure that is easily exploited by a variety of architectures. To demonstrate this, we construct modified versions of standard datasets in which (i) the generative factors are perfectly preserved; (ii) each image undergoes a mild transformation causing a small change of variance; (iii) the leading VAE-based disentanglement architectures fail to produce disentangled representations whilst the performance of a non-variational method remains unchanged.

Author(s): Dominik Zietlow and Michal Rolinek and Georg Martius
Book Title: Proceedings of the 2021 International Conference on Machine Learning (ICML)
Volume: 139
Pages: 12945--12954
Year: 2021
Month: July
Series: Proceedings of Machine Learning Research
Project(s):
Bibtex Type: Conference Paper (inproceedings)
Event Name: The 38th International Conference on Machine Learning (ICML 2021)
Event Place: Virtual
State: Published
URL: https://proceedings.mlr.press/v139/zietlow21a.html
Electronic Archiving: grant_archive
Links:

BibTex

@inproceedings{zietld2021:demystifying,
  title = {Demystifying Inductive Biases for (Beta-)VAE Based Architectures},
  booktitle = {Proceedings of the 2021 International Conference on Machine Learning (ICML)},
  abstract = {The performance of Beta-Variational-Autoencoders and their variants on learning semantically meaningful, disentangled representations is unparalleled. On the other hand, there are theoretical arguments suggesting the impossibility of unsupervised disentanglement. In this work, we shed light on the inductive bias responsible for the success of VAE-based architectures. We show that in classical datasets the structure of variance, induced by the generating factors, is conveniently aligned with the latent directions fostered by the VAE objective. This builds the pivotal bias on which the disentangling abilities of VAEs rely. By small, elaborate perturbations of existing datasets, we hide the convenient correlation structure that is easily exploited by a variety of architectures. To demonstrate this, we construct modified versions of standard datasets in which (i) the generative factors are perfectly preserved; (ii) each image undergoes a mild transformation causing a small change of variance; (iii) the leading VAE-based disentanglement architectures fail to produce disentangled representations whilst the performance of a non-variational method remains unchanged. },
  volume = {139},
  pages = {12945--12954},
  series = {Proceedings of Machine Learning Research },
  month = jul,
  year = {2021},
  slug = {zietld2021-demystifying},
  author = {Zietlow, Dominik and Rolinek, Michal and Martius, Georg},
  url = {https://proceedings.mlr.press/v139/zietlow21a.html},
  month_numeric = {7}
}