Entity-Relation Guided Random Walk for Link Prediction in Knowledge Graphs

Weisheng Li, Hao Zhong, Ronghua Lin, Chao Chang, Zhihong Pan, Yong Tang

Abstract: Knowledge graphs (KGs) are structured knowledge bases that represent information as a collection of interconnected entities and relations. Link prediction in KGs aims to infer missing or potential links between entities based on triple facts. Among different link prediction methods, knowledge graph embedding (KGE) has gained widespread popularity, with the goal of learning low-dimensional representations for KGs. However, most present KGE methods struggle to capture both local and global neighborhood information efficiently. Additionally, many hybrid methods have limitations in modeling and capturing interactions between triples. In this article, we propose an entity-relation-guided random walk (ERGRW) method for link prediction in KGs. Unlike conventional approaches that solely focus on entity-based walks, ERGRW creatively introduces relations as objects to walk as well. Inspired by distance-based methods, we design novel random walk rules based on the translation principle within triples. Thus, the ERGRW not only captures local and global neighborhood information but also discovers potential semantic relationships and interactions in the KGs. Furthermore, the encoder–decoder framework of ERGRW is able to learn comprehensive representation and improve link prediction performance. Extensive experiments conducted on four standard datasets demonstrate the superiority of ERGRW for link prediction.