ComDensE : Combined Dense Embedding of Relation-aware and Common Features for Knowledge Graph Completion
This work addresses the problem of incomplete knowledge graphs for AI applications, presenting an incremental architectural improvement over existing methods.
The paper tackles knowledge graph completion by proposing ComDensE, a model that combines relation-aware and common features using dense neural networks, achieving state-of-the-art performance with improved MRR and HIT@1 scores on FB15k-237 and WN18RR datasets.
Real-world knowledge graphs (KG) are mostly incomplete. The problem of recovering missing relations, called KG completion, has recently become an active research area. Knowledge graph (KG) embedding, a low-dimensional representation of entities and relations, is the crucial technique for KG completion. Convolutional neural networks in models such as ConvE, SACN, InteractE, and RGCN achieve recent successes. This paper takes a different architectural view and proposes ComDensE which combines relation-aware and common features using dense neural networks. In the relation-aware feature extraction, we attempt to create relational inductive bias by applying an encoding function specific to each relation. In the common feature extraction, we apply the common encoding function to all input embeddings. These encoding functions are implemented using dense layers in ComDensE. ComDensE achieves the state-of-the-art performance in the link prediction in terms of MRR, HIT@1 on FB15k-237 and HIT@1 on WN18RR compared to the previous baseline approaches. We conduct an extensive ablation study to examine the effects of the relation-aware layer and the common layer of the ComDensE. Experimental results illustrate that the combined dense architecture as implemented in ComDensE achieves the best performance.