On the Scalability of Temporal Relative Positional Encoding for Dynamic Link Prediction

The combination of temporal graph neural networks (TGNNs) and relative positional features is effective for dynamic link prediction tasks because it improves the modeling of relationships between nodes. We summarize the positional feature as Temporal Relative Positional Encoding (TRPE), which incorporates local subgraph computation and temporal decay effects compared to RPE in static graph learning. However, existing dynamic graph learning methods with TRPE are limited by high computational costs and poor scalability due to information loss in neighborhood compression. To address this, we introduce a scalable framework that enhances TRPE computation by integrating it with temporal clusters. Our method replaces high-order relative position information with cluster-based computation to reduce computation costs. Each node is assigned to a cluster based on its structural role, rather than just randomly sketched, reducing information loss during neighborhood compression. TGNNs with our proposed framework are more expressive than existing dynamic graph learning methods and offer greater scalability than existing TRPE models. Our experimental results on seven standard temporal link prediction benchmarks demonstrate that our proposed model achieves comparable or superior performance compared to state-of-the-art models.