Research on Link Prediction Model Based on Hierarchical Attention Mechanism

ZHAO Xiaojuan,JIA Yan,LI Aiping,CHEN Kai
DOI: https://doi.org/10.11959/j.issn.1000-436x.2021057
2021-01-01
Abstract:In order to solve the problem that the existing graph attention mechanism tends to cause attention distribution to certain relations with high frequency when performing link prediction related tasks, a new link prediction model based on hierarchical attention mechanism was proposed.In the link prediction task, a hierarchical attention mechanism was designed to give different attention to the relationships of different relationship types connected to a given entity in the knowledge graph according to the relationship in the prediction task.While the characteristics of multi-hop neighbor entities were pay attention to, the relationship characteristics was pay more attention to find the relationship type that matches the target relationship.Through comparison experiments with the mainstream models on multiple benchmark data sets, the results show that the performance of the model is better than the mainstream models and has good robustness.
What problem does this paper attempt to address?