Implicit Gaussian Process Representation of Vector Fields Over Arbitrary Latent Mani-Folds
Gaussian processes (GPs) are popular nonparametric statistical models for learning unknown functions and quantifying the spatiotemporal uncertainty in data. Recent works have extended GPs to model scalar and vector quantities distributed over non-Euclidean domains, including smooth manifolds appearing in numerous fields such as computer vision, dynamical systems, and neuroscience. However, these approaches assume that the manifold underlying the data is known, limiting their practical utility. We introduce RVGP, a generalisation of GPs for learning vector signals over latent Riemannian manifolds. Our method uses positional encoding with eigenfunctions of the connection Laplacian, associated with the tangent bundle, readily derived from common graph-based data approximation. We demonstrate that RVGP possesses global regularity over the manifold, which allows it to super-resolve and inpaint vector fields while preserving singularities. Furthermore, we use RVGP to reconstruct high-density neural dynamics derived from low-density EEG recordings in healthy individuals and Alzheimer's patients. We show that vector field singularities are important disease markers and that their reconstruction leads to a classification accuracy of disease states comparable to high-density recordings. Thus, our method overcomes a significant practical limitation in experimental and clinical applications.
ICLR-2024-implicit-gaussian-process-representation-of-vector-fields-over-arbitrary-latent-manifolds-Paper-Conference.pdf
Main Document
Accepted version
openaccess
CC BY
3.14 MB
Adobe PDF
9309156f212aff339f15e102e99bbfc8