English

CausE: Towards Causal Knowledge Graph Embedding

Computation and Language 2023-07-25 v2 Artificial Intelligence

Abstract

Knowledge graph embedding (KGE) focuses on representing the entities and relations of a knowledge graph (KG) into the continuous vector spaces, which can be employed to predict the missing triples to achieve knowledge graph completion (KGC). However, KGE models often only briefly learn structural correlations of triple data and embeddings would be misled by the trivial patterns and noisy links in real-world KGs. To address this issue, we build the new paradigm of KGE in the context of causality and embedding disentanglement. We further propose a Causality-enhanced knowledge graph Embedding (CausE) framework. CausE employs causal intervention to estimate the causal effect of the confounder embeddings and design new training objectives to make stable predictions. Experimental results demonstrate that CausE could outperform the baseline models and achieve state-of-the-art KGC performance. We release our code in https://github.com/zjukg/CausE.

Keywords

Cite

@article{arxiv.2307.11610,
  title  = {CausE: Towards Causal Knowledge Graph Embedding},
  author = {Yichi Zhang and Wen Zhang},
  journal= {arXiv preprint arXiv:2307.11610},
  year   = {2023}
}

Comments

Accepted by CCKS 2023 as a research paper

R2 v1 2026-06-28T11:37:01.142Z