Quasi-Monte Carlo Graph Random Features
Isaac Reid, Adrian Weller, Krzysztof Marcin Choromanski
Abstract
We present a novel mechanism to improve the accuracy of the recently-introduced class of graph random features (GRFs) [Choromanski, 2023] . Our method induces negative correlations between the lengths of the algorithm's random walks by imposing antithetic termination: a procedure to sample more diverse random walks which may be of independent interest. It has a trivial drop-in implementation. We derive strong theoretical guarantees on the properties of these quasi-Monte Carlo GRFs (q-GRFs), proving that they yield lower-variance estimators of the 2-regularised Laplacian kernel under mild conditions. Remarkably, our results hold for any graph topology. We demonstrate empirical accuracy improvements on a variety of tasks including a new practical application: time-efficient approximation of the graph diffusion process. To our knowledge, q-GRFs constitute the first rigorously studied quasi-Monte Carlo scheme for kernels defined on combinatorial objects, inviting new research on correlations between graph random walks. 1 * Senior lead. 1 We will make all code publicly available. Preprint. Under review.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fdd6a5c4-7ee3-4aa5-8190-2b43879b076cCited by top-tier papers6
- General Graph Random FeaturesIsaac Reid, Krzysztof Marcin Choromanski, Eli Berger, Adrian WellerICLR 2024 · 11 citations
- Repelling Random WalksIsaac Reid, Eli Berger, Krzysztof Marcin Choromanski, Adrian WellerICLR 2024 · 6 citations
- Computationally-efficient Graph Modeling with Refined Graph Random FeaturesKrzysztof Choromanski, Kumar Avinava Dubey, Arijit Sehanobish, Isaac ReidICML 2026 · 2 citations
- Rapid Training of Hamiltonian Graph Networks Using Random FeaturesAtamert Rahma, Chinmay Datar, Ana Cukarska, Felix DietrichICLR 2026 · 2 citations
- Variance-Reducing Couplings for Random FeaturesIsaac Reid, Stratis Markou, Krzysztof Marcin Choromanski, Richard E. Turner et al.ICLR 2025
Builds on5
- Random Walk Graph Neural NetworksGiannis Nikolentzos, Michalis VazirgiannisNeurIPS 2020 · 172 citations
- Rethinking Attention with PerformersKrzysztof Marcin Choromanski, Valerii Likhosherstov, David Dohan, Xingyou Song et al.ICLR 2021 · 122 citations
- Taming graph kernels with random featuresKrzysztof Marcin ChoromanskiICML 2023 · 21 citations
- Structure-Aware Random Fourier Kernel for GraphsJinyuan Fang, Qiang Zhang, Zaiqiao Meng, Shangsong LiangNeurIPS 2021 · 13 citations
- Simplex Random FeaturesIsaac Reid, Krzysztof Marcin Choromanski, Valerii Likhosherstov, Adrian WellerICML 2023 · 10 citations
Related papers
- Quasi-Monte Carlo Features for Kernel ApproximationZhen Huang, Jiajin Sun, Yian HuangICML 2024 · 6 citations
- Graph Random Features for Scalable Gaussian ProcessesMatthew Zhang, Jihao Andreas Lin, Krzysztof Choromanski, Adrian Weller et al.ICLR 2026 · 4 citations
- SWING: Unlocking Implicit Graph Representations for Graph Random FeaturesAlessandro Manenti, Kumar Avinava Dubey, Arijit Sehanobish, Cesare Alippi et al.ICML 2026
- Graph Random Neural Features for Distance-Preserving Graph RepresentationsDaniele Zambon, Cesare Alippi, Lorenzo LiviICML 2020 · 17 citations
- Weisfeiler and Leman Go Walking: Random Walk Kernels RevisitedNils M. KriegeNeurIPS 2022 · 22 citations
