WWW 2019

Universal Graph Transformer Self-Attention Networks

D. Q. Nguyen, T. Nguyen, Dinh Q. Phung

Main figure for Universal Graph Transformer Self-Attention Networks

Abstract

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

Proposed method: Universal Graph Transformer

Benchmark Results

Tested on 1 dataset

Tables show presented results in global context (best 3 papers + GCN + MLP).

TU Dortmund

#ModelAccuracy ↑Paper
1
MSH-GNN
0.9910 ±0.0030
2
ID-GNN+BEC
0.9830 ±0.0360
3
SupCosine
0.9830 ±0.0250
39
GIN
0.9330 ±0.0290
198
Universal Graph Transformer
0.8997 ±0.0365
This paper
403
MLP
0.8660 ±0.0495
SOTA performance (not significantly different from the best) Proposed method (this paper) Global context (other papers)