English

CPDG: A Contrastive Pre-Training Method for Dynamic Graph Neural Networks

Machine Learning 2023-12-27 v3 Social and Information Networks

Abstract

Dynamic graph data mining has gained popularity in recent years due to the rich information contained in dynamic graphs and their widespread use in the real world. Despite the advances in dynamic graph neural networks (DGNNs), the rich information and diverse downstream tasks have posed significant difficulties for the practical application of DGNNs in industrial scenarios. To this end, in this paper, we propose to address them by pre-training and present the Contrastive Pre-Training Method for Dynamic Graph Neural Networks (CPDG). CPDG tackles the challenges of pre-training for DGNNs, including generalization capability and long-short term modeling capability, through a flexible structural-temporal subgraph sampler along with structural-temporal contrastive pre-training schemes. Extensive experiments conducted on both large-scale research and industrial dynamic graph datasets show that CPDG outperforms existing methods in dynamic graph pre-training for various downstream tasks under three transfer settings.

Keywords

Cite

@article{arxiv.2307.02813,
  title  = {CPDG: A Contrastive Pre-Training Method for Dynamic Graph Neural Networks},
  author = {Yuanchen Bei and Hao Xu and Sheng Zhou and Huixuan Chi and Haishuai Wang and Mengdi Zhang and Zhao Li and Jiajun Bu},
  journal= {arXiv preprint arXiv:2307.02813},
  year   = {2023}
}

Comments

14 pages, 8 figures, accepted by ICDE2024

R2 v1 2026-06-28T11:23:25.736Z