Skip to yearly menu bar Skip to main content


Poster

Riemannian Neural SDE: Learning Stochastic Representations on Manifolds

Sung Woo Park · Hyomin Kim · Kyungjae Lee · Junseok Kwon

Hall J (level 1) #924

Keywords: [ Riemannian neural stochastic differential equation ] [ Stochastic representation on Manifolds ]


Abstract:

In recent years, the neural stochastic differential equation (NSDE) has gained attention for modeling stochastic representations with great success in various types of applications. However, it typically loses expressivity when the data representation is manifold-valued. To address this issue, we suggest a principled method for expressing the stochastic representation with the Riemannian neural SDE (RNSDE), which extends the conventional Euclidean NSDE. Empirical results for various tasks demonstrate that the proposed method significantly outperforms baseline methods.

Chat is not available.