1673-159X

CN 51-1686/N

融合权重和结构的加权非负矩阵分解的链路预测

Link Prediction Based on Weighted Structure and Weighted Nonnegative Matrix Factorization

  • 摘要: 链路预测的目标是根据已知网络结构信息去预测尚未连接的节点间形成链接的可能性。大部分现存链路预测方法仅关注无向无权网络,忽略自然权重与网络结构,从而导致预测精度下降。为此,文章提出一个加权非负矩阵分解(WNMF)的链路预测模型。该模型同时保持自然权重和加权网络局部结构。首先,将权重网络的邻接矩阵分解映射到低维潜在空间,以保持原始网络自然链接权重,然后将3个经典的加权共同邻居(WCN)、加权Adamic-Adar(WAA)和加权资源分配(WRA)作为指示矩阵分配给非负矩阵分解模型,以保持网络局部结构,并融合以上两类信息提出3个基于加权非负矩阵分解框架(WNMF框架)的链路预测模型:WNMF-WCN、WNMF-WAA和WNMF-WRA。此外,采用拉格朗日乘法规则学习所提3个模型参数。在6个真实世界加权网络上将现有链路预测模型与本文链路预测模型相比较,其结果表明,所提模型的PCC和Precision值最高可分别提升22.8%和23.5%。

     

    Abstract: The goal of link prediction is to predict missing links and possible future links based on known network structure information. However, most existing link prediction algorithms only focus on undirected and unweighted networks and ignores the natural weights and network structure which leads to the decrease of prediction accuracy. To address this problem, we propose a link prediction model of weighted non-negative matrix factorization to preserve natural weights and local structure of weighted network at the same time. Firstly, the adjacency matrix decomposition of the weighted network is mapped to a low-dimensional latent space to preserve the natural link weights of the original network, and then three classical weighted common neighbors (WCN), weighted Adamic-Adar (WAA) and weighted resource allocation (WRA) are used as indicators and matrix is assigned to the non-negative matrix factorization model to maintain the local structure of the network, and the above two types of information are fused to propose three link prediction models based on the weighted non-negative matrix factorization (WNMF) framework, namely WNMF-WCN, WNMF-WAA and WNMF-WRA . Furthermore, the lagrange multiplication rule is enabled to learn the proposed three model parameters. Compared with existing link prediction methods on 6 real-world weighted networks, the experimental results show that the proposed model improves the PCC and Precision values by 22.8% and 23.5%, respectively.

     

/

返回文章
返回