Neumann eigenmaps for landmark embedding
Shashank Sule, Wojciech Czaja
TL;DR
NeuMaps address the scalability and robustness limitations of diffusion-map embeddings by embedding a landmark subset of the data through normalized Neumann eigenvectors of the landmark subgraph, yielding a diffusion distance for the reflecting random walk and a Nyström-compatible out-of-sample extension. The method reduces computation by solving a smaller eigenproblem while preserving geometric structure and exhibits improved stability to removal of highly significant points. Theoretical results establish an isometric embedding via the Neumann Laplacian and connect the extension to Nyström, with empirical validation on digit classification and molecular dynamics showing superior performance to Roseland and standard diffusion maps. Together, these contributions provide a boundary-aware, landmark-driven diffusion framework with practical impact for scalable, robust manifold learning.
Abstract
We present Neumann eigenmaps (NeuMaps), a novel approach for enhancing the standard diffusion map embedding using landmarks, i.e distinguished samples within the dataset. By interpreting these landmarks as a subgraph of the larger data graph, NeuMaps are obtained via the eigendecomposition of a renormalized Neumann Laplacian. We show that NeuMaps offer two key advantages: (1) they provide a computationally efficient embedding that accurately recovers the diffusion distance associated with the reflecting random walk on the subgraph, and (2) they naturally incorporate the Nyström extension within the diffusion map framework through the discrete Neumann boundary condition. Through examples in digit classification and molecular dynamics, we demonstrate that NeuMaps not only improve upon existing landmark-based embedding methods but also enhance the stability of diffusion map embeddings to the removal of highly significant points.
