Riemannian Continuous Normalizing Flows

by   Emile Mathieu, et al.

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.


Neural Ordinary Differential Equations on Manifolds

Normalizing flows are a powerful technique for obtaining reparameterizab...

Normalizing Flows on Riemannian Manifolds

We consider the problem of density estimation on Riemannian manifolds. D...

Riemannian Convex Potential Maps

Modeling distributions on Riemannian manifolds is a crucial component in...

Neural Manifold Ordinary Differential Equations

To better conform to data geometry, recent deep generative modelling tec...

Implicit Riemannian Concave Potential Maps

We are interested in the challenging problem of modelling densities on R...

Gradient Flows in Dataset Space

The current practice in machine learning is traditionally model-centric,...

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