Journal of Computer Applications ›› 2019, Vol. 39 ›› Issue (6): 1632-1638.DOI: 10.11772/j.issn.1001-9081.2018112279

• Artificial intelligence • Previous Articles     Next Articles

Link prediction model based on densely connected convolutional network

WANG Wentao, WU Lintao, HUANG Ye, ZHU Rongbo   

  1. College of Computer Science, South-Central University for Nationalities, Wuhan Hubei 430074, China
  • Received:2018-11-15 Revised:2019-01-18 Online:2019-06-10 Published:2019-06-17
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61772562), the Fundamental Research Funds for the Central Universities of South-Central University for Nationalities (CZY18014), the Academic Innovation Research Fund for Graduates of South-Central University for Nationalities (3212018hqzz029).

基于密集连接卷积神经网络的链路预测模型

王文涛, 吴淋涛, 黄烨, 朱容波   

  1. 中南民族大学 计算机科学学院, 武汉 430074
  • 通讯作者: 吴淋涛
  • 作者简介:王文涛(1967-),男,河北邯郸人,副教授,博士,主要研究方向:计算机网络与控制;吴淋涛(1994-),男,湖南茶陵人,硕士研究生,主要研究方向:计算机网络、数据挖掘;黄烨(1993-),男,湖北大悟人,硕士研究生,主要研究方向:知识表示、神经网络;朱容波(1978-),男,湖北潜江人,教授,博士,主要研究方向:移动计算、无线网络协议设计与性能优化。
  • 基金资助:
    国家自然科学基金资助项目(61772562);中南民族大学中央高校基本科研业务费专项基金资助项目(CZY18014);中南民族大学研究生学术创新基金后期资助项目(3212018hqzz029)。

Abstract: The current link prediction algorithms based on network representation learning mainly construct feature vectors by capturing the neighborhood topology information of network nodes for link prediction. However, those algorithms usually only focus on learning information from the single neighborhood topology of network nodes, while ignore the researches on similarity between multiple nodes in link structure. Aiming at these problems, a new Link Prediction model based on Densely connected convolutional Network (DenseNet-LP) was proposed. Firstly, the node representation vectors were generated by the network representation learning algorithm called node2vec, and the structure information of the network nodes was mapped into three dimensional feature information by these vectors. Then, DenseNet was used to to capture the features of link structure and establish a two-category classification model to realize link prediction. The experimental results on four public datasets show that, the Area Under the Receiver Operating Characteristic (ROC) Curve (AUC) value of the prediction result of the proposed model is increased by up to 18 percentage points compared to the result of network representation learning algorithm.

Key words: link prediction, network representation learning, node representation, convolutional neural network, deep learning

摘要: 现有的基于网络表示学习的链路预测算法主要通过捕获网络节点的邻域拓扑信息构造特征向量来进行链路预测,该类算法通常只注重从网络节点的单一邻域拓扑结构中学习信息,而对多个网络节点在链路结构上的相似性方面研究不足。针对此问题,提出一种基于密集连接卷积神经网络(DenseNet)的链路预测模型(DenseNet-LP)。首先,利用基于网络表示学习算法node2vec生成节点表示向量,并利用该表示向量将网络节点的结构信息映射为三维特征数据;然后,利用密集连接卷积神经网络来捕捉链路结构的特征,并建立二分类模型实现链路预测。在四个公开的数据集上的实验结果表明,相较于网络表示学习算法,所提模型链路预测结果的ROC曲线下方面积(AUC)值最大提高了18个百分点。

关键词: 链路预测, 网络表示学习, 节点表示, 卷积神经网络, 深度学习

CLC Number: