摘要
学习连续时间动态图(CTDG)的有效表示是一个挑战,因为需要同时跨越节点交互的长时间历史并捕捉细微的时间细节。具体而言,出现了两个问题:(1)对更长时间历史进行编码需要更多的计算资源,因此CTDG模型必须保持低计算复杂度以确保效率;(2)同时,需要更强大的模型来识别和选择更长时间历史提供的扩展上下文中最重要的时序信息。为了解决这些问题,我们提出了一种名为DyGMamba的CTDG表示学习模型,该模型源于流行的Mamba状态空间模型(SSM)。DyGMamba首先利用节点级SSM对历史节点交互序列进行编码。然后,另一个时间级SSM被用来利用隐藏在历史图中的时间模式,其输出被用来动态地从交互历史中选择关键信息。我们在动态链接预测任务上对DyGMamba进行了实验验证。结果表明,我们的模型在大多数情况下都达到了最先进的水平。DyGMamba在计算资源方面也保持了较高的效率,这使得在有限的计算预算下捕捉长时间依赖关系成为可能。