Gradient Flow of Energy: A General and Efficient Approach for Entity Alignment Decoding

Yuanyi Wang, Haifeng Sun, Jingyu Wang, Qi Qi, Shaoling Sun +1 more
1/23/2024
cs.IRcs.CL

No official code found. Best implementation: wyy-code/TFP (1 star, last commit 2026-01-20).

Abstract

Entity alignment (EA), a pivotal process in integrating multi-source Knowledge Graphs (KGs), seeks to identify equivalent entity pairs across these graphs. Most existing approaches regard EA as a graph representation learning task, concentrating on enhancing graph encoders. However, the decoding process in EA - essential for effective operation and alignment accuracy - has received limited attention and remains tailored to specific datasets and model architectures, necessitating both entity and additional explicit relation embeddings. This specificity limits its applicability, particularly in GNN-based models. To address this gap, we introduce a novel, generalized, and efficient decoding approach for EA, relying solely on entity embeddings. Our method optimizes the decoding process by minimizing Dirichlet energy, leading to the gradient flow within the graph, to maximize graph homophily. The discretization of the gradient flow produces a fast and scalable approach, termed Triple Feature Propagation (TFP). TFP innovatively generalizes adjacency matrices to multi-views matrices:entity-to-entity, entity-to-relation, relation-to-entity, and relation-to-triple. The gradient flow through generalized matrices enables TFP to harness the multi-view structural information of KGs. Rigorous experimentation on diverse public datasets demonstrates that our approach significantly enhances various EA methods. Notably, the approach achieves these advancements with less than 6 seconds of additional computational time, establishing a new benchmark in efficiency and adaptability for future EA methods.

View on arXivView PDF

Code Implementations(1)

wyy-code/TFPCommunity100%

This is code of Triple Feature Propagation (TFP) introduced in our paper: "Rethinking Smoothness for Fast and Adaptable Entity Alignment Decoding", accpeted by NAACL 2025

10PythonDec 26, 20239 months ago

Similar Papers

Cite this paper

@article{wang2024gradient,
  title  = {Gradient Flow of Energy: A General and Efficient Approach for Entity Alignment Decoding},
  author = {Yuanyi Wang and Haifeng Sun and Jingyu Wang and Qi Qi and Shaoling Sun and Jianxin Liao},
  year   = {2024},
  eprint = {2401.12798},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.12798v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2401.12798.svg)](https://codeofpaper.com/paper/2401.12798)

Discussion