Selective Temporal Knowledge Graph Reasoning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Hou, Zhongni, Jin, Xiaolong, Li, Zixuan, Bai, Long, Guo, Jiafeng, Cheng, Xueqi
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910395774533632
author Hou, Zhongni
Jin, Xiaolong
Li, Zixuan
Bai, Long
Guo, Jiafeng
Cheng, Xueqi
author_facet Hou, Zhongni
Jin, Xiaolong
Li, Zixuan
Bai, Long
Guo, Jiafeng
Cheng, Xueqi
contents Temporal Knowledge Graph (TKG), which characterizes temporally evolving facts in the form of (subject, relation, object, timestamp), has attracted much attention recently. TKG reasoning aims to predict future facts based on given historical ones. However, existing TKG reasoning models are unable to abstain from predictions they are uncertain, which will inevitably bring risks in real-world applications. Thus, in this paper, we propose an abstention mechanism for TKG reasoning, which helps the existing models make selective, instead of indiscriminate, predictions. Specifically, we develop a confidence estimator, called Confidence Estimator with History (CEHis), to enable the existing TKG reasoning models to first estimate their confidence in making predictions, and then abstain from those with low confidence. To do so, CEHis takes two kinds of information into consideration, namely, the certainty of the current prediction and the accuracy of historical predictions. Experiments with representative TKG reasoning models on two benchmark datasets demonstrate the effectiveness of the proposed CEHis.
format Preprint
id arxiv_https___arxiv_org_abs_2404_01695
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Selective Temporal Knowledge Graph Reasoning
Hou, Zhongni
Jin, Xiaolong
Li, Zixuan
Bai, Long
Guo, Jiafeng
Cheng, Xueqi
Machine Learning
Temporal Knowledge Graph (TKG), which characterizes temporally evolving facts in the form of (subject, relation, object, timestamp), has attracted much attention recently. TKG reasoning aims to predict future facts based on given historical ones. However, existing TKG reasoning models are unable to abstain from predictions they are uncertain, which will inevitably bring risks in real-world applications. Thus, in this paper, we propose an abstention mechanism for TKG reasoning, which helps the existing models make selective, instead of indiscriminate, predictions. Specifically, we develop a confidence estimator, called Confidence Estimator with History (CEHis), to enable the existing TKG reasoning models to first estimate their confidence in making predictions, and then abstain from those with low confidence. To do so, CEHis takes two kinds of information into consideration, namely, the certainty of the current prediction and the accuracy of historical predictions. Experiments with representative TKG reasoning models on two benchmark datasets demonstrate the effectiveness of the proposed CEHis.
title Selective Temporal Knowledge Graph Reasoning
topic Machine Learning
url https://arxiv.org/abs/2404.01695