HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

SEEK: Segmented Embedding of Knowledge Graphs

Wentao Xu Shun Zheng Liang He Bin Shao Jian Yin Tie-Yan Liu

SEEK: Segmented Embedding of Knowledge Graphs

Abstract

In recent years, knowledge graph embedding becomes a pretty hot research topic of artificial intelligence and plays increasingly vital roles in various downstream applications, such as recommendation and question answering. However, existing methods for knowledge graph embedding can not make a proper trade-off between the model complexity and the model expressiveness, which makes them still far from satisfactory. To mitigate this problem, we propose a lightweight modeling framework that can achieve highly competitive relational expressiveness without increasing the model complexity. Our framework focuses on the design of scoring functions and highlights two critical characteristics: 1) facilitating sufficient feature interactions; 2) preserving both symmetry and antisymmetry properties of relations. It is noteworthy that owing to the general and elegant design of scoring functions, our framework can incorporate many famous existing methods as special cases. Moreover, extensive experiments on public benchmarks demonstrate the efficiency and effectiveness of our framework. Source codes and data can be found at \url{https://github.com/Wentao-Xu/SEEK}.

Code Repositories

Wentao-Xu/SEEK
Official
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
link-prediction-on-fb15kSEEK
Hits@1: 0.792
Hits@10: 0.886
Hits@3: 0.841
MRR: 0.825
link-prediction-on-yago37SEEK
Hits@1: 0.370
Hits@10: 0.622
Hits@3: 0.498
MRR: 0.454

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
SEEK: Segmented Embedding of Knowledge Graphs | Papers | HyperAI