HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

ClusterEA: Scalable Entity Alignment with Stochastic Training and Normalized Mini-batch Similarities

Yunjun Gao; Xiaoze Liu; Junyang Wu; Tianyi Li; Pengfei Wang; Lu Chen

ClusterEA: Scalable Entity Alignment with Stochastic Training and Normalized Mini-batch Similarities

Abstract

Entity alignment (EA) aims at finding equivalent entities in different knowledge graphs (KGs). Embedding-based approaches have dominated the EA task in recent years. Those methods face problems that come from the geometric properties of embedding vectors, including hubness and isolation. To solve these geometric problems, many normalization approaches have been adopted for EA. However, the increasing scale of KGs renders it hard for EA models to adopt the normalization processes, thus limiting their usage in real-world applications. To tackle this challenge, we present ClusterEA, a general framework that is capable of scaling up EA models and enhancing their results by leveraging normalization methods on mini-batches with a high entity equivalent rate. ClusterEA contains three components to align entities between large-scale KGs, including stochastic training, ClusterSampler, and SparseFusion. It first trains a large-scale Siamese GNN for EA in a stochastic fashion to produce entity embeddings. Based on the embeddings, a novel ClusterSampler strategy is proposed for sampling highly overlapped mini-batches. Finally, ClusterEA incorporates SparseFusion, which normalizes local and global similarity and then fuses all similarity matrices to obtain the final similarity matrix. Extensive experiments with real-life datasets on EA benchmarks offer insight into the proposed framework, and suggest that it is capable of outperforming the state-of-the-art scalable EA framework by up to 8 times in terms of Hits@1.

Code Repositories

immortals88/demo-sea
Mentioned in GitHub
joker-xii/clusterea
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
entity-alignment-on-dbp1m-de-enClusterEA-D
Hit@1: 0.288
entity-alignment-on-dbp1m-de-enClusterEA-R
Hit@1: 0.260
entity-alignment-on-dbp1m-de-enClusterEA-G
Hit@1: 0.100
entity-alignment-on-dbp1m-fr-enClusterEA
Hit@1: 0.281

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
ClusterEA: Scalable Entity Alignment with Stochastic Training and Normalized Mini-batch Similarities | Papers | HyperAI