Saved in:
Bibliographic Details
Main Authors: Song, Yue, Keller, Thomas Anderson, Yue, Yisong, Perona, Pietro, Welling, Max
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.05564
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908874549755904
author Song, Yue
Keller, Thomas Anderson
Yue, Yisong
Perona, Pietro
Welling, Max
author_facet Song, Yue
Keller, Thomas Anderson
Yue, Yisong
Perona, Pietro
Welling, Max
contents There is a vast literature on representation learning based on principles such as coding efficiency, statistical independence, causality, controllability, or symmetry. In this paper we propose to learn representations from sequence data by factorizing the transformations of the latent variables into sparse components. Input data are first encoded as distributions of latent activations and subsequently transformed using a probability flow model, before being decoded to predict a future input state. The flow model is decomposed into a number of rotational (divergence-free) vector fields and a number of potential flow (curl-free) fields. Our sparsity prior encourages only a small number of these fields to be active at any instant and infers the speed with which the probability flows along these fields. Training this model is completely unsupervised using a standard variational objective and results in a new form of disentangled representations where the input is not only represented by a combination of independent factors, but also by a combination of independent transformation primitives given by the learned flow fields. When viewing the transformations as symmetries one may interpret this as learning approximately equivariant representations. Empirically we demonstrate that this model achieves state of the art in terms of both data likelihood and unsupervised approximate equivariance errors on datasets composed of sequence transformations.
format Preprint
id arxiv_https___arxiv_org_abs_2410_05564
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Unsupervised Representation Learning from Sparse Transformation Analysis
Song, Yue
Keller, Thomas Anderson
Yue, Yisong
Perona, Pietro
Welling, Max
Machine Learning
Computer Vision and Pattern Recognition
There is a vast literature on representation learning based on principles such as coding efficiency, statistical independence, causality, controllability, or symmetry. In this paper we propose to learn representations from sequence data by factorizing the transformations of the latent variables into sparse components. Input data are first encoded as distributions of latent activations and subsequently transformed using a probability flow model, before being decoded to predict a future input state. The flow model is decomposed into a number of rotational (divergence-free) vector fields and a number of potential flow (curl-free) fields. Our sparsity prior encourages only a small number of these fields to be active at any instant and infers the speed with which the probability flows along these fields. Training this model is completely unsupervised using a standard variational objective and results in a new form of disentangled representations where the input is not only represented by a combination of independent factors, but also by a combination of independent transformation primitives given by the learned flow fields. When viewing the transformations as symmetries one may interpret this as learning approximately equivariant representations. Empirically we demonstrate that this model achieves state of the art in terms of both data likelihood and unsupervised approximate equivariance errors on datasets composed of sequence transformations.
title Unsupervised Representation Learning from Sparse Transformation Analysis
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.05564