Relation Prediction Based on Source-Entity Behavior Preference Modeling Via Heterogeneous Graph Pooling

Yashen Wang,Xiaoling Zhu,Huanhuan Zhang
DOI: https://doi.org/10.1007/978-3-031-10983-6_33
2022-01-01
Abstract:Recent years have witnessed great advance of representation learning (RL) based models for the knowledge graph relation prediction task. Most existing approaches represent graph nodes as vectors in a low-dimensional embedding space, ignoring the entity behavior preference and the beneficial semantic interactions in the real-world graphs. To address this challenge, this paper proposes a novel relation prediction model based on source-entity behavior preference modeling, which represents each source-entity as a heterogeneous graph released from their structure and semantic perspectives to better capture the behavior relatedness. Especially, a heterogeneous graph pooling method is leveraged for learning source-entity behavior embedding representation from this personalized heterogeneous graph. In our comprehensive experiments, we evaluate our model on real-world graphs, and the results demonstrate that the proposed model significantly outperforms existing state-of-the-art methods on benchmark datasets for the relation prediction and knowledge graph completion task.
What problem does this paper attempt to address?