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

Generative Graph Pattern Machine

About

Graph neural networks (GNNs) have been predominantly driven by message-passing, where node representations are iteratively updated via local neighborhood aggregation. Despite their success, message-passing suffers from fundamental limitations -- including constrained expressiveness, over-smoothing, over-squashing, and limited capacity to model long-range dependencies. These issues hinder scalability: increasing data size or model size often fails to yield improved performance. To this end, we explore pathways beyond message-passing and introduce Generative Graph Pattern Machine (G$^2$PM), a generative Transformer pre-training framework for graphs. G$^2$PM represents graph instances (nodes, edges, or entire graphs) as sequences of substructures, and employs generative pre-training over the sequences to learn generalizable and transferable representations. Empirically, G$^2$PM demonstrates strong scalability: on the ogbn-arxiv benchmark, it continues to improve with model sizes up to 60M parameters, outperforming prior generative approaches that plateau at significantly smaller scales (e.g., 3M). In addition, we systematically analyze the model design space, highlighting key architectural choices that contribute to its scalability and generalization. Across diverse tasks -- including node/link/graph classification, transfer learning, and cross-graph pretraining -- G$^2$PM consistently outperforms strong baselines, establishing a compelling foundation for scalable graph learning. The code and dataset are available at https://github.com/Zehong-Wang/G2PM.

Zehong Wang, Zheyuan Zhang, Tianyi Ma, Chuxu Zhang, Yanfang Ye• 2025

Related benchmarks

TaskDatasetResultRank
Node ClassificationPhoto
Mean Accuracy72.19
447
Node ClassificationRoman-Empire
Accuracy16.81
398
Link PredictionPubmed
AUC53.08
173
Node-level classificationFacebook
Accuracy54.68
65
Node ClassificationPubmed
Mean Accuracy65.63
38
Node ClassificationPhoto 1-shot (test)
Accuracy59.9
30
Node ClassificationPubmed 1-shot
Accuracy49.71
27
Node ClassificationCora 1-shot
Accuracy40.04
27
Node ClassificationCiteseer 1-shot
Accuracy31.04
27
Node ClassificationWisconsin 1-shot
Accuracy34.7
19
Showing 10 of 15 rows

Other info

Follow for update