Disentanglement of Latent Representations via Causal Interventions

Disentanglement of Latent Representations via Causal Interventions

Gaël Gendron, Michael Witbrock, Gillian Dobbie

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence
Main Track. Pages 3239-3247. https://doi.org/10.24963/ijcai.2023/361

The process of generating data such as images is controlled by independent and unknown factors of variation. The retrieval of these variables has been studied extensively in the disentanglement, causal representation learning, and independent component analysis fields. Recently, approaches merging these domains together have shown great success. Instead of directly representing the factors of variation, the problem of disentanglement can be seen as finding the interventions on one image that yield a change to a single factor. Following this assumption, we introduce a new method for disentanglement inspired by causal dynamics that combines causality theory with vector-quantized variational autoencoders. Our model considers the quantized vectors as causal variables and links them in a causal graph. It performs causal interventions on the graph and generates atomic transitions affecting a unique factor of variation in the image. We also introduce a new task of action retrieval that consists of finding the action responsible for the transition between two images. We test our method on standard synthetic and real-world disentanglement datasets. We show that it can effectively disentangle the factors of variation and perform precise interventions on high-level semantic attributes of an image without affecting its quality, even with imbalanced data distributions.
Keywords:
Knowledge Representation and Reasoning: KRR: Causality
Computer Vision: CV: Representation learning
Machine Learning: ML: Autoencoders