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

GraphAIR: Graph Representation Learning with Neighborhood Aggregation and Interaction

About

Graph representation learning is of paramount importance for a variety of graph analytical tasks, ranging from node classification to community detection. Recently, graph convolutional networks (GCNs) have been successfully applied for graph representation learning. These GCNs generate node representation by aggregating features from the neighborhoods, which follows the "neighborhood aggregation" scheme. In spite of having achieved promising performance on various tasks, existing GCN-based models have difficulty in well capturing complicated non-linearity of graph data. In this paper, we first theoretically prove that coefficients of the neighborhood interacting terms are relatively small in current models, which explains why GCNs barely outperforms linear models. Then, in order to better capture the complicated non-linearity of graph data, we present a novel GraphAIR framework which models the neighborhood interaction in addition to neighborhood aggregation. Comprehensive experiments conducted on benchmark tasks including node classification and link prediction using public datasets demonstrate the effectiveness of the proposed method.

Fenyu Hu, Yanqiao Zhu, Shu Wu, Weiran Huang, Liang Wang, Tieniu Tan• 2019

Related benchmarks

TaskDatasetResultRank
Link PredictionCiteseer
AUC95
146
Link PredictionPubmed
AUC99.2
123
Link PredictionCora
AUC0.954
116
Transductive Node ClassificationPubmed (transductive)
Accuracy80
95
Transductive Node ClassificationCora (transductive)
Accuracy84.7
72
Transductive Node ClassificationCiteseer (transductive)
Accuracy73.5
61
Inductive Node ClassificationPPI (inductive)
Accuracy98.6
14
Transductive Node ClassificationNELL
Accuracy68.5
9
Showing 8 of 8 rows

Other info

Code

Follow for update