Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Universal Graph Transformer Self-Attention Networks

About

We introduce a transformer-based GNN model, named UGformer, to learn graph representations. In particular, we present two UGformer variants, wherein the first variant (publicized in September 2019) is to leverage the transformer on a set of sampled neighbors for each input node, while the second (publicized in May 2021) is to leverage the transformer on all input nodes. Experimental results demonstrate that the first UGformer variant achieves state-of-the-art accuracies on benchmark datasets for graph classification in both inductive setting and unsupervised transductive setting; and the second UGformer variant obtains state-of-the-art accuracies for inductive text classification. The code is available at: \url{https://github.com/daiquocnguyen/Graph-Transformer}.

Dai Quoc Nguyen, Tu Dinh Nguyen, Dinh Phung• 2019

Related benchmarks

TaskDatasetResultRank
Graph ClassificationPROTEINS
Accuracy78.53
742
Graph ClassificationMUTAG
Accuracy89.97
697
Graph ClassificationCOLLAB
Accuracy77.84
329
Graph ClassificationIMDB-B
Accuracy77.04
322
Graph ClassificationIMDB-M
Accuracy53.6
218
Graph ClassificationDD
Accuracy80.23
175
Graph ClassificationPTC
Accuracy69.63
167
Text ClassificationMR (test)
Accuracy79.29
99
Text ClassificationR8 (test)
Accuracy97.05
56
Document ClassificationOhsumed (test)
Accuracy70.63
54
Showing 10 of 20 rows

Other info

Code

Follow for update