基于联合注意力机制的动态图表示学习模型

    Dynamic graph representation learning model based on joint attention mechanism

    • 摘要: 动态图广泛存在于实际应用场景中,对其进行表示学习可以有效地应用于各种下游分析任务。动态图的拓扑结构和节点特征随时间不断演变,节点之间存在结构相关性和特征相关性。现有研究忽略了以上两种相关性对动态图表示学习的影响,针对这一问题,本文提出基于联合注意力机制的动态图表示学习模型(DyGRL-JAM),从动态拓扑结构中学习结构相关性,并结合特征相关性构造联合注意力,使用图张量卷积实现节点时空消息传递获得节点表示。在六个动态图数据上进行实验,结果表明DyGRL-JAM对动态图的表示学习能力优于当前先进方法。

       

      Abstract: Dynamic graphs widely exist in practical application scenarios, and their representation learning can be effectively applied to various downstream analysis tasks. The topological structures and node features of dynamic graphs are evolving, and there are structural correlations and feature correlations between nodes. Existing research ignores the impact of the above two types of correlation on dynamic graph representation learning. To address this problem, a dynamic graph representation learning model based on a joint attention mechanism(DyGRL-JAM) is proposed in this paper. DyGRL-JAM learns structural correlations from dynamic topological structures and combines them with feature correlations to construct the joint attention. Then, a graph tensor convolution is adopted to implement node spatiotemporal message passing to obtain node representation. Experiments are conducted on six dynamic graph datasets, and the results show that DyGRL-JAM has better representation learning capabilities for dynamic graphs than state-of-the-art methods.

       

    /

    返回文章
    返回