Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper tackles the challenge of Entity Alignment (EA) in knowledge graph integration by introducing two innovative modules: Predicate Importance Estimation (PIE) and Decoupled Rationale-Score Distillation (DRSD). PIE enhances entity embeddings by focusing on predicate importance while eliminating subject information from triples, and DRSD leverages a distilled small language model to convert binary EA labels into text-based supervision, fostering task-specific reasoning. The combined approach significantly improves EA classification performance and introduces a mechanism for flagging uncertain predictions for human review, thus enhancing the reliability of automated systems.
By decoupling confidence scores from decision-making, this approach not only boosts EA accuracy but also flags uncertain predictions for human review, enhancing system reliability.
Knowledge graphs (KGs) are increasingly used as structured context for Large Language Models (LLMs), but industrial KG-RAG systems often need to integrate public and domain-specific KGs constructed from heterogeneous databases. This integration relies on Entity Alignment (EA), where lexical matching alone is insufficient under predicate-name variation and incomplete local neighborhoods. We address EA for KG integration by constructing a pairwise EA dataset and proposing two complementary modules: Predicate Importance Estimation (PIE) and Decoupled Rationale-Score Distillation (DRSD). PIE is a compact embedding-based approach that removes the subject information from each 1-hop triple, encodes the resulting subjectless triples, and aggregates them with learnable predicate-importance weights to build predicate-aware entity embeddings. DRSD trains a distilled small language model (SLM) with pseudo-answers produced by a teacher LLM through distinct prompts. By converting binary EA labels into text-based supervision and decoupling confidence-score estimation from label-consistent rationales, DRSD enables the SLM to learn task-specific reasoning while retaining a less label-biased confidence signal. Experiments show that PIE and DRSD improve EA classification. Moreover, because DRSD decouples confidence-score estimation from the decision, a discrepancy between the two flags an uncertain prediction for human review, thereby enabling a practical discrepancy between automatic acceptance and human-in-the-loop verification.