Guardado en:
Detalles Bibliográficos
Autores principales: Boll, Bastian, Gonzalez-Alvarado, Daniel, Petra, Stefania, Schnörr, Christoph
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:https://arxiv.org/abs/2406.04527
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929674465050624
author Boll, Bastian
Gonzalez-Alvarado, Daniel
Petra, Stefania
Schnörr, Christoph
author_facet Boll, Bastian
Gonzalez-Alvarado, Daniel
Petra, Stefania
Schnörr, Christoph
contents We introduce a novel generative model for the representation of joint probability distributions of a possibly large number of discrete random variables. The approach uses measure transport by randomized assignment flows on the statistical submanifold of factorizing distributions, which enables to represent and sample efficiently from any target distribution and to assess the likelihood of unseen data points. The complexity of the target distribution only depends on the parametrization of the affinity function of the dynamical assignment flow system. Our model can be trained in a simulation-free manner by conditional Riemannian flow matching, using the training data encoded as geodesics on the assignment manifold in closed-form, with respect to the e-connection of information geometry. Numerical experiments devoted to distributions of structured image labelings demonstrate the applicability to large-scale problems, which may include discrete distributions in other application areas. Performance measures show that our approach scales better with the increasing number of classes than recent related work.
format Preprint
id arxiv_https___arxiv_org_abs_2406_04527
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generative Assignment Flows for Representing and Learning Joint Distributions of Discrete Data
Boll, Bastian
Gonzalez-Alvarado, Daniel
Petra, Stefania
Schnörr, Christoph
Machine Learning
We introduce a novel generative model for the representation of joint probability distributions of a possibly large number of discrete random variables. The approach uses measure transport by randomized assignment flows on the statistical submanifold of factorizing distributions, which enables to represent and sample efficiently from any target distribution and to assess the likelihood of unseen data points. The complexity of the target distribution only depends on the parametrization of the affinity function of the dynamical assignment flow system. Our model can be trained in a simulation-free manner by conditional Riemannian flow matching, using the training data encoded as geodesics on the assignment manifold in closed-form, with respect to the e-connection of information geometry. Numerical experiments devoted to distributions of structured image labelings demonstrate the applicability to large-scale problems, which may include discrete distributions in other application areas. Performance measures show that our approach scales better with the increasing number of classes than recent related work.
title Generative Assignment Flows for Representing and Learning Joint Distributions of Discrete Data
topic Machine Learning
url https://arxiv.org/abs/2406.04527