Poster
in
Workshop: Geometrical and Topological Representation Learning
Riemannian Neural SDE: Learning Stochastic Representations on Manifolds
Sung Woo Park · Hyomin Kim · Hyeseong Kim · Junseok Kwon
Keywords: [ stochastic differential equation ] [ density estimation ]
In recent years, the neural stochastic differential equation (NSDE) has gained attention in modeling stochastic representations, while NSDE brings a great success in various types of applications. However, it typically loses the expressivity when the data representation is manifold-valued. To overcome such an issue, we suggest a principled way to express the stochastic representation with the Riemannian neural SDE (RNSDE), which extends the conventional Euclidean NSDE. Empirical results on the density estimation on manifolds show that the proposed method significantly outperforms baseline methods.