Mei Peng, Zhao Yu Hong
School of Information Engineering, Inner Mongolia University of Science and Technology, Baotou, 014010, China.
Sci Rep. 2024 Jan 4;14(1):538. doi: 10.1038/s41598-023-50977-6.
Dynamic network link prediction is extensively applicable in various scenarios, and it has progressively emerged as a focal point in data mining research. The comprehensive and accurate extraction of node information, as well as a deeper understanding of the temporal evolution pattern, are particularly crucial in the investigation of link prediction in dynamic networks. To address this issue, this paper introduces a node representation learning framework based on Graph Convolutional Networks (GCN), referred to as GCN_MA. This framework effectively combines GCN, Recurrent Neural Networks (RNN), and multi-head attention to achieve comprehensive and accurate representations of node embedding vectors. It aggregates network structural features and node features through GCN and incorporates an RNN with multi-head attention mechanisms to capture the temporal evolution patterns of dynamic networks from both global and local perspectives. Additionally, a node representation algorithm based on the node aggregation effect (NRNAE) is proposed, which synthesizes information including node aggregation and temporal evolution to comprehensively represent the structural characteristics of the network. The effectiveness of the proposed method for link prediction is validated through experiments conducted on six distinct datasets. The experimental outcomes demonstrate that the proposed approach yields satisfactory results in comparison to state-of-the-art baseline methods.
动态网络链接预测在各种场景中都有广泛应用,并且已逐渐成为数据挖掘研究的一个焦点。在动态网络的链接预测研究中,全面准确地提取节点信息以及更深入地理解时间演化模式尤为关键。为了解决这个问题,本文介绍了一种基于图卷积网络(GCN)的节点表示学习框架,称为GCN_MA。该框架有效地结合了GCN、循环神经网络(RNN)和多头注意力,以实现节点嵌入向量的全面准确表示。它通过GCN聚合网络结构特征和节点特征,并结合具有多头注意力机制的RNN,从全局和局部视角捕捉动态网络的时间演化模式。此外,还提出了一种基于节点聚合效应的节点表示算法(NRNAE),该算法综合了包括节点聚合和时间演化在内的信息,以全面表示网络的结构特征。通过在六个不同数据集上进行的实验验证了所提方法在链接预测方面的有效性。实验结果表明,与现有最先进的基线方法相比,所提方法取得了令人满意的结果。