NeurIPS2023
Gacs-Korner Common Information Variational Autoencoder
Michael Kleinman, Alessandro Achille, Stefano Soatto, Jonathan C. Kao
20 citations
Abstract
We propose a notion of common information that allows one to quantify and separate the information that is shared between two random variables from the information that is unique to each. Our notion of common information is defined by an optimization problem over a family of functions and recovers the Gács-Körner common information as a special case. Importantly, our notion can be approximated empirically using samples from the underlying data distribution. We then provide a method to partition and quantify the common and unique information using a simple modification of a traditional variational auto-encoder. Empirically, we demonstrate that our formulation allows us to learn semantically meaningful common and unique factors of variation even on high-dimensional data such as images and videos. Moreover, on datasets where ground-truth latent factors are known, we show that we can accurately quantify the common information between the random variables. 2 * Work performed as external collaboration not related to Amazon 2 Code available at: https://github.com/mjkleinman/common-vae 37th Conference on Neural Information Processing Systems (NeurIPS 2023).