English

Message Passing Query Embedding

Artificial Intelligence 2020-06-25 v2 Computation and Language Machine Learning

Abstract

Recent works on representation learning for Knowledge Graphs have moved beyond the problem of link prediction, to answering queries of an arbitrary structure. Existing methods are based on ad-hoc mechanisms that require training with a diverse set of query structures. We propose a more general architecture that employs a graph neural network to encode a graph representation of the query, where nodes correspond to entities and variables. The generality of our method allows it to encode a more diverse set of query types in comparison to previous work. Our method shows competitive performance against previous models for complex queries, and in contrast with these models, it can answer complex queries when trained for link prediction only. We show that the model learns entity embeddings that capture the notion of entity type without explicit supervision.

Keywords

Cite

@article{arxiv.2002.02406,
  title  = {Message Passing Query Embedding},
  author = {Daniel Daza and Michael Cochez},
  journal= {arXiv preprint arXiv:2002.02406},
  year   = {2020}
}

Comments

Presented at ICML 2020 - GRL+ Workshop