Skip to main navigation Skip to search Skip to main content

Multi-hop knowledge graph reasoning in few-shot scenarios

  • Shangfei Zheng
  • , Wei Chen
  • , Weiqing Wang
  • , Pengpeng Zhao
  • , Hongzhi Yin
  • , Lei Zhao

Research output: Contribution to journalArticleResearchpeer-review

Abstract

Reinforcement learning (RL)-based multi-hop reasoning has become an interpretable way for knowledge graph reasoning owing to its persuasive explanations for the predicted results, but the reasoning performance of these methods drops significantly over few-shot relations (only contain few triplets). To address this problem, recent studies introduce meta-learning into RL-based reasoning methods. However, the performance of these studies is still limited due to the following points: (1) the overall reasoning accuracy is impaired due to the low reasoning accuracies over some hard relations; (2) the reasoning process becomes laborious and ineffective owing to the existence of noisy data; (3) the generalizability is negatively affected due to the lack of knowledge-sharing. To tackle these challenges, we propose a novel model HMLS consisting of two modules HHML (Hierarchical Hardness-aware Meta-reinforcement Learning) and HHS (Hierarchical Hardness-aware Sampling). Specifically, HHML contains the following two components: (1) a hardness-aware RL conducts multi-hop reasoning by training hardness-aware batches and reducing noise; (2) a knowledge-sharing meta-learning adapts to few-shot relations by exploiting common features in the hierarchical relation structure. The other module HHS generates hardness-aware batches from relation and relation-cluster levels. The experimental results demonstrate that this work notably outperforms the state-of-the-art approaches in few-shot scenarios.

Original languageEnglish
Pages (from-to)1713-1727
Number of pages14
JournalIEEE Transactions on Knowledge and Data Engineering
Volume36
Issue number4
DOIs
Publication statusPublished - Apr 2024

Keywords

  • Adaptation models
  • Cognition
  • Few-shot relations
  • hard sample mining
  • Knowledge graphs
  • meta-reinforcement learning
  • Metalearning
  • multi-hop knowledge graph reasoning
  • Reinforcement learning
  • Task analysis
  • Training

Cite this