Abstract:Knowledge graphs (KGs) have recently become increasingly popular due to the broad range of essential applications in various downstream tasks including intelligent search, personalized recommendations, intelligent financial data analytics, etc. During an automated construction of a KG, the knowledge facts from multiple knowledge sources are automatically extracted in the form of triples, and these observed triples are used to derive new unobserved triples for KG completion (also known as link prediction). State-of-the-art link prediction methods are known to be primarily KG embedding models, among which tensor factorization models have recently drawn much attention due to their scalability and expressive feature embeddings, and hence, perform well for link prediction. However, these embedding models consider each KG triple individually and fail to capture the useful information present in the neighborhood of a node. To this end, we propose a novel end-to-end KG embedding learning framework that consists of an encoder of a dual weighted graph convolutional network, and a decoder of a novel fully expressive tensor factorization model. The proposed encoder extends weighted graph convolutional network to generate two rich and high quality embedding vectors for each node by aggregating information from the neighboring nodes. The proposed decoder has a flexible and powerful tensor representation form of the Tensor Train decomposition that takes benefit of the two representations of each node in its embedding space to accurately model the KG triples. We also derive a bound on the size of the embeddings for full expressivity and show that our proposed tensor factorization model is fully expressive. Additionally, we show the relationship of our tensor factorization model to previous tensor factorization models. The experimental results show the effectiveness of the proposed framework that consistently marks performance gains over several previous models on recent standard link prediction datasets.

Dual Graph Embedding for Object-Tag Link Prediction on the Knowledge Graph.

Learning hierarchy-aware complex knowledge graph embeddings for link prediction

Interaction Embeddings for Prediction and Explanation in Knowledge Graphs.

Knowledge Graph Embedding with Diversity of Structures

A Knowledge Graph Embedding Method Based on Neural Network

HGCGE: Hyperbolic Graph Convolutional Networks-Based Knowledge Graph Embedding for Link Prediction

KGEL: A novel end-to-end embedding learning framework for knowledge graph completion

Efficiently Embedding Dynamic Knowledge Graphs

Link Prediction Based on Data Augmentation and Metric Learning Knowledge Graph Embedding

Dual Quaternion Knowledge Graph Embeddings

Path-KGE: Preference-Aware Knowledge Graph Embedding with Path Semantics for Link Prediction.

Knowledge Graph Embedding Based on Subgraph-Aware Proximity

Relation Embedding with Dihedral Group in Knowledge Graph

Link Prediction using Embedded Knowledge Graphs

A Survey on Knowledge Graph Embeddings for Link Prediction

Learning knowledge graph embedding with a dual-attention embedding network

Analogical Inference Enhanced Knowledge Graph Embedding

Knowledge graph embedding based on dynamic adaptive atrous convolution and attention mechanism for link prediction

Temporal Knowledge Graph Embedding for Link Prediction.

Two Birds with One Stone: A Link Prediction Model for Knowledge Hypergraph Based on Fully-Connected Tensor Decomposition.

Dual Quaternion Embeddings for Link Prediction