基于元路径的动态异质网络表示学习  

Dynamic Heterogeneous Network Representation Method Based on Meta-Path

作　　者：刘群[1] 谭洪胜 张优敏 王国胤[1] LIU Qun;TAN Hong-sheng;ZHANG You-min;WANG Guo-yin(College of Computer Science and Technology,Chongqing University of Posts and Telecommunications,Chongqing 400065,China)

机构地区：[1]重庆邮电大学计算机科学与技术学院,重庆400065

出　　处：《电子学报》2022年第8期1830-1839,共10页Acta Electronica Sinica

基　　金：国家自然科学基金重点项目(No.61936001);重庆市教委重点合作项目(No.HZ2021008);重庆市自然科学基金(No.cstc2021ycjh-bg-zxm0013)。

摘　　要：对网络表示学习的研究已经取得了很多成果,但是大部分网络表示学习模型忽略了网络动态性和异质性,无法区分网络中耦合的时间和空间(结构)特征,也不能捕获网络的丰富语义信息.本文提出了基于元路径的动态异质网络表示学习方法.首先将节点的邻域结构按照时间划分出不同的子空间结构,并为每个节点采样出所有时间加权元路径的序列.其次通过门控循环单元将节点的全部时间加权元路径序列上的邻域信息进行集成,最后利用带注意力机制的双向门控循环单元对融合后的节点序列进行时空上下文信息学习,获得每个节点的最终表示向量.通过在真实数据集上的实验表明,在节点分类、聚类和可视化的下游任务测试中,本文提出的算法较基线方法在性能上均有较大提升.节点分类任务中的Micro-F1平均提高了1.09%~3.72%,节点聚类任务中的ARI值提高了3.23%~14.49%.The researches of network representation learning have made many achievements.Since most of the researches ignore the dynamics and heterogeneity of the networks,coupled temporal and spatial structure features can not be distinguished,and rich semantic information of the network cannot be captured well.In this paper,meta-path based dynamic heterogeneous network representation learning method is proposed.Firstly,the neighborhood structures of nodes are divided into different sub-spaces according to their time,then the sequences of all time-weighted meta-paths for each node are sampled.Secondly,the neighborhood information on all time-weighted meta-paths of each node is integrated by a gated recurrent unit network(GRU).Furthermore,a bi-directional gated recurrent unit network(Bi-GRU)with an attention mechanism is used to learn the spatio-temporal contextual information from the merged sequences,and the final node representation will be received.Experiments on real data sets show that our algorithm has greatly improved performance on the downstream network tasks,such as node classification,clustering and visualization.Compared with state-of-the-art baseline methods,the Micro-F1 value has been raised by 1.09%~3.72%averagely on classification tasks,and the ARI value has been increased by 3.23%~14.49%on clustering tasks.

关 键 词：网络表示学习 动态异质网络 元路径 注意力机制 门控循环单元 

分 类 号：TP391[自动化与计算机技术—计算机应用技术]