EasyDGL: Encode, Train and Interpret for Continuous-Time Dynamic Graph Learning

被引:0
|
作者
Chen, Chao [1 ,2 ]
Geng, Haoyu [1 ,2 ]
Yang, Nianzu [1 ,2 ]
Yang, Xiaokang [1 ,2 ]
Yan, Junchi [1 ,2 ,3 ]
机构
[1] Shanghai Jiao Tong Univ, Dept Comp Sci & Engn, Shanghai 200240, Peoples R China
[2] Shanghai Jiao Tong Univ, MoE Key Lab Artificial Intelligence, Shanghai 200240, Peoples R China
[3] Shanghai Jiao Tong Univ, Sch Artificial Intelligence, Shanghai 200030, Peoples R China
基金
中国国家自然科学基金;
关键词
Continuous-time dynamic graph; graph attention networks; graph signal processing; temporal point process;
D O I
10.1109/TPAMI.2024.3443110
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Dynamic graphs arise in various real-world applications, and it is often welcomed to model the dynamics in continuous time domain for its flexibility. This paper aims to design an easy-to-use pipeline (EasyDGL which is also due to its implementation by DGL toolkit) composed of three modules with both strong fitting ability and interpretability, namely encoding, training and interpreting: i) a temporal point process (TPP) modulated attention architecture to endow the continuous-time resolution with the coupled spatiotemporal dynamics of the graph with edge-addition events; ii) a principled loss composed of task-agnostic TPP posterior maximization based on observed events, and a task-aware loss with a masking strategy over dynamic graph, where the tasks include dynamic link prediction, dynamic node classification and node traffic forecasting; iii) interpretation of the outputs (e.g., representations and predictions) with scalable perturbation-based quantitative analysis in the graph Fourier domain, which could comprehensively reflect the behavior of the learned model. Empirical results on public benchmarks show our superior performance for time-conditioned predictive tasks, and in particular EasyDGL can effectively quantify the predictive power of frequency content that a model learns from evolving graph data.
引用
收藏
页码:10845 / 10862
页数:18
相关论文
empty
未找到相关数据