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

GraphFormers: GNN-nested Transformers for Representation Learning on Textual Graph

About

The representation learning on textual graph is to generate low-dimensional embeddings for the nodes based on the individual textual features and the neighbourhood information. Recent breakthroughs on pretrained language models and graph neural networks push forward the development of corresponding techniques. The existing works mainly rely on the cascaded model architecture: the textual features of nodes are independently encoded by language models at first; the textual embeddings are aggregated by graph neural networks afterwards. However, the above architecture is limited due to the independent modeling of textual features. In this work, we propose GraphFormers, where layerwise GNN components are nested alongside the transformer blocks of language models. With the proposed architecture, the text encoding and the graph aggregation are fused into an iterative workflow, {making} each node's semantic accurately comprehended from the global perspective. In addition, a {progressive} learning strategy is introduced, where the model is successively trained on manipulated data and original data to reinforce its capability of integrating information on graph. Extensive evaluations are conducted on three large-scale benchmark datasets, where GraphFormers outperform the SOTA baselines with comparable running efficiency.

Junhan Yang, Zheng Liu, Shitao Xiao, Chaozhuo Li, Defu Lian, Sanjay Agrawal, Amit Singh, Guangzhong Sun, Xing Xie• 2021

Related benchmarks

TaskDatasetResultRank
Node ClassificationarXiv
Accuracy72.27
219
Node ClassificationDBLP
Accuracy92.6
67
Node ClassificationProducts
Accuracy84.18
56
Molecular Property ClassificationMoleculeNet BBBP
ROC AUC70.12
56
Molecular property predictionBACE
ROC-AUC76.95
55
Node Classificationogbn-proteins
Accuracy72.17
35
Link PredictionDBLP (test)--
22
RegressionMoleculeNet Lipophilicity
RMSE1.2367
21
Molecular property predictionMoleculeNet ESOL
RMSE0.926
15
Category RetrievalAmazon Economics (test)
Recall@5024.48
15
Showing 10 of 45 rows

Other info

Code

Follow for update