Riemannian Continuous Normalizing Flows

06/18/2020
by   Emile Mathieu, et al.
14

Normalizing flows have shown great promise for modelling flexible probability distributions in a computationally tractable way. However, whilst data is often naturally described on Riemannian manifolds such as spheres, torii, and hyperbolic spaces, most normalizing flows implicitly assume a flat geometry, making them either misspecified or ill-suited in these situations. To overcome this problem, we introduce Riemannian continuous normalizing flows, a model which admits the parametrization of flexible probability measures on smooth manifolds by defining flows as the solution to ordinary differential equations. We show that this approach can lead to substantial improvements on both synthetic and real-world data when compared to standard flows or previously introduced projected flows.

READ FULL TEXT
research
06/11/2020

Neural Ordinary Differential Equations on Manifolds

Normalizing flows are a powerful technique for obtaining reparameterizab...
research
11/07/2016

Normalizing Flows on Riemannian Manifolds

We consider the problem of density estimation on Riemannian manifolds. D...
research
06/18/2021

Riemannian Convex Potential Maps

Modeling distributions on Riemannian manifolds is a crucial component in...
research
06/18/2020

Neural Manifold Ordinary Differential Equations

To better conform to data geometry, recent deep generative modelling tec...
research
10/04/2021

Implicit Riemannian Concave Potential Maps

We are interested in the challenging problem of modelling densities on R...
research
10/24/2020

Gradient Flows in Dataset Space

The current practice in machine learning is traditionally model-centric,...
research
01/22/2019

Fast and Robust Shortest Paths on Manifolds Learned from Data

We propose a fast, simple and robust algorithm for computing shortest pa...

Please sign up or login with your details

Forgot password? Click here to reset