Link Prediction Model Based on Adversarial Graph Convolutional Network

被引：0
|
作者：
Tang C. [1 ]
Zhao J. [1 ]
Ye X. [1 ]
Yu S. [1 ]
机构：
[1] Faculty of Electrical Engineering and Computer Science, Ning-bo University, Ningbo
来源：
Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence | 2021年 / 34卷 / 02期
基金：
中国国家自然科学基金;
关键词：
Adversarial Network; Graph Convolutional Network; Hidden Space; Link Prediction;
D O I：
10.16451/j.cnki.issn1003-6059.202102001
中图分类号：
学科分类号：
摘要：
Most link prediction models rely too much on the known link information while mining node similarity. However, the number of the known observed links is small in the real world. To improve the robustness of the model, it is crucial to decouple the dependence of the model on the link information and mine the underlying features of nodes. In this paper, a link prediction model based on adversarial graph convolutional network is proposed with the consideration of the potential relationship between node features and links. Firstly, the similarity metric between nodes is utilized to fill in some unknown links in the adjacency matrix to alleviate the influence of link sparsity on the graph convolution model. Then, the adversarial network is employed to deeply mine the underlying connections between node features and links to reduce the dependence of the model on links. Experiments on real datasets show that the proposed model achieves better performance on link prediction problem and the performance remains relatively stable under link sparsity. Moreover, the proposed model is applicable to large-scale datasets. © 2021, Science Press. All right reserved.
引用
收藏
页码：95 / 105
页数：10
相关论文
共 38 条
  • [1] STANFIELD Z, COSKUN M, KOYUTURK M., Drug Response Prediction as a Link Prediction Problem, Scientific Reports, 7, 1, pp. 1-13, (2017)
  • [2] GUO L, MA J, CHEN Z M, Et al., Learning to Recommend with Social Contextual Information from Implicit Feedback, Soft Computing, 19, 5, pp. 1351-1362, (2015)
  • [3] LI X, CHEN H., Recommendation as Link Prediction in Bipartite Graphs: A Graph Kernel-Based Machine Learning Approach, Decision Support Systems, 54, 2, pp. 880-890, (2013)
  • [4] GENG X, LI Y G, WANG L Y, Et al., Spatiotemporal Multi-graph Convolution Network for Ride-Hailing Demand Forecasting, Proc of the AAAI Conference on Artificial Intelligence, pp. 3656-3663, (2019)
  • [5] GUO S N, LIN Y F, FENG N, Et al., Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting, Proc of the AAAI Conference on Artificial Intelligence, pp. 922-929, (2019)
  • [6] NEWMAN M E J., Clustering and Preferential Attachment in Growing Networks, Physical Review E, 64, (2001)
  • [7] LIBEN-NOWELL D, KLEINBERG J., The link-Prediction Problem for Social Networks, Journal of the American Society for Information Science and Technology, 58, 7, pp. 1019-1031, (2007)
  • [8] JEH G, WIDOM J., SimRank: A Measure of Structural-Context Si-milarity, Proc of the 8th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 538-543, (2002)
  • [9] AHMED N M, CHEN L, WANG Y L, Et al., DeepEye: Link Prediction in Dynamic Networks Based on Non-negative Matrix Factorization, Big Data Mining and Analytics, 1, 1, pp. 19-33, (2018)
  • [10] MENON A K, ELKAN C., Link Prediction via Matrix Factorization, Proc of the European Conference on Machine Learning and Knowledge Discovery in Databases, II, pp. 437-452, (2011)
1234