Skip to yearly menu bar Skip to main content


Symmetry-induced Disentanglement on Graphs

Giangiacomo Mercatali · Andre Freitas · Vikas Garg

Hall J (level 1) #523

Keywords: [ Generative Models ] [ graph neural networks ] [ Disentanglement ] [ Latent Variable Models ]


Learning disentangled representations is important for unraveling the underlying complex interactions between latent generative factors. Disentanglement has been formalized using a symmetry-centric notion for unstructured spaces, however, graphs have eluded a similarly rigorous treatment. We fill this gap with a new notion of conditional symmetry for disentanglement, and leverage tools from Lie algebras to encode graph properties into subgroups using suitable adaptations of generative models such as Variational Autoencoders. Unlike existing works on disentanglement, the proposed models segregate the latent space into uncoupled and entangled parts. Experiments on synthetic and real datasets suggest that these models can learn effective disengaged representations, and improve performance on downstream tasks such as few-shot classification and molecular generation.

Chat is not available.