Learnable Data Augmentation and Contrastive Pre-training for Temporal Link Prediction
Learnable Data Augmentation and Contrastive Pre-training for Temporal Link Prediction
Canghong Jin, Jiafeng Zhao, Feng Xu, Tongya Zheng, Zemin Liu, Lina Wei, Mingli Song
Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence
Main Track. Pages 2726-2734.
https://doi.org/10.24963/ijcai.2026/303
Link prediction is a foundational task in temporal graphs. While temporal graph neural networks exhibit commendable performance, they are often criticized for providing inadequate representations, especially under limited data. Contrastive learning has been introduced as a solution for graph pre-training to mitigate data scarcity. However, the data augmentation techniques on which they depend frequently lead to suboptimal augmentations, manifesting either as over-augmentation or under-augmentation.
In light of these challenges, we explore the under-explored domain of contrastive learning of temporal graph transformers and propose a novel model, ContraTGT, which employs a dual-view graph transformer. This transformer is fine-tuned to extract sequences tailored for specific target nodes, encapsulating both spatial and temporal perspectives. To optimize the contrastive learning of this dual-view transformer, we put forth an innovative, learnable data augmentation method. This technique, which involves selective masking of elements within dual-view sequences, generates superior augmentations, thereby amplifying the potency of the contrastive learning approach.
Extensive experiments on five public temporal network datasets demonstrate that our model can consistently outperform all baselines, especially in small training set conditions.
Keywords:
Data Mining: Mining graphs
Data Mining: Mining heterogenous data
