Disentangling Node Metric Factors for Temporal Link Prediction

Tianli Zhang,Tongya Zheng,Yuanyu Wan,Ying Li,Wenqi Huang
DOI: https://doi.org/10.1007/978-981-99-8082-6_27
2024-01-01
Abstract:Temporal Link Prediction (TLP), as one of the highly concerned tasks in graph mining, requires predicting the future link probability based on historical interactions. On the one hand, traditional methods based on node metrics, such as Common Neighbor, achieve satisfactory performance in the TLP task. On the other hand, node metrics overly focus on the global impact of nodes while neglecting the personalization of different node pairs, which can sometimes mislead link prediction results. However, mainstream TLP methods follow the standard paradigm of learning node embedding, entangling favorable and harmful node metric factors in the representation, reducing the model's robustness. In this paper, we propose a plug-and-play plugin called Node Metric Disentanglement, which can apply to most TLP methods and boost their performance. It explicitly accounts for node metrics and disentangles them from the embedding representations generated by TLP methods. We adopt the attention mechanism to reasonably select information conducive to the TLP task and integrate it into the node embedding. Experiments on various state-of-the-art methods and dynamic graphs verify the effectiveness and universality of our NMD plugin.
What problem does this paper attempt to address?