期刊
KNOWLEDGE-BASED SYSTEMS
卷 232, 期 -, 页码 -出版社
ELSEVIER
DOI: 10.1016/j.knosys.2021.107431
关键词
Hierarchical attention; Neural network; Link prediction
The proposed HalpNet model utilizes a hierarchical attention mechanism to comprehensively explore neighborhood information and achieve effective link prediction, demonstrating significant superiority over existing approaches on various benchmark datasets.
In this paper, a novel end-to-end neural link prediction model, named Hierarchical Attention Link Prediction Neural Network (HalpNet), is proposed. HalpNet comprehensively explores neighborhood information, which has proved important for link prediction, via the core component, i.e., hierarchical attention mechanism. The proposed hierarchical attention mechanism consists of two neural attention layers, modeling crucial structure information at node level and subgraph level, respectively. At node level, a structure-preserving attention is developed to preserve structure features of each node in the neighborhood subgraph. Based on the latent node features, at subgraph level, a structure-aggregating attention is designed to learn how important that each node in the subgraph is for the linkage of the target node pair and aggregate node features with learned attentions as a comprehensive subgraph representation. Given this expressive representation of neighborhood subgraph, HalpNet is able to predict link score of target node pair effectively. We evaluate HalpNet on 8 benchmark datasets against 14 popular and state-of-the-art approaches. The experimental results demonstrate its significant superiority and wide applicability on link prediction problem. (c) 2021 Elsevier B.V. All rights reserved.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据