Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Unifying Dual-Space Embedding for Entity Alignment via Contrastive Learning

About

Entity alignment aims to match identical entities across different knowledge graphs (KGs). Graph neural network-based entity alignment methods have achieved promising results in Euclidean space. However, KGs often contain complex structures, including both local and hierarchical ones, which make it challenging to efficiently represent them within a single space. In this paper, we proposed a novel method UniEA, which unifies dual-space embedding to preserve the intrinsic structure of KGs. Specifically, we learn graph structure embedding in both Euclidean and hyperbolic spaces simultaneously to maximize the consistency between the embedding in both spaces. Moreover, we employ contrastive learning to mitigate the misalignment issues caused by similar entities, where embedding of similar neighboring entities within the KG become too close in distance. Extensive experiments on benchmark datasets demonstrate that our method achieves state-of-the-art performance in structure-based EA. Our code is available at https://github.com/wonderCS1213/UniEA.

Cunda Wang, Weihua Wang, Qiuyu Liang, Feilong Bao, Guanglai Gao• 2024

Related benchmarks

TaskDatasetResultRank
Entity AlignmentDBP15K FR-EN
Hits@10.964
184
Entity AlignmentDWYDBP-WIKI
Hits@195.7
11
Entity AlignmentICEWS-WIKI
Hits@10.884
10
Entity AlignmentICEWS-YAGO
Hits@186.2
10
Showing 4 of 4 rows

Other info

Follow for update