MTdyg: Multi-scale transformers with continuous time dynamic graph model for link prediction

被引:0
|
作者
Xu, Long [1 ]
Pan, Zhiqiang [1 ]
Chen, Honghui [1 ]
Wang, Shen [1 ]
机构
[1] Natl Key Lab Informat Syst Engn, Changsha 410000, Peoples R China
关键词
Link prediction; Continuous time dynamic graph; Multi-scale patches; Transformer;
D O I
10.1016/j.knosys.2024.112245
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Link prediction through continuous dynamic graph neural networks is a challenging endeavour. Previous studies have considered historic interaction sequences among pairs of nodes. However, this approach does not sufficiently model the links between them. Thus, the frequency of historical common neighbours was proposed to obtain more information about node pairs. Nonetheless, this measure fails to consider the timing of interactions because the relative importance of different interactions shifts over time. Multi-scale transformers with a continuous time dynamic graph model (MTdyg) are our response to this challenge to enhance link prediction. The novel MTdyg model employs a multi-scale patching strategy that dynamically adjusts based on the interaction frequency, feeding segmented interaction sequences into the transformer. This methodology significantly improves the model's ability to assimilate historical data. Furthermore, we developed a temporal attention-based historical common neighbour encoding technique to effectively identify linkages between source and target nodes in interaction sequences. Extensive testing on eight public datasets confirmed that MTdyg delivers superior performance on most datasets, demonstrating its effectiveness in capturing the nuances of node interactions over time.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] A Multi-Scale Approach for Graph Link Prediction
    Cai, Lei
    Ji, Shuiwang
    [J]. THIRTY-FOURTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, THE THIRTY-SECOND INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE AND THE TENTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2020, 34 : 3308 - 3315
  • [2] Multi-Scale Variational Graph AutoEncoder for Link Prediction
    Guo, Zhihao
    Wang, Feng
    Yao, Kaixuan
    Liang, Jiye
    Wang, Zhiqiang
    [J]. WSDM'22: PROCEEDINGS OF THE FIFTEENTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2022, : 334 - 342
  • [3] Multi-scale Fusion Dynamic Graph Neural Network for Traffic Flow Prediction
    Weng, Wenchao
    Chen, Qikai
    Dai, Yu
    Chen, Jingyang
    Chen, Dongliang
    [J]. ACM International Conference Proceeding Series, 2023, : 85 - 90
  • [4] Multi-scale Subgraph Contrastive Learning for Link Prediction
    Sun, Shilin
    Zhang, Zehua
    Wang, Runze
    Tian, Hua
    [J]. ROUGH SETS, IJCRS 2022, 2022, 13633 : 217 - 223
  • [5] Multi-Scale Dynamic Graph Learning for Time Series Anomaly Detection (Student Abstract)
    Jin, Yixuan
    Wei, Yutao
    Cheng, Zhangtao
    Tai, Wenxin
    Xiao, Chunjing
    Zhong, Ting
    [J]. THIRTY-EIGTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE, VOL 38 NO 21, 2024, : 23523 - 23524
  • [6] Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction
    Gao, Ming
    Du, Zhuoran
    Qin, Hongmao
    Wang, Wei
    Jin, Guangyin
    Xie, Guotao
    [J]. Knowledge-Based Systems, 2024, 305
  • [7] Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction
    Hu, Na
    Zhang, Dafang
    Xie, Kun
    Liang, Wei
    Li, Kuan-Ching
    Zomaya, Albert Y.
    [J]. FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2024, 158 : 323 - 332
  • [8] A dynamic multi-scale Markov model based methodology for remaining life prediction
    Yan, Jihong
    Guo, Chaozhong
    Wang, Xing
    [J]. MECHANICAL SYSTEMS AND SIGNAL PROCESSING, 2011, 25 (04) : 1364 - 1376
  • [9] Multi-Scale Dynamic Convolutional Network for Knowledge Graph Embedding
    Zhang, Zhaoli
    Li, Zhifei
    Liu, Hai
    Xiong, Neal N.
    [J]. IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2022, 34 (05) : 2335 - 2347
  • [10] Multi-scale time-stepping of Partial Differential Equations with transformers
    Hemmasian, Amirpouya
    Farimani, Amir Barati
    [J]. COMPUTER METHODS IN APPLIED MECHANICS AND ENGINEERING, 2024, 426