Future Link Prediction Without Memory or Aggregation
Lu Yi, Runlin Lei, Fengran Mo, Yanping Zheng, Zhewei Wei, Yuhang Ye
Abstract
Future link prediction on temporal graphs is a fundamental task with wide applicability in real-world dynamic systems. These scenarios often involve both recurring (seen) and novel (unseen) interactions, requiring models to generalize effectively across both types of edges. However, existing methods typically rely on complex memory and aggregation modules, yet struggle to handle unseen edges. In this paper, we revisit the architecture of existing temporal graph models and identify two essential but overlooked modeling requirements for future link prediction: representing nodes with unique identifiers and performing target-aware matching between source and destination nodes. To this end, we propose Cross-Attention based Future Link Predictor on Temporal Graphs (CRAFT), a simple yet effective architecture that discards memory and aggregation modules and instead builds on two components: learnable node embeddings and cross-attention between the destination and the source's recent interactions. This design provides strong expressive power and enables target-aware modeling of the compatibility between candidate destinations and the source's interaction patterns. Extensive experiments on diverse datasets demonstrate that CRAFT consistently achieves superior performance with high efficiency, making it well-suited for large-scale real-world applications.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on12
- MLP-Mixer: An all-MLP Architecture for VisionIlya O. Tolstikhin, Neil Houlsby, Alexander Kolesnikov, Lucas Beyer et al.NeurIPS 2021 · 3,862 citations
- Inductive representation learning on temporal graphsDa Xu, Chuanwei Ruan, Evren Körpeoglu, Sushant Kumar et al.ICLR 2020 · 901 citations
- What graph neural networks cannot learn: depth vs widthAndreas LoukasICLR 2020 · 336 citations
- Inductive Representation Learning in Temporal Networks via Causal Anonymous WalksYanbang Wang, Yen-Yu Chang, Yunyu Liu, Jure Leskovec et al.ICLR 2021 · 326 citations
- Towards Better Dynamic Graph Learning: New Architecture and Unified LibraryLe Yu, Leilei Sun, Bowen Du, Weifeng LvNeurIPS 2023 · 323 citations
Related papers
- TAMI: Taming Heterogeneity in Temporal Interactions for Temporal Graph Link PredictionZhongyi Yu, Jianqiu Wu, Zhenghao Wu, Shuhan Zhong et al.NeurIPS 2025 · 3 citations
- On the Cross-Graph Transferability of Dynamic Link PredictionZhiqiang Pan, Chen Gao, Fei Cai, Wanyu Chen et al.WWW 2025 · 6 citations
- When Speed meets Accuracy: an Efficient and Effective Graph Model for Temporal Link PredictionHaoyang Li, Yuming Xu, Yiming Li, Hanmo Liu et al.VLDB 2025 · 1 citation
- FreeDyG: Frequency Enhanced Continuous-Time Dynamic Graph Model for Link PredictionYuxing Tian, Yiyan Qi, Fan GuoICLR 2024 · 58 citations
- Forecasting Interaction Order on Temporal GraphsWenwen Xia, Yuchen Li, Jianwei Tian, Shenghong LiKDD 2021 · 8 citations
