GC-LSTM: graph convolution embedded LSTM for dynamic network link prediction

被引:77
|
作者
Chen, Jinyin [1 ,2 ]
Wang, Xueke [2 ]
Xu, Xuanheng [2 ]
机构
[1] Zhejiang Univ Technol, Inst Cyberspace Secur, Hangzhou, Peoples R China
[2] Zhejiang Univ Technol, Coll Informat Engn, Hangzhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Dynamic network link prediction; Graph convolution network; Long short-term memory network; Network embedding; Deep learning;
D O I
10.1007/s10489-021-02518-9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Dynamic network link prediction is becoming a hot topic in network science, due to its wide applications in biology, sociology, economy and industry. However, it is a challenge since network structure evolves with time, making long-term prediction of adding/deleting links especially difficult. Inspired by the great success of deep learning frameworks, especially the convolution neural network (CNN) and long short-term memory (LSTM) network, we propose a novel end-to-end model with a Graph Convolution Network(GCN) embedded LSTM, named GC-LSTM, for dynamic network link prediction. Thereinto, LSTM is adopted as the main framework to learn the temporal features of all snapshots of a dynamic network. While for each snapshot, GCN is applied to capture the local structural properties of nodes as well as the relationship between them. One benefit is that our GC-LSTM can predict both added and removed links, making it more practical in reality, while most existing dynamic link prediction methods can only handle removed links. Extensive experiments demonstrated that GC-LSTM achieves outstanding performance and outperforms existing state-of-the-art methods.
引用
收藏
页码:7513 / 7528
页数:16
相关论文
共 50 条
  • [21] An LSTM network for highway trajectory prediction
    [J]. Altche, Florent (florent.altche@mines-paristech.fr), 1600, Institute of Electrical and Electronics Engineers Inc., United States (2018-March):
  • [22] Efficient Link Prediction with Clusterized Batch Training in Graph Convolution Network
    Khadka E.
    Adhikari N.B.
    [J]. Journal of The Institution of Engineers (India): Series B, 2023, 104 (03) : 693 - 701
  • [23] An LSTM Network for Highway Trajectory Prediction
    Altche, Florent
    de La Fortelle, Arnaud
    [J]. 2017 IEEE 20TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION SYSTEMS (ITSC), 2017,
  • [24] A Dynamic Heterogeneous Graph Convolution Network For Traffic Flow Prediction
    Li, He
    Jin, Duo
    Li, XueJiao
    Qiao, Shaojie
    [J]. COMPUTER JOURNAL, 2024, 67 (01): : 31 - 44
  • [25] Survey of Dynamic Graph Neural Network for Link Prediction
    Zhang, Qi
    Chen, Xu
    Wang, Shuyang
    Jing, Yongjun
    Song, Jifei
    [J]. Computer Engineering and Applications, 60 (20): : 49 - 67
  • [26] An Improved Convolutional LSTM Network with Directional Convolution Layer for Prediction of Water Quality with Spatial Information
    Zhao, Ziqi
    Geng, Yuxin
    Ni, Qingjian
    [J]. ADVANCES IN SWARM INTELLIGENCE, ICSI 2022, PT II, 2022, : 94 - 105
  • [27] Microwave Link Failures Prediction via LSTM-based Feature Fusion Network
    Ruan, Zichan
    Yang, Shuiqiao
    Pan, Lei
    Ma, Xingjun
    Luo, Wei
    Grobler, Marthie
    [J]. 2021 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2021,
  • [28] Social graph convolutional LSTM for pedestrian trajectory prediction
    Zhou, Yutao
    Wu, Huayi
    Cheng, Hongquan
    Qi, Kunlun
    Hu, Kai
    Kang, Chaogui
    Zheng, Jie
    [J]. IET INTELLIGENT TRANSPORT SYSTEMS, 2021, 15 (03) : 396 - 405
  • [29] A Directed Acyclic Graph Network Combined With CNN and LSTM for Remaining Useful Life Prediction
    Li, Jialin
    Li, Xueyi
    He, David
    [J]. IEEE ACCESS, 2019, 7 : 75464 - 75475
  • [30] A Heterogeneous Graph Enhanced LSTM Network for Hog Price Prediction Using Online Discussion
    Ye, Kai
    Piao, Yangheran
    Zhao, Kun
    Cui, Xiaohui
    [J]. AGRICULTURE-BASEL, 2021, 11 (04):