Liu Jie, Song Lingyun, Wang Guangtao, Shang Xuequn
School of Computer Science, Northwestern Polytechnical University, Xi'an, 710000, China; Key Laboratory of Big Data Storage and Management, Ministry of Industry and Information Technology, Northwestern Polytechnical University, Xi'an, 710000, China.
School of Computer Science, Northwestern Polytechnical University, Xi'an, 710000, China; Key Laboratory of Big Data Storage and Management, Ministry of Industry and Information Technology, Northwestern Polytechnical University, Xi'an, 710000, China.
Neural Netw. 2023 Jan;157:65-76. doi: 10.1016/j.neunet.2022.08.028. Epub 2022 Sep 22.
Heterogeneous information network embedding aims to learn low-dimensional node vectors in heterogeneous information networks (HINs), concerning not only structural information but also heterogeneity of diverse node and relation types. Most existing HIN embedding models mainly rely on metapath to define composite relations between node pairs and thus extract substructures from the original HIN. However, due to the pairwise structure of metapath, these models fail to capture the high-order relations (such as "Multiple authors co-authoring a paper") implicitly contained in HINs. To tackle the limitation, this paper proposes a Metapath-aware HyperGraph Transformer (Meta-HGT) for node embedding in HINs. Meta-HGT first extends metapath to guide the high-order relation extraction from original HIN and constructs multiple metapath based hypergraphs with diverse composite semantics. Then, Meta-HGT learns the latent node and hyperedge embeddings in each metapath based hypergraph through Meta-HGT layers. Each layer consists of two types of components, i.e., intra-hyperedge aggregation and inter-hyperedge aggregation, in which a novel type-dependent attention mechanism is proposed for node and hyperedge feature aggregation. Finally, it fuses multiple node embeddings learned from different metapath based hypergraphs via a semantic attention layer and generates the final node embeddings. Extensive experiments have been conducted on three HIN benchmarks for node classification. The results demonstrate that Meta-HGT achieves state-of-the-art performance on all three datasets.
异质信息网络嵌入旨在在异质信息网络(HIN)中学习低维节点向量,不仅考虑结构信息,还考虑不同节点和关系类型的异质性。大多数现有的HIN嵌入模型主要依靠元路径来定义节点对之间的复合关系,从而从原始HIN中提取子结构。然而,由于元路径的成对结构,这些模型无法捕捉HIN中隐含的高阶关系(如“多位作者共同撰写一篇论文”)。为了解决这一局限性,本文提出了一种用于HIN中节点嵌入的元路径感知超图变换器(Meta-HGT)。Meta-HGT首先扩展元路径以指导从原始HIN中提取高阶关系,并构建具有不同复合语义的多个基于元路径的超图。然后,Meta-HGT通过Meta-HGT层学习每个基于元路径的超图中的潜在节点和超边嵌入。每一层由两种类型的组件组成,即超边内聚合和超边间聚合,其中提出了一种新颖的依赖类型的注意力机制用于节点和超边特征聚合。最后,它通过语义注意力层融合从不同的基于元路径的超图中学习到的多个节点嵌入,并生成最终的节点嵌入。针对节点分类在三个HIN基准上进行了广泛的实验。结果表明,Meta-HGT在所有三个数据集上都取得了领先的性能。