LLM2D
轨迹编码时序图网络
Trajectory Encoding Temporal Graph Networks
作者: Jiafeng Xiong, Rizos Sakellariou
发布日期: 4/16/2025
arXiv ID: oai:arXiv.org:2504.11386v1

摘要

arXiv:2504.11386v1 Announce Type: cross 摘要:时间图网络(TGNs)在链接预测和节点分类等动态图任务中展现了显著的成功。这两个任务包括归纳设置,模型在已知节点之间预测链接,以及_transductive_设置,模型将学习到的模式泛化到以前未见过的节点。现有的TGN设计在这两种场景下遇到了困境。匿名TGNs仅依赖于时间和结构信息,提供了强大的归纳泛化能力,但难以区分已知节点。相反,非匿名TGNs利用节点特征在_transductive_任务中表现出色,但在适应新节点方面失败。为了解决这一挑战,我们提出了一种轨迹编码TGN(TETGN)。我们的方法引入了自动生成可扩展的节点标识符(IDs)作为可学习的时间位置特征,并通过这些标识符进行消息传递,以捕捉每个节点的历史上下文。通过将此轨迹感知模块与标准TGN结合使用并采用多头注意力机制,TETGN有效地平衡了_transductive_准确性与归纳泛化能力。在三个真实世界数据集上的实验结果表明,TETGN在链接预测和节点分类任务中显著优于强基线模型,证明了其将匿名和非匿名模型的优势统一起来进行动态图学习的能力。