Optimal Time Complexity Algorithms for Computing General Random Walk Graph Kernels on Sparse Graphs

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Choromanski, Krzysztof, Reid, Isaac, Sehanobish, Arijit, Dubey, Avinava
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910650690699264
author Choromanski, Krzysztof
Reid, Isaac
Sehanobish, Arijit
Dubey, Avinava
author_facet Choromanski, Krzysztof
Reid, Isaac
Sehanobish, Arijit
Dubey, Avinava
contents We present the first linear time complexity randomized algorithms for unbiased approximation of the celebrated family of general random walk kernels (RWKs) for sparse graphs. This includes both labelled and unlabelled instances. The previous fastest methods for general RWKs were of cubic time complexity and not applicable to labelled graphs. Our method samples dependent random walks to compute novel graph embeddings in $\mathbb{R}^d$ whose dot product is equal to the true RWK in expectation. It does so without instantiating the direct product graph in memory, meaning we can scale to massive datasets that cannot be stored on a single machine. We derive exponential concentration bounds to prove that our estimator is sharp, and show that the ability to approximate general RWKs (rather than just special cases) unlocks efficient implicit graph kernel learning. Our method is up to $\mathbf{27\times}$ faster than its counterparts for efficient computation on large graphs and scales to graphs $\mathbf{128 \times}$ bigger than largest examples amenable to brute-force computation.
format Preprint
id arxiv_https___arxiv_org_abs_2410_10368
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Optimal Time Complexity Algorithms for Computing General Random Walk Graph Kernels on Sparse Graphs
Choromanski, Krzysztof
Reid, Isaac
Sehanobish, Arijit
Dubey, Avinava
Machine Learning
We present the first linear time complexity randomized algorithms for unbiased approximation of the celebrated family of general random walk kernels (RWKs) for sparse graphs. This includes both labelled and unlabelled instances. The previous fastest methods for general RWKs were of cubic time complexity and not applicable to labelled graphs. Our method samples dependent random walks to compute novel graph embeddings in $\mathbb{R}^d$ whose dot product is equal to the true RWK in expectation. It does so without instantiating the direct product graph in memory, meaning we can scale to massive datasets that cannot be stored on a single machine. We derive exponential concentration bounds to prove that our estimator is sharp, and show that the ability to approximate general RWKs (rather than just special cases) unlocks efficient implicit graph kernel learning. Our method is up to $\mathbf{27\times}$ faster than its counterparts for efficient computation on large graphs and scales to graphs $\mathbf{128 \times}$ bigger than largest examples amenable to brute-force computation.
title Optimal Time Complexity Algorithms for Computing General Random Walk Graph Kernels on Sparse Graphs
topic Machine Learning
url https://arxiv.org/abs/2410.10368