English

Riemannian Diffusion Models

Machine Learning 2022-08-18 v1

Abstract

Diffusion models are recent state-of-the-art methods for image generation and likelihood estimation. In this work, we generalize continuous-time diffusion models to arbitrary Riemannian manifolds and derive a variational framework for likelihood estimation. Computationally, we propose new methods for computing the Riemannian divergence which is needed in the likelihood estimation. Moreover, in generalizing the Euclidean case, we prove that maximizing this variational lower-bound is equivalent to Riemannian score matching. Empirically, we demonstrate the expressive power of Riemannian diffusion models on a wide spectrum of smooth manifolds, such as spheres, tori, hyperboloids, and orthogonal groups. Our proposed method achieves new state-of-the-art likelihoods on all benchmarks.

Keywords

Cite

@article{arxiv.2208.07949,
  title  = {Riemannian Diffusion Models},
  author = {Chin-Wei Huang and Milad Aghajohari and Avishek Joey Bose and Prakash Panangaden and Aaron Courville},
  journal= {arXiv preprint arXiv:2208.07949},
  year   = {2022}
}