Repository logo
 

Learning disentangled representations with semi-supervised deep generative models

Accepted version
Peer-reviewed

Type

Conference Object

Change log

Authors

Narayanaswamy, Siddharth 
Paige, T Brooks 
Van de Meent, Jan-Willem 
Desmaison, Alban 
Goodman, Noah 

Abstract

Variational autoencoders (VAEs) learn representations of data by jointly training a probabilistic encoder and decoder network. Typically these models encode all features of the data into a single variable. Here we are interested in learning disentangled representations that encode distinct aspects of the data into separate variables. We propose to learn such representations using model architectures that generalise from standard VAEs, employing a general graphical model structure in the encoder and decoder. This allows us to train partially-specified models that make relatively strong assumptions about a subset of interpretable variables and rely on the flexibility of neural networks to learn representations for the remaining variables. We further define a general objective for semi-supervised learning in this model class, which can be approximated using an importance sampling procedure. We evaluate our framework's ability to learn disentangled representations, both by qualitative exploration of its generative capacity, and quantitative evaluation of its discriminative ability on a variety of models and datasets.

Description

Keywords

Journal Title

Proceedings of the 31st International Conference on Neural Information Processing Systems

Conference Name

31st Conference on Neural Information Processing Systems

Journal ISSN

Volume Title

Publisher

Curran Associates Inc.
Sponsorship
Alan Turing Institute (AT/I00009/16)