A link prediction method for heterogeneous networks based on BP neural network

被引:105
作者
Li, Ji-chao [1 ]
Zhao, Dan-ling [1 ]
Ge, Bing-Feng [1 ]
Yang, Ke-Wei [1 ]
Chen, Ying-Wu [1 ]
机构
[1] Natl Univ Def Technol, Coll Syst Engn, Changsha 410073, Hunan, Peoples R China
基金
中国国家自然科学基金;
关键词
Link prediction; Heterogeneous networks; Meta-path; BP neural network;
D O I
10.1016/j.physa.2017.12.018
中图分类号
O4 [物理学];
学科分类号
0702 ;
摘要
Most real-world systems, composed of different types of objects connected via many interconnections, can be abstracted as various complex heterogeneous networks. Link prediction for heterogeneous networks is of great significance for mining missing links and reconfiguring networks according to observed information, with considerable applications in, for example, friend and location recommendations and disease-gene candidate detection. In this paper, we put forward a novel integrated framework, called MPBP (Meta-Path feature-based BP neural network model), to predict multiple types of links for heterogeneous networks. More specifically, the concept of meta-path is introduced, followed by the extraction of meta-path features for heterogeneous networks. Next, based on the extracted meta-path features, a supervised link prediction model is built with a three-layer BP neural network. Then, the solution algorithm of the proposed link prediction model is put forward to obtain predicted results by iteratively training the network. Last, numerical experiments on the dataset of examples of a gene-disease network and a combat network are conducted to verify the effectiveness and feasibility of the proposed MPBP. It shows that the MPBP with very good performance is superior to the baseline methods. (C) 2017 Elsevier B.V. All rights reserved.
引用
收藏
页码:1 / 17
页数:17
相关论文
共 37 条
[1]  
[Anonymous], PROCEDIA ENG
[2]  
[Anonymous], 2011, P 17 ACM SIGKDD INT, DOI [DOI 10.1145/2020408.2020581, 10.1145/2020408.2020581]
[3]  
[Anonymous], 2011, SURVEY LINK PREDICTI
[4]  
[Anonymous], PHYS REV E
[5]  
[Anonymous], 2012, ACM International Conference on Web Search and Data Mining
[6]  
[Anonymous], 2011, P 2011 SIAM INT C DA, DOI DOI 10.1137/1.9781611972818.5
[7]  
[Anonymous], 2012, P 5 ACM INT C WEB SE, DOI DOI 10.1145/2124295.2124373
[8]  
Backstrom L, 2011, P 4 ACM INT C WEB SE, P635, DOI [DOI 10.1145/1935826.1935914, 10.1145/1935826.1935914]
[9]   Efficiency of attack strategies on complex model and real-world networks [J].
Bellingeri, Michele ;
Cassi, Davide ;
Vincenzi, Simone .
PHYSICA A-STATISTICAL MECHANICS AND ITS APPLICATIONS, 2014, 414 :174-180
[10]   Collective Prediction of Multiple Types of Links in Heterogeneous Information Networks [J].
Cao, Bokai ;
Kong, Xiangnan ;
Yu, Philip S. .
2014 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2014, :50-59