新模型统一处理已知与未知节点,提升动态图学习的准确性与泛化能力。
Trajectory Encoding Temporal Graph Networks
- 用可自动扩展的节点ID作为时序位置特征,捕捉节点历史轨迹
- 在三个真实数据集上,链接预测和节点分类均超越现有基线
- 适合需要兼顾已知节点精度与新节点泛化的动态图任务
动态图网络(TGN)在链接预测和节点分类等任务中表现优异,涵盖需预测已知节点间关系的归纳设置,以及需推广至未见节点的转导设置。现有设计在此双重场景下存在困境:仅依赖时序与结构信息的匿名模型虽具备强归纳能力,却难以区分已知节点;而利用节点特征的非匿名模型虽在转导任务中表现良好,但无法适应新节点。为此,本文提出轨迹编码图网络(TETGN),引入可自动扩展的节点标识符作为可学习的时序位置特征,并通过消息传递机制建模节点历史上下文。结合多头注意力机制,该方法有效平衡了转导精度与归纳泛化能力。在三个真实世界数据集上的实验表明,TETGN在链接预测和节点分类任务中显著优于多个强基线模型,实现了匿名与非匿名模型优势的统一。
原文摘要 · Abstract (English)
Temporal Graph Networks (TGNs) have demonstrated significant success in dynamic graph tasks such as link prediction and node classification. Both tasks comprise transductive settings, where the model predicts links among known nodes, and in inductive settings, where it generalises learned patterns to previously unseen nodes. Existing TGN designs face a dilemma under these dual scenarios. Anonymous TGNs, which rely solely on temporal and structural information, offer strong inductive generalisation but struggle to distinguish known nodes. In contrast, non-anonymous TGNs leverage node features to excel in transductive tasks yet fail to adapt to new nodes. To address this challenge, we propose Trajectory Encoding TGN (TETGN). Our approach introduces automatically expandable node identifiers (IDs) as learnable temporal positional features and performs message passing over these IDs to capture each node's historical context. By integrating this trajectory-aware module with a standard TGN using multi-head attention, TETGN effectively balances transductive accuracy with inductive generalisation. Experimental results on three real-world datasets show that TETGN significantly outperforms strong baselines on both link prediction and node classification tasks, demonstrating its ability to unify the advantages of anonymous and non-anonymous models for dynamic graph learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。