Share your thoughts, 1 month free Claude Pro on usSee more
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
994
Graph ClassificationMUTAG
Accuracy89.97
862
Graph ClassificationCOLLAB
Accuracy77.84
422
Graph ClassificationIMDB-B
Accuracy77.04
378
Graph ClassificationIMDB-M
Accuracy53.6
275
Graph ClassificationDD
Accuracy80.23
273
Graph ClassificationPTC
Accuracy69.63
167
Text ClassificationMR (test)
Accuracy79.29
148
Text ClassificationR8 (test)
Accuracy97.05
56
Document ClassificationOhsumed (test)
Accuracy70.63
54
Showing 10 of 21 rows

Other info

Code

Follow for update