论文标题

邻居增强图形卷积网络用于节点分类和建议

Neighbor Enhanced Graph Convolutional Networks for Node Classification and Recommendation

论文作者

Chen, Hao, Huang, Zhong, Xu, Yue, Deng, Zengde, Huang, Feiran, He, Peng, Li, Zhoujun

论文摘要

最近提出的图形卷积网络(GCN)在各种与图形相关的任务(例如节点分类和建议)上取得了显着卓越的性能。但是,当前对GCN模型的研究通常会递归地汇总来自所有邻居或随机采样的邻居子集的信息,而无需明确识别汇总的邻居是否在图形卷积期间提供了有用的信息。在本文中,我们理论上分析了邻居质量对GCN模型的性能的感情,并提出了邻居增强的图形卷积网络(NEGCN)框架,以提高现有GCN模型的性能。我们的贡献是三倍。首先,我们第一次提出了在一般理论框架中的节点分类和推荐任务的邻居质量概念。具体而言,对于节点分类,我们提出了三个命题,以理论上分析邻居质量如何影响GCN模型的节点分类性能。其次,根据提出的三个命题,我们介绍了图形完善过程,包括专门设计的邻居评估方法,以提高邻居质量,从而提高节点分类和建议任务。第三,我们对几个基准数据集进行了广泛的节点分类和建议实验。实验结果验证了我们提出的NEGCN框架可以显着提高节点分类和推荐任务上各种典型GCN模型的性能。

The recently proposed Graph Convolutional Networks (GCNs) have achieved significantly superior performance on various graph-related tasks, such as node classification and recommendation. However, currently researches on GCN models usually recursively aggregate the information from all the neighbors or randomly sampled neighbor subsets, without explicitly identifying whether the aggregated neighbors provide useful information during the graph convolution. In this paper, we theoretically analyze the affection of the neighbor quality over GCN models' performance and propose the Neighbor Enhanced Graph Convolutional Network (NEGCN) framework to boost the performance of existing GCN models. Our contribution is three-fold. First, we at the first time propose the concept of neighbor quality for both node classification and recommendation tasks in a general theoretical framework. Specifically, for node classification, we propose three propositions to theoretically analyze how the neighbor quality affects the node classification performance of GCN models. Second, based on the three proposed propositions, we introduce the graph refinement process including specially designed neighbor evaluation methods to increase the neighbor quality so as to boost both the node classification and recommendation tasks. Third, we conduct extensive node classification and recommendation experiments on several benchmark datasets. The experimental results verify that our proposed NEGCN framework can significantly enhance the performance for various typical GCN models on both node classification and recommendation tasks.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源