arXiv · 2106.03500
Density estimation on smooth manifolds with normalizing flows
Abstract
We present a framework for learning probability distributions on topologically non-trivial manifolds, utilizing normalizing flows. Current methods focus on manifolds that are homeomorphic to Euclidean space, enforce strong structural priors on the learned models or use operations that do not easily scale to high dimensions. In contrast, our method learns distributions on a data manifold by "gluing" together multiple local models, thus defining an open cover of the data manifold. We demonstrate the efficiency of our approach on synthetic data of known manifolds, as well as higher dimensional manifolds of unknown topology, where our method exhibits better sample efficiency and competitive or superior performance against baselines in a number of tasks.
Explore related subjects
Keep this discovery
Dimitris Kalatzis, Johan Ziruo Ye, Alison Pouplin, Jesper Wohlert, Søren Hauberg. 2021-06-07. Density estimation on smooth manifolds with normalizing flows. https://arxiv.org/abs/2106.03500
Cite the original work for its findings. Save a collection to share your selection of sources.