• •    

CCML2021+286: 基于深度神经网路和门控循环单元的动态图表示学习方法

李慧博1,赵云霄1,白亮2   

  1. 1. 山西大学
    2. 山西大学计算机与信息技术学院
  • 收稿日期:2021-06-10 修回日期:2021-06-28 发布日期:2021-06-28
  • 通讯作者: 李慧博

CCML2021+286: Dynamic graph representation learning method based on deep neural network and gated recurrent unit

  • Received:2021-06-10 Revised:2021-06-28 Online:2021-06-28

摘要: 学习图中节点的潜在向量表示是一项重要且普遍存在的任务,旨在捕捉图中节点的各种属性。大量工作证明静态图表示已经能够学习到节点的部分信息,然而,真实世界的图是随着时间的推移而演变的。为了解决多数动态网络算法不能有效保留节点邻域结构和时态信息的问题,本文提出了基于深度神经网路(DNN)和门控循环单元(GRU)的动态网络表示学习方法:DynAEGRU。该算法以自编码器作为框架,首先编码器使用深度神经网络聚集邻域信息,得到低维特征向量,其次使用GRU网络提取节点时态信息,最后解码器重构邻接矩阵并与真实图对比构建损失。通过该模型与8种图表示算法在3个数据集上实验分析,结果表明DynAEGRU模型具有较好的性能增益。

关键词: 动态网络表示学习, 深度神经网络, 自编码器, 门控循环单元, 链路预测

Abstract: Learning the latent vector representation of nodes in the graph is an important and ubiquitous task, which aims to capture various attributes of the nodes in the graph. A lot of work proves that static graphs represent part of the node information that has been learned. However, real-world graphs evolve over time. In order to solve the problem that most dynamic network algorithms cannot effectively retain node neighborhood structure and temporal information, this paper proposes a dynamic network representation learning method based on deep neural network (DNN) and gated recurrent unit (GRU): DynAEGRU. The algorithm uses the autoencoder as the framework. First, the encoder uses a deep neural network to gather neighborhood information to obtain low-dimensional feature vectors. Secondly, the GRU network is used to gather node temporal information, and the decoder reconstructs the adjacency matrix and compares it with the real graph to construct the loss. Through the experimental analysis of this model and 8 graph representation learning algorithms on 3 datasets, the results show that the DynAEGRU model has a good performance.

Key words: Keywords: dynamic network representation learning, deep neural network, autoencoder, gated recurrent unit, link prediction

中图分类号: