Link prediction by continuous spatiotemporal representation via neural differential equations

被引:0
作者
Huang, Liyi [1 ,2 ]
Pang, Bowen [2 ,3 ]
Yang, Qiming [2 ,3 ]
Feng, Xiangnan [4 ]
Wei, Wei [1 ,2 ,3 ,5 ]
机构
[1] Beihang Univ, Inst Artificial Intelligence, Beijing 100191, Peoples R China
[2] Minist Educ, Key Lab Math Informat & Behav Semant, Beijing, Peoples R China
[3] Beihang Univ, Sch Math Sci, Beijing 100191, Peoples R China
[4] Complex Sci Hub Vienna, A-1080 Vienna, Austria
[5] Zhongguancun Lab, Beijing 100094, Peoples R China
基金
中国国家自然科学基金;
关键词
Graph neural networks; Neural differential equation; Temporal link prediction; Dynamic network;
D O I
10.1016/j.knosys.2024.111619
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
With the continuous advancement of data science and machine learning, temporal link prediction has emerged as a crucial aspect of dynamic network analysis, providing significant research and application potential across various domains. While deep learning techniques have achieved remarkable results in temporal link prediction, most existing studies have focused on discrete model frameworks. These frameworks face limitations in capturing deep structural features and effectively aggregating temporal information. To address these limitations, we draw inspiration from neural differential equations to propose a Continuous Temporal Graph Neural Differential Equation (CTGNDE) network model for temporal link prediction. Specifically, we design a spatial graph Ordinary Differential Equation (ODE) to capture the spatial correlations inherent in complex spatiotemporal information. Then we employ Neural Controlled Differential Equation (Neural CDE) to learn complex evolution patterns and effectively aggregate temporal information. Finally, we characterize completely continuous and more accurate hidden state trajectories by coupling spatial and temporal messages. Experiments conducted on 10 real -world network datasets validated the superior performance of the CTGNDE model over the state-of-the-art baselines.
引用
收藏
页数:11
相关论文
共 35 条
  • [1] Friends and neighbors on the Web
    Adamic, LA
    Adar, E
    [J]. SOCIAL NETWORKS, 2003, 25 (03) : 211 - 230
  • [2] Bell M.G., 1997, TRANSPORTATION NETWO, DOI DOI 10.1002/9781118903032
  • [3] Line Graph Neural Networks for Link Prediction
    Cai, Lei
    Li, Jundong
    Wang, Jie
    Ji, Shuiwang
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2022, 44 (09) : 5103 - 5113
  • [4] Chen R. T. Q., 2018, Advances in Neural Information Processing Systems, V31
  • [5] Common neighbours and the local-community-paradigm for topological link prediction in bipartite networks
    Daminelli, Simone
    Thomas, Josephine Maria
    Duran, Claudio
    Cannistraci, Carlo Vittorio
    [J]. NEW JOURNAL OF PHYSICS, 2015, 17
  • [6] Demsar J, 2006, J MACH LEARN RES, V7, P1
  • [7] Temporal Link Prediction: A Survey
    Divakaran, Aswathy
    Mohan, Anuraj
    [J]. NEW GENERATION COMPUTING, 2020, 38 (01) : 213 - 258
  • [8] Dupont E, 2019, ADV NEUR IN, V32
  • [9] Spatial-Temporal Graph ODE Networks for Traffic Flow Forecasting
    Fang, Zheng
    Long, Qingqing
    Song, Guojie
    Xie, Kunqing
    [J]. KDD '21: PROCEEDINGS OF THE 27TH ACM SIGKDD CONFERENCE ON KNOWLEDGE DISCOVERY & DATA MINING, 2021, : 364 - 373
  • [10] A Novel Representation Learning for Dynamic Graphs Based on Graph Convolutional Networks
    Gao, Chao
    Zhu, Junyou
    Zhang, Fan
    Wang, Zhen
    Li, Xuelong
    [J]. IEEE TRANSACTIONS ON CYBERNETICS, 2023, 53 (06) : 3599 - 3612