Manifold Random Features
Ananya Parashar ⋅ Derek Long ⋅ Dwaipayan Saha ⋅ Krzysztof M Choromanski
Abstract
We present a new paradigm for creating random features to approximate bi-variate functions (in particular, kernels) defined on general manifolds. This new mechanism of $\textit{Manifold Random Features}$ (MRFs) leverages discretization of the manifold and the recently introduced technique of $\textit{Graph Random Features}$ (GRFs) to learn continuous fields on manifolds. Those fields are used to find continuous approximation mechanisms that otherwise, in general scenarios, cannot be derived analytically. MRFs provide positive and bounded features, a key property for accurate, low-variance approximation. We show deep asymptotic connection between GRFs, defined on discrete graph objects, and continuous random features used for regular kernels. As a by-product of our method, we re-discover recently introduced mechanism of Gaussian kernel approximation applied in particular to improve linear-attention Transformers, considering simple random walks on graphs and by-passing original complex mathematical computations. We complement our algorithm with a rigorous theoretical analysis and verify in thorough experimental studies.
Chat is not available.
Successful Page Load