atlas

Research map / Graph learning / GNN architectures

From Theory to Practice: Rethinking Green and Martin Kernels for Unleashing Graph Transformers

ICML 2025 · Graph learning › GNN architectures · approach: Expressive equivariant message passing

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces Green- and Martin-kernel-based structural encodings (GKSE, MKSE), redefined with finite-step random walk quantities, and plugs them into GRIT to improve graph transformer performance on molecular and circuit benchmarks.

Problem

Graph Transformers depend on structural encodings for inductive bias, but existing random-walk encodings (RRWP) are unstable on non-aperiodic substructures and sparse on DAGs; classical Green and Martin kernels are ill-defined for recurrent finite graphs.

Approach

Redefine Green and Martin kernels as finite-step quantities (expected visits within k steps; probability of returning within k steps via first-return time), compute them recursively, and use them as relative and absolute structural encodings in GRIT.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Replace RRWP with GKSEGRIT · ZINC testMAE: 0.059 ± 0.001 -> 0.058 ± 0.002Helped
Replace RRWP with MKSEGRIT · ZINC testMAE: 0.059 ± 0.001 -> 0.056 ± 0.021Helped
Replace RRWP with GKSEGRIT · MNIST testaccuracy: 98.108 ± 0.111 -> 98.305 ± 0.125Helped
Replace RRWP with MKSEGRIT · CIFAR10 testaccuracy: 76.468 ± 0.881 -> 77.365 ± 0.640Helped
Replace RRWP with GKSEGRIT · Peptides-func testAP: 0.6988 ± 0.0082 -> 0.6976 ± 0.0097No clear effect
Replace RRWP with MKSEGRIT · Peptides-func testAP: 0.6988 ± 0.0082 -> 0.6784 ± 0.0057Hurt
Replace RRWP with GKSEGRIT · Peptides-struct testMAE: 0.2460 ± 0.0012 -> 0.2452 ± 0.0012Helped
Replace RRWP with MKSEGRIT · Peptides-struct testMAE: 0.2460 ± 0.0012 -> 0.2457 ± 0.0013Helped
Replace GPS-medium with GKSEGRIT · PCQM4Mv2 testMAE: 0.0858 -> 0.0837 ± 0.0002Helped
Replace RRWP with GKSEGRIT · Ckt-Bench101 testMAE: 0.0418 ± 0.0021 -> 0.0395 ± 0.0033Helped
Replace RRWP with MKSEGRIT · Ckt-Bench101 testMAE: 0.0418 ± 0.0021 -> 0.0409 ± 0.0016Helped
Replace RRWP with GKSEGRIT · Ckt-Bench301 testMAE: 0.0190 ± 0.0005 -> 0.0188 ± 0.0004Helped

Surprising findings

Limitations

Experimental setup

Datasets: ZINC MNIST CIFAR10 Peptides-func Peptides-struct PCQM4Mv2 Ckt-Bench101 Ckt-Bench301

Benchmarks: ZINC MNIST CIFAR10 Peptides-func Peptides-struct PCQM4Mv2 Ckt-Bench101 Ckt-Bench301

Metrics: MAE accuracy AP

Baselines: GRIT+RRWP GPS EGT PNA GatedGCN TokenGT Graphormer GRPE Specformer

Builds on: GRIT RRWP GraphGPS Generalized Distance Weisfeiler-Lehman

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy