ConTIG: Continuous representation learning on temporal interaction graphs

No Thumbnail Available
File version
Author(s)
Wang, Zihui
Yang, Peizhen
Fan, Xiaoliang
Yan, Xu
Wu, Zonghan
Pan, Shirui
Chen, Longbiao
Zang, Yu
Wang, Cheng
Yu, Rongshan
Griffith University Author(s)
Primary Supervisor
Other Supervisors
Editor(s)
Date
2024
Size
File type(s)
Location
License
Abstract

Representation learning on temporal interaction graphs (TIG) aims to model complex networks with the dynamic evolution of interactions on a wide range of web and social graph applications. However, most existing works on TIG either (a) rely on discretely updated node embeddings merely when an interaction occurs that fail to capture the continuous evolution of embedding trajectories of nodes, or (b) overlook the rich temporal patterns hidden in the ever-changing graph data that presumably lead to sub-optimal models. In this paper, we propose a two-module framework named ConTIG, a novel representation learning method on TIG that captures the continuous dynamic evolution of node embedding trajectories. With two essential modules, our model exploits three-fold factors in dynamic networks including latest interaction, neighbor features, and inherent characteristics. In the first update module, we employ a continuous inference block to learn the nodes' state trajectories from time-adjacent interaction patterns using ordinary differential equations. In the second transform module, we introduce a self-attention mechanism to predict future node embeddings by aggregating historical temporal interaction information. Experiment results demonstrate the superiority of ConTIG on temporal link prediction, temporal node recommendation, and dynamic node classification tasks of four datasets compared with a range of state-of-the-art baselines, especially for long-interval interaction prediction.

Journal Title

Neural Networks

Conference Title
Book Title
Edition
Volume

172

Issue
Thesis Type
Degree Program
School
Publisher link
Patent number
Funder(s)
Grant identifier(s)
Rights Statement
Rights Statement
Item Access Status
Note
Access the data
Related item(s)
Subject

Artificial intelligence

Machine learning

Statistics

Graph embedding

Graph neural networks

Graph representation

Mining and learning in graphs

Temporal interaction graph

Persistent link to this record
Citation

Wang, Z; Yang, P; Fan, X; Yan, X; Wu, Z; Pan, S; Chen, L; Zang, Y; Wang, C; Yu, R, ConTIG: Continuous representation learning on temporal interaction graphs, Neural Networks, 2024, 172, pp. 106151

Collections