Graph Cascades: Contagion-Based Mesoscopic Rewiring for Structure-Aware Graph Machine Learning
About
We introduce Graph Cascades, a mesoscopic rewiring strategy for Graph Neural Networks (GNNs) and Graph Transformers (GTs) that captures intermediate-scale graph structure beyond purely local edges or fully global attention. Using contagion-based diffusion processes, Graph Cascades constructs, in O(|V|+|E|) time, an auxiliary graph where node pairs supported by repeated multi-hop reinforcement are promoted to direct neighbors. We theoretically characterize when reinforcement-based rewiring helps: sufficient conditions under which reinforcement-based edge selection is more label-aligned than direct adjacency, an SBM witness in which two-hop reinforcement is perfectly homophilic, and a formalization of mesoscopic connectivity via graph effective resistance. Empirically, across node-classification benchmarks, Graph Cascades improves multiple GNN and sparse-GT backbones, with the most reliable gains observed on heterophilic and moderate- to high-degree homophilic graphs. The theoretical conditions also identify regimes where mesoscopic rewiring is unlikely to be beneficial -- low-degree regular graphs and graphs with structural bottlenecks -- and these predictions match the observed failures. We additionally observe tight correlations between performance and structural properties in the rewired graphs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Citeseer (test) | Accuracy0.757 | 1013 | |
| Node Classification | PubMed (test) | Accuracy88.7 | 628 | |
| Node Classification | Chameleon (test) | Mean Accuracy63.65 | 425 | |
| Node Classification | Cornell (test) | Mean Accuracy75.02 | 403 | |
| Node Classification | Texas (test) | Mean Accuracy82.31 | 402 | |
| Node Classification | Roman-Empire | -- | 398 | |
| Node Classification | Squirrel (test) | Mean Accuracy47.31 | 353 | |
| Node Classification | Wisconsin (test) | Mean Accuracy81.02 | 346 | |
| Node Classification | Actor (test) | Mean Accuracy0.3646 | 339 | |
| Node Classification | Cora (test) | Accuracy87.96 | 254 |