Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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}.

Qin Jiang, Chengjia Wang, Michael Lones, Dongdong Chen, Wei Pang• 2024

Related benchmarks

TaskDatasetResultRank
Node ClassificationChameleon
Accuracy80.1
640
Node ClassificationSquirrel
Accuracy76
591
Node ClassificationCiteseer
Accuracy68.3
393
Node ClassificationwikiCS
Accuracy79.3
317
Node ClassificationCora-ML
Accuracy82.3
232
Node ClassificationTelegram
Accuracy96.8
32
Node ClassificationCora-ML imbalanced, ratio 100:1 (test)
Accuracy64.9
7
Node ClassificationCiteSeer imbalanced, ratio 100:1 (test)
Accuracy43.1
7
Node ClassificationWikiCS imbalanced, ratio 100:1 (test)
Accuracy71
7
Showing 9 of 9 rows

Other info

Code

Follow for update