arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04282cs.SI
收藏

通过丰富的节点属性信息提升网络嵌入算法的链接预测准确率

Improving link prediction accuracy of network embedding algorithms via rich node attribute information

Weiwei Gu, Jinqiang Hou, Weiyi Gu

AI总结:

针对现有基于网络嵌入的链接预测算法未充分利用节点属性信息的问题,提出属性图增强嵌入算法AGEE,通过构建属性网络接入典型算法并自动学习结构与属性网络的权重权衡,可将链接预测准确率较现有基准提升约3%。

摘要:

复杂网络被广泛用于表征从社交网络到脑网络等大量现实世界中的关系。基于当前观测到的网络推断缺失链接或预测未来链接的任务被称为链接预测任务。近期基于网络嵌入的链接预测算法已在链接预测准确率上展现出突破性性能。这类算法通常将节点属性作为初始特征输入,以加快训练过程中的收敛速度,但它们并未充分利用节点特征信息。本文中,除了将特征属性作为初始输入外,我们通过构建属性网络并将其接入若干典型链接预测算法,进一步优化了节点属性信息的利用,将该算法命名为Attributive Graph Enhanced Embedding(AGEE,属性图增强嵌入)。AGEE能够自动学习结构网络与属性网络之间的权重权衡。数值实验表明,与链接预测框架SEAL、Variational Graph AutoEncoder(VGAE,变分图自编码器)以及Node2vec相比,AGEE可将链接预测准确率提升约3%。

英文摘要:

Complex networks are widely used to represent an abundance of real-world relations ranging from social networks to brain networks. Inferring missing links or predicting future ones based on the currently observed network is known as the link prediction task.Recent network embedding based link prediction algorithms have demonstrated ground-breaking performance on link prediction accuracy. Those algorithms usually apply node attributes as the initial feature input to accelerate the convergence speed during the training process. However, they do not take full advantage of node feature information. In this paper,besides applying feature attributes as the initial input, we make better utilization of node attribute information by building attributable networks and plugging attributable networks into some typical link prediction algorithms and naming this algorithm Attributive Graph Enhanced Embedding (AGEE). AGEE is able to automatically learn the weighting trades-off between the structure and the attributive networks. Numerical experiments show that AGEE can improve the link prediction accuracy by around 3% compared with link prediction framework SEAL, Variational Graph AutoEncoder (VGAE), and Node2vec.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑