Taming graph kernels with random features
Krzysztof Marcin Choromanski
Abstract
We introduce in this paper the mechanism of graph random features (GRFs). GRFs can be used to construct unbiased randomized estimators of several important kernels defined on graphs' nodes, in particular the regularized Laplacian kernel. As regular RFs for non-graph kernels, they provide means to scale up kernel methods defined on graphs to larger networks. Importantly, they give substantial computational gains also for smaller graphs, while applied in downstream applications. Consequently, GRFs address the notoriously difficult problem of cubic (in the number of the nodes of the graph) time complexity of graph kernels algorithms. We provide a detailed theoretical analysis of GRFs and an extensive empirical evaluation: from speed tests, through Frobenius relative error analysis to kmeans graph-clustering with graph kernels. We show that the computation of GRFs admits an embarrassingly simple distributed algorithm that can be applied if the graph under consideration needs to be split across several machines. We also introduce a (still unbiased) quasi Monte Carlo variant of GRFs, q-GRFs, relying on the so-called reinforced random walks, that might be used to optimize the variance of GRFs. As a byproduct, we obtain a novel approach to solve certain classes of linear equations with positive and symmetric matrices.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers14
- General Graph Random FeaturesIsaac Reid, Krzysztof Marcin Choromanski, Eli Berger, Adrian WellerICLR 2024 · 11 citations
- Expectation-Complete Graph Representations with HomomorphismsPascal Welke, Maximilian Thiessen, Fabian Jogl, Thomas GärtnerICML 2023 · 11 citations
- Quasi-Monte Carlo Graph Random FeaturesIsaac Reid, Adrian Weller, Krzysztof Marcin ChoromanskiNeurIPS 2023 · 11 citations
- Federated Graph-Level Clustering NetworkJingxin Liu, Jieren Cheng, Renda Han, Wenxuan Tu et al.AAAI 2025 · 9 citations
- Repelling Random WalksIsaac Reid, Eli Berger, Krzysztof Marcin Choromanski, Adrian WellerICLR 2024 · 6 citations
Builds on6
- Rethinking Attention with PerformersKrzysztof Marcin Choromanski, Valerii Likhosherstov, David Dohan, Xingyou Song et al.ICLR 2021 · 122 citations
- From block-Toeplitz matrices to differential equations on graphs: towards a general theory for scalable masked TransformersKrzysztof Choromanski, Han Lin, Haoxian Chen, Tianyi Zhang et al.ICML 2022 · 47 citations
- Hybrid Random FeaturesKrzysztof Marcin Choromanski, Han Lin, Haoxian Chen, Arijit Sehanobish et al.ICLR 2022 · 27 citations
- Chefs' Random Tables: Non-Trigonometric Random FeaturesValerii Likhosherstov, Krzysztof Marcin Choromanski, Kumar Avinava Dubey, Frederick Liu et al.NeurIPS 2022 · 21 citations
- Structure-Aware Random Fourier Kernel for GraphsJinyuan Fang, Qiang Zhang, Zaiqiao Meng, Shangsong LiangNeurIPS 2021 · 13 citations
Related papers
- Computationally-efficient Graph Modeling with Refined Graph Random FeaturesKrzysztof Choromanski, Kumar Avinava Dubey, Arijit Sehanobish, Isaac ReidICML 2026 · 2 citations
- Graph Random Features for Scalable Gaussian ProcessesMatthew Zhang, Jihao Andreas Lin, Krzysztof Choromanski, Adrian Weller et al.ICLR 2026 · 4 citations
- Decentralised Learning with Random Features and Distributed Gradient DescentDominic Richards, Patrick Rebeschini, Lorenzo RosascoICML 2020 · 20 citations
- SWING: Unlocking Implicit Graph Representations for Graph Random FeaturesAlessandro Manenti, Kumar Avinava Dubey, Arijit Sehanobish, Cesare Alippi et al.ICML 2026
- Variance-Reducing Couplings for Random FeaturesIsaac Reid, Stratis Markou, Krzysztof Marcin Choromanski, Richard E. Turner et al.ICLR 2025
