arXiv Machine Learning
22h ago

MambaTG: Candidate-Aware State Space Modeling for Future Link Prediction on Temporal Graphs

MambaTG is a new framework for future link prediction on temporal graphs that addresses both repeated and unseen edge prediction. It leverages two complementary signals: the historical-sequence evolution of a node’s interactions and the compatibility between a candidate edge and the node’s evolving history. By using a candidate-aware Mamba architecture, the model captures ordered interaction dynamics and adapts state transitions to different candidates, achieving strong performance across 13 datasets while remaining computationally efficient.

By Nguyen Hong Quan, Sitao Luan, Yilun Zheng, Lihui Chen
Hugging Face Trending Papers
Sep 28

Do Temporal Link Predictors Need Learned Memory? A Smoothed-Count Baseline with a Handful of Parameters

The paper investigates whether temporal link predictors can forgo learned node representations in favor of simple statistical counts of interaction patterns. It introduces a predictor that aggregates transition and co‑occurrence counts, smooths them with destination frequencies or Kneser‑Ney continuation counts, and combines these with popularity, source history, and recency via a shared log‑linear rule. With only 9–13 learned parameters, the model achieves the best mean reciprocal rank on 7 of 16 datasets and outperforms several baselines across all evaluated datasets, demonstrating that a lightweight, count‑based approach can rival more complex neural methods.