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

GraphGPT: Generative Pre-trained Graph Eulerian Transformer

About

We introduceGraphGPT, a novel self-supervised generative pre-trained model for graph learning based on the Graph Eulerian Transformer (GET). First, we propose GET, which combines a standard transformer encoder or decoder architecture with an innovative graph-to-sequence transformation method. This method converts graphs or sampled subgraphs into sequences of tokens representing nodes, edges, and attributes in a reversible manner using Eulerian paths. We pre-train GET using either of the two self-supervised tasks: next-token prediction (NTP) and scheduled masked-token prediction (SMTP). The pre-trained model is then fine-tuned for downstream tasks such as graph-, edge-, and node-level prediction. Despite its simplicity, GraphGPT achieves performance comparable to or surpassing state-of-the-art methods on multiple large-scale Open Graph Benchmark (OGB) datasets. It demonstrates exceptional results on the molecular property prediction dataset PCQM4Mv2 and the protein-protein interaction dataset ogbl-ppa. Notably, generative pre-training enables scaling GraphGPT to 2 billion parameters while maintaining performance gains - a breakthrough that overcomes the scalability limitations of traditional Graph Neural Networks (GNNs) and prior graph transformers (GTs). To advance research in graph foundation models and facilitate scientific discovery in chemistry, materials science, and related fields, we will release the source code (https://github.com/alibaba/graph-gpt) and pre-trained checkpoints.

Qifang Zhao, Weidong Ren, Tianyu Li, Hong Liu, Xingsheng He, Xiaoxiao Xu• 2023

Related benchmarks

TaskDatasetResultRank
Node Classificationogbn-arxiv (test)
Accuracy72.1
542
Node ClassificationPhoto
Mean Accuracy72.44
447
Node ClassificationRoman-Empire
Accuracy15.38
398
Graph Classificationogbg-molpcba (test)
AP31.81
215
Link PredictionPubmed
AUC48.82
173
Link Predictionogbl-ppa (test)
Hits@1000.7655
77
Node-level classificationFacebook
Accuracy54.06
65
Node Classificationogbn-proteins (test)
ROC-AUC0.8533
62
Link Predictionogbl-citation2 (test)
MRR0.9305
57
Node ClassificationPubmed
Mean Accuracy64.55
38
Showing 10 of 17 rows

Other info

Code

Follow for update