解决时间图链接预测中交互异质性问题,提升稀疏交互对的预测效果。
TAMI: Taming Heterogeneity in Temporal Interactions for Temporal Graph Link Prediction
- 引入对数时间编码和历史聚合机制,平衡时间间隔并保留稀疏交互记录。
- 在13个经典数据集和3个新基准上,显著提升现有模型的链接预测准确率。
- 适用于需要精准预测长期或低频交互的场景,如社交网络、推荐系统。
时间图链接预测旨在基于节点历史交互预测未来互动,其核心依赖于节点嵌入。我们观察到时间交互存在天然异质性:少数节点对贡献了大量交互事件,且交互间隔差异显著。这导致现有方法难以有效编码时间信息,尤其在预测低频交互节点对时易遗忘历史。为此,本文提出TAMI框架,包含对数时间编码(LTE)与链接历史聚合(LHA)两个组件。LTE通过转换交互间隔使时间分布更均衡,LHA防止目标节点对的历史交互被遗忘。可无缝集成现有先进时间图神经网络,在13个经典数据集及3个最新时间图基准(TGB)上均实现跨归纳与直推设置的性能提升。代码已开源。
原文摘要 · Abstract (English)
Temporal graph link prediction aims to predict future interactions between nodes in a graph based on their historical interactions, which are encoded in node embeddings. We observe that heterogeneity naturally appears in temporal interactions, e.g., a few node pairs can make most interaction events, and interaction events happen at varying intervals. This leads to the problems of ineffective temporal information encoding and forgetting of past interactions for a pair of nodes that interact intermittently for their link prediction. Existing methods, however, do not consider such heterogeneity in their learning process, and thus their learned temporal node embeddings are less effective, especially when predicting the links for infrequently interacting node pairs. To cope with the heterogeneity, we propose a novel framework called TAMI, which contains two effective components, namely log time encoding function (LTE) and link history aggregation (LHA). LTE better encodes the temporal information through transforming interaction intervals into more balanced ones, and LHA prevents the historical interactions for each target node pair from being forgotten. State-of-the-art temporal graph neural networks can be seamlessly and readily integrated into TAMI to improve their effectiveness. Experiment results on 13 classic datasets and three newest temporal graph benchmark (TGB) datasets show that TAMI consistently improves the link prediction performance of the underlying models in both transductive and inductive settings. Our code is available at https://github.com/Alleinx/TAMI_temporal_graph.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。