Improving Few-Shot Inductive Learning on Temporal Knowledge Graphs using Confidence-Augmented Reinforcement Learning

by   Zifeng Ding, et al.

Temporal knowledge graph completion (TKGC) aims to predict the missing links among the entities in a temporal knwoledge graph (TKG). Most previous TKGC methods only consider predicting the missing links among the entities seen in the training set, while they are unable to achieve great performance in link prediction concerning newly-emerged unseen entities. Recently, a new task, i.e., TKG few-shot out-of-graph (OOG) link prediction, is proposed, where TKGC models are required to achieve great link prediction performance concerning newly-emerged entities that only have few-shot observed examples. In this work, we propose a TKGC method FITCARL that combines few-shot learning with reinforcement learning to solve this task. In FITCARL, an agent traverses through the whole TKG to search for the prediction answer. A policy network is designed to guide the search process based on the traversed path. To better address the data scarcity problem in the few-shot setting, we introduce a module that computes the confidence of each candidate action and integrate it into the policy for action selection. We also exploit the entity concept information with a novel concept regularizer to boost model performance. Experimental results show that FITCARL achieves stat-of-the-art performance on TKG few-shot OOG link prediction.


page 1

page 2

page 3

page 4


Few-Shot Inductive Learning on Temporal Knowledge Graphs using Concept-Aware Information

Knowledge graph completion (KGC) aims to predict the missing links among...

MDistMult: A Multiple Scoring Functions Model for Link Prediction on Antiviral Drugs Knowledge Graph

Knowledge graphs (KGs) on COVID-19 have been constructed to accelerate t...

Learning Meta Representations of One-shot Relations for Temporal Knowledge Graph Link Prediction

Few-shot relational learning for static knowledge graphs (KGs) has drawn...

Adaptive Pseudo-Siamese Policy Network for Temporal Knowledge Prediction

Temporal knowledge prediction is a crucial task for the event early warn...

ReInform: Selecting paths with reinforcement learning for contextualized link prediction

We propose to use reinforcement learning to inform transformer-based con...

PubGraph: A Large Scale Scientific Temporal Knowledge Graph

Research publications are the primary vehicle for sharing scientific pro...

Exploring the Limits of Few-Shot Link Prediction in Knowledge Graphs

Real-world knowledge graphs are often characterized by low-frequency rel...

Please sign up or login with your details

Forgot password? Click here to reset