Multi-hop Knowledge Graph Reasoning Based on Hyperbolic Knowledge Graph Embedding and Reinforcement Learning

被引:6
作者
Zhou, Xingchen [1 ]
Wang, Peng [1 ]
Luo, Qiqing [1 ]
Pan, Zhe [1 ]
机构
[1] Southeast Univ, Sch Comp Sci & Engn, Nanjing, Peoples R China
来源
PROCEEDINGS OF THE 10TH INTERNATIONAL JOINT CONFERENCE ON KNOWLEDGE GRAPHS (IJCKG 2021) | 2021年
关键词
knowledge graph reasoning; reinforcement learning; knowledge graph embedding;
D O I
10.1145/3502223.3502224
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Some unseen facts in knowledge graphs (KGs) can be complemented by multi-hop knowledge graph reasoning. The process of multi-hop reasoning can be presented as a serialized decision problem, and then be solved with reinforcement learning (RL). However, existing RL-based reasoning methods cannot deal with the hierarchical information in KGs effectively. To solve this issue, we propose a novel RL-based multi-hop KG reasoning model Path Additional Actionspace Ranking (PAAR). Our model first proposes a hyperbolic knowledge graph embedding (KGE) model which can effectively capture hierarchical information in KGs. To introduce hierarchical information into RL, the hyperbolic KGE vector is subsequently added to the state space and helps to expand the action space. In order to alleviate the reward sparsity problem in RL, our model utilizes the score function of hyperbolic KGE model as a soft reward. Finally, the metric of hyperbolic space is added to the training of RL as a penalty strategy to constrain the sufficiency of multi-hop reasoning paths. Experimental results on two real-world datasets show that our proposed model not only provides effective answers but also offers sufficient paths.
引用
收藏
页码:1 / 9
页数:9
相关论文
共 28 条
  • [1] Poincare ball embeddings of the optical geometry
    Abramowicz, MA
    Bengtsson, I
    Karas, V
    Rosquist, K
    [J]. CLASSICAL AND QUANTUM GRAVITY, 2002, 19 (15) : 3963 - 3976
  • [2] Alberto Nicolas Usunier, 2013, ADV NEURAL INF PROCE, P2787
  • [3] DBpedia: A nucleus for a web of open data
    Auer, Soeren
    Bizer, Christian
    Kobilarov, Georgi
    Lehmann, Jens
    Cyganiak, Richard
    Ives, Zachary
    [J]. SEMANTIC WEB, PROCEEDINGS, 2007, 4825 : 722 - +
  • [4] Balazevic I, 2019, ADV NEUR IN, V32
  • [5] Bollacker K, 2008, P 2008 ACM SIGMOD IN, P1247, DOI 10.1145/1376616.1376746
  • [6] Stochastic Gradient Descent on Riemannian Manifolds
    Bonnabel, Silvere
    [J]. IEEE TRANSACTIONS ON AUTOMATIC CONTROL, 2013, 58 (09) : 2217 - 2229
  • [7] Chami I, 2019, ADV NEUR IN, V32
  • [8] Cohen W, 2011, P 2011 C EMPIRICAL M, P529
  • [9] Das R, 2018, P 6 INT C LEARNING R
  • [10] Das R, 2017, 15TH CONFERENCE OF THE EUROPEAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS (EACL 2017), VOL 1: LONG PAPERS, P132