计算机科学与探索 ›› 2023, Vol. 17 ›› Issue (9): 2198-2208.DOI: 10.3778/j.issn.1673-9418.2206102

• 人工智能·模式识别 • 上一篇    下一篇

融合多元信息的社交网络节点分类方法

刘超,梁安婷,刘小洋,黄贤英   

  1. 重庆理工大学 计算机科学与工程学院,重庆 400054
  • 出版日期:2023-09-01 发布日期:2023-09-01

Social Network Nodes Classification Method Based on Multi-information Fusion

LIU Chao, LIANG Anting, LIU Xiaoyang, HUANG Xianying   

  1. School of Computer Science and Engineering, Chongqing University of Technology, Chongqing 400054, China
  • Online:2023-09-01 Published:2023-09-01

摘要: 针对社交网络节点分类效果不佳的问题,提出一种融合多元信息的图卷积网络节点分类模型(IMIGCN)。首先,利用特征向量X和邻接矩阵A,分别构造包含节点间同质信息的同质矩阵FA和共引信息的共引矩阵CoA;分析网络中的三角结构,通过转换公式构造包含节点间三角信息的三角矩阵TriA。在此基础上,融入节点自身信息。接着,改进传统图卷积网络(GCN)模型。将GCN的单核改进为适应性多核,通过注意力机制将多核学习的结果自适应融合为一个嵌入,达到一次卷积同时融合多元信息的效果。为了学习更多信息,将模型过程中的嵌入设计为多头,通过多头嵌入注意力自适应学习多头嵌入的权重分配。实验结果表明,与现有较优的节点分类模型相比,提出的IMIGCN在社交网络上的分类精确度提高0.009 8~0.053 2,F1指标提升0.012 7~0.053 6,证明了提出的IMIGCN合理有效。

关键词: 社交网络, 信息提取, 信息融合, 节点分类, 图卷积网络(GCN)

Abstract: For the problem of poor performance of social network nodes, a model of integrating multiple information graph convolutional networks (IMIGCN) for node classification is proposed. Firstly, using the eigenvector X and the adjacency matrix A, this paper respectively constructs the homogeneous matrix FA containing the homogeneous information between nodes and the co-citation matrix CoA containing the co-citation information. The triangular structure in the network is analyzed, and the triangular matrix TriA containing the triangular information between nodes is constructed by the transformation formula. On this basis, the information of the node itself is integrated. Next, this paper improves the traditional graph convolutional networks (GCN) model. The single kernel of GCN is improved to adaptive multi-kernel, and the results of multi-kernel learning are adaptively fused into one embedding through the attention mechanism, so as to achieve the effect of integrating multiple information at the same time by one convolution. In order to learn more information, the embedding in the model process is designed as multi-head, and the weight assignment of multi-head embedding is adaptively learned through multi-head embedding attention. Experimental results show that, compared with the existing node classification models with better performance, the classification accuracy of the proposed IMIGCN on social networks is improved by 0.0098 to 0.0532, and the F1 index is improved by 0.0127 to 0.0536, which proves that the proposed IMIGCN is reasonable and effective.

Key words: social network, information extraction, information fusion, node classification, graph convolutional network (GCN)