HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Boosting the Speed of Entity Alignment 10*: Dual Attention Matching Network with Normalized Hard Sample Mining

Xin Mao Wenting Wang Yuanbin Wu Man Lan

Boosting the Speed of Entity Alignment 10*: Dual Attention Matching Network with Normalized Hard Sample Mining

Abstract

Seeking the equivalent entities among multi-source Knowledge Graphs (KGs) is the pivotal step to KGs integration, also known as \emph{entity alignment} (EA). However, most existing EA methods are inefficient and poor in scalability. A recent summary points out that some of them even require several days to deal with a dataset containing 200,000 nodes (DWY100K). We believe over-complex graph encoder and inefficient negative sampling strategy are the two main reasons. In this paper, we propose a novel KG encoder -- Dual Attention Matching Network (Dual-AMN), which not only models both intra-graph and cross-graph information smartly, but also greatly reduces computational complexity. Furthermore, we propose the Normalized Hard Sample Mining Loss to smoothly select hard negative samples with reduced loss shift. The experimental results on widely used public datasets indicate that our method achieves both high accuracy and high efficiency. On DWY100K, the whole running process of our method could be finished in 1,100 seconds, at least 10* faster than previous work. The performances of our method also outperform previous works across all datasets, where Hits@1 and MRR have been improved from 6% to 13%.

Code Repositories

MaoXinn/Dual-AMN
Official
tf
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
entity-alignment-on-dbp15k-fr-enDual-AMN
Hits@1: 0.954
entity-alignment-on-dbp15k-ja-enDual-AMN
Hits@1: 0.892
entity-alignment-on-dbp15k-zh-enDual-AMN
Hits@1: 0.861
entity-alignment-on-dicews-1kDual-AMN
Hit@1: 71.6
entity-alignment-on-yago-wiki50kDual-AMN
Hit@1: 89.7

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
Boosting the Speed of Entity Alignment 10*: Dual Attention Matching Network with Normalized Hard Sample Mining | Papers | HyperAI