Reason more like human: Incorporating meta information into hierarchical reinforcement learning for knowledge graph reasoning

Yi Xia,Junyong Luo,Mingjing Lan,Gang Zhou,Zhibo Li,Shuo Liu
DOI: https://doi.org/10.1007/s10489-022-04147-2
IF: 5.3
2022-10-09
Applied Intelligence
Abstract:Nowadays, reasoning over knowledge graphs (KGs) has been widely adapted to empower retrieval systems, recommender systems, and question answering systems, generating a surge in research interest. However, recently developed reasoning methods usually lack interpretability, and can hardly tackle the large-scale action space problem over KGs. Inspired by the ability of human hierarchical decision making, we propose a multi-hop reasoning framework with deep reinforcement learning (RL) to fill this gap, which incorporates meta information into hierarchical reasoning over KGs. We first use optimization-based meta learning method to initialize parameters for RL agents, allowing for efficient adaptation for tasks in a few gradient steps. Then, a hierarchical RL framework is designed to decompose reasoning tasks into several sub-tasks and solve them separately, performed more efficient and natural than directly solving the entire problem. We further evaluated our model through different tasks on five real world datasets. The experimental results indicate that our method outperforms state-of-the-art baseline models without losing interpretability.
computer science, artificial intelligence
What problem does this paper attempt to address?