arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.02643cs.SIcs.LG
收藏

基于注意力和元路径隐式利用的GCN for HIN

GCN for HIN via Implicit Utilization of Attention and Meta-paths

Tianjin University(天津大学);Emory University(埃默里大学);University of Illinois at Chicago(伊利诺伊大学芝加哥分校);University of Illinois at Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校)

Di Jin, Zhizhi Yu, Dongxiao He, Carl Yang, Philip S. Yu, Jiawei Han

摘要:

异构信息网络(HIN)嵌入,旨在映射的结构和语义信息在HIN的分布式表示,引起了相当大的研究关注。用于HIN嵌入的图神经网络通常采用分层注意(包括节点级和元路径级注意)来捕获来自基于元路径的邻居的信息。然而,这种复杂的注意力结构往往由于严重的过拟合而无法实现元路径选择的功能。此外,当传播信息时,这些方法不区分直接(单跳)元路径和间接(多跳)元路径。但从网络科学的角度来看,直接关系往往被认为是更重要的,它只能用来模拟直接的信息传播。针对这些局限性,本文提出了一种隐式利用注意力和元路径的神经网络方法,该方法可以有效缓解HIN中过度参数化的注意力机制带来的严重过拟合问题。首先采用多层图卷积网络(GCN)框架,在每一层进行判别式聚合,并将直接连接的元路径的信息传播逐层叠加,实现了间接选择元路径的注意力功能。然后,我们通过引入一个新的传播操作,可以从聚集分离,给出了一个有效的放松和改进。也就是说,我们首先用定义良好的概率扩散动力学对整个传播过程进行建模,然后引入一个基于随机图的约束,使其能够随着层的增加而减少噪声。大量的实验证明了新方法的优越性比国家的最先进的方法。

英文摘要:

Heterogeneous information network (HIN) embedding, aiming to map the structure and semantic information in a HIN to distributed representations, has drawn considerable research attention. Graph neural networks for HIN embeddings typically adopt a hierarchical attention (including node-level and meta-path-level attentions) to capture the information from meta-path-based neighbors. However, this complicated attention structure often cannot achieve the function of selecting meta-paths due to severe overfitting. Moreover, when propagating information, these methods do not distinguish direct (one-hop) meta-paths from indirect (multi-hop) ones. But from the perspective of network science, direct relationships are often believed to be more essential, which can only be used to model direct information propagation. To address these limitations, we propose a novel neural network method via implicitly utilizing attention and meta-paths, which can relieve the severe overfitting brought by the current over-parameterized attention mechanisms on HIN. We first use the multi-layer graph convolutional network (GCN) framework, which performs a discriminative aggregation at each layer, along with stacking the information propagation of direct linked meta-paths layer-by-layer, realizing the function of attentions for selecting meta-paths in an indirect way. We then give an effective relaxation and improvement via introducing a new propagation operation which can be separated from aggregation. That is, we first model the whole propagation process with well-defined probabilistic diffusion dynamics, and then introduce a random graph-based constraint which allows it to reduce noise with the increase of layers. Extensive experiments demonstrate the superiority of the new approach over state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑