RLPath: a knowledge graph link prediction method using reinforcement learning based attentive relation path searching and representation learning

Ling Chen,Jun Cui,Xing Tang,Yuntao Qian,Yansheng Li,Yongjun Zhang
DOI: https://doi.org/10.1007/s10489-021-02672-0
IF: 5.3
2021-07-27
Applied Intelligence
Abstract:Due to containing rich patterns between entities, relation paths have been widely used in knowledge graph link prediction. The state-of-the-art link prediction methods considering relation paths obtain relation paths by reinforcement learning with an untrainable reward setting, and realize link prediction by path-ranking algorithm (PRA), which ignores information in entities. In this paper, we propose a new link prediction method RLPath to employ information in both relation paths and entities, which alternately trains a reinforcement learning model with a trainable reward setting to search high-quality relation paths, and a translation-based model to realize link prediction. Simultaneously, we propose a novel reward setting for the reinforcement learning model, which shares the parameters with the attention of the translation-based model, so that these parameters can not only measure the contributions of relation paths, but also guide agents to search relation paths that have high contributions for link prediction, forming mutual promotion. In experiments, we compare RLPath with the state-of-the-art link prediction methods. The results show that RLPath has competitive performance.
computer science, artificial intelligence
What problem does this paper attempt to address?