通过语义筛选邻居,提升知识图谱补全精度
Context-Driven Knowledge Graph Completion with Semantic-Aware Relational Message Passing
- 按语义相关性选最相关的边进行消息传递
- 在多个基准上优于现有方法,有效减少噪声干扰
- 适合需要精准链接预测的场景,如智能问答
三元组(h, r, t)周围的语义上下文对知识图谱补全(KGC)至关重要,为预测提供关键线索。然而,传统基于节点的消息传递机制在知识图谱中常因无差别聚合所有邻接边信息而引入噪声,导致信息稀释或过平滑。为此,我们提出一种语义感知的关系消息传递框架。核心创新在于引入语义感知的Top-K邻居选择策略:首先在共享隐空间中评估中心节点与其关联边的语义相关性,仅选取最相关的前K条边;随后使用多头注意力聚合器将这些选中边的信息与中心节点表示融合,生成语义聚焦的节点消息。该方法不仅利用图结构和边特征,还更准确地捕捉并传播与特定链接预测任务相关的上下文信息,有效抑制无关信息干扰。大量实验表明,本方法在多个主流基准上均优于现有方法。
原文摘要 · Abstract (English)
Semantic context surrounding a triplet $(h, r, t)$ is crucial for Knowledge Graph Completion (KGC), providing vital cues for prediction. However, traditional node-based message passing mechanisms, when applied to knowledge graphs, often introduce noise and suffer from information dilution or over-smoothing by indiscriminately aggregating information from all neighboring edges. To address this challenge, we propose a semantic-aware relational message passing. A core innovation of this framework is the introduction of a semantic-aware Top-K neighbor selection strategy. Specifically, this strategy first evaluates the semantic relevance between a central node and its incident edges within a shared latent space, selecting only the Top-K most pertinent ones. Subsequently, information from these selected edges is effectively fused with the central node's own representation using a multi-head attention aggregator to generate a semantically focused node message. In this manner, our model not only leverages the structure and features of edges within the knowledge graph but also more accurately captures and propagates the contextual information most relevant to the specific link prediction task, thereby effectively mitigating interference from irrelevant information. Extensive experiments demonstrate that our method achieves superior performance compared to existing approaches on several established benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。