arXiv · 2409.14851
Disentanglement with Factor Quantized Variational Autoencoders
Abstract
Disentangled representation learning aims to represent the underlying generative factors of a dataset in a latent representation independently of one another. In our work, we propose a discrete variational autoencoder (VAE) based model where the ground truth information about the generative factors are not provided to the model. We demonstrate the advantages of learning discrete representations over learning continuous representations in facilitating disentanglement. Furthermore, we propose incorporating an inductive bias into the model to further enhance disentanglement. Precisely, we propose scalar quantization of the latent variables in a latent representation with scalar values from a global codebook, and we add a total correlation term to the optimization as an inductive bias. Our method called FactorQVAE combines optimization based disentanglement approaches with discrete representation learning, and it outperforms the former disentanglement methods in terms of two disentanglement metrics (DCI and InfoMEC) while improving the reconstruction performance. Our code can be found at https://github.com/ituvisionlab/FactorQVAE.
Explore related subjects
Keep this discovery
Gulcin Baykal, Melih Kandemir, Gozde Unal. 2024-09-23. Disentanglement with Factor Quantized Variational Autoencoders. https://doi.org/10.1016/j.neucom.2025.131968
Cite the original work for its findings. Save a collection to share your selection of sources.