Scale-aware Message Passing For Graph Node Classification
About
Most Graph Neural Networks (GNNs) operate at the first-order scale, even though multi-scale representations are known to be crucial in domains such as image classification. In this work, we investigate whether GNNs can similarly benefit from multi-scale learning, rather than being limited to a fixed depth of $k$-hop aggregation. We begin by formalizing scale invariance in graph learning, providing theoretical guarantees and empirical evidence for its effectiveness. Building on this principle, we introduce ScaleNet, a scale-aware message-passing architecture that combines directed multi-scale feature aggregation with an adaptive self-loop mechanism. ScaleNet achieves state-of-the-art performance on six benchmark datasets, covering both homophilic and heterophilic graphs. To handle scalability, we further propose LargeScaleNet, which extends multi-scale learning to large graphs and sets new state-of-the-art results on three large-scale benchmarks. We also show that FaberNet's strength largely arises from multi-scale feature integration. Together with these state-of-the-art results, our findings suggest that scale invariance may serve as a valuable principle for improving the performance of single-order GNNs. The code for all experiments is available at \href{https://github.com/Qin87/ScaleNet/tree/iclr_scale_aware/}{this link}.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Node Classification | Chameleon | Accuracy80.1 | 640 | |
| Node Classification | Squirrel | Accuracy76 | 591 | |
| Node Classification | Citeseer | Accuracy68.3 | 393 | |
| Node Classification | wikiCS | Accuracy79.3 | 317 | |
| Node Classification | Cora-ML | Accuracy82.3 | 232 | |
| Node Classification | Telegram | Accuracy96.8 | 32 | |
| Node Classification | Cora-ML imbalanced, ratio 100:1 (test) | Accuracy64.9 | 7 | |
| Node Classification | CiteSeer imbalanced, ratio 100:1 (test) | Accuracy43.1 | 7 | |
| Node Classification | WikiCS imbalanced, ratio 100:1 (test) | Accuracy71 | 7 |