论文精选

GA-S2S:融合图结构的Seq2Seq模型提升知识图谱链接预测

Leveraging Graph Structure in Seq2Seq Models for Knowledge Graph Link Prediction

精选理由

做知识图谱推理或链接预测的团队,GA-S2S用图结构补上了Seq2Seq模型的盲区,效果提升明显,值得在CoDEx等数据集上复现试试。

AI 摘要

本文提出GA-S2S框架,将T5-small编码器-解码器与关系图注意力网络(RGAT)结合,用于知识图谱链接预测。现有Seq2Seq模型仅依赖实体和关系的文本描述,最多将查询实体的邻域展平为线性序列,丢失了图结构信息。GA-S2S同时编码文本特征和查询实体周围的完整k跳子图拓扑,通过融合原始编码器输出与RGAT的关系感知嵌入,捕获更丰富的多跳关系模式和文本信息。在CoDEx数据集上的初步实验表明,GA-S2S相比竞争性的Seq2Seq基线模型,链接预测准确率最高提升19%。

AI 翻译 · 中文

本文提出GA-S2S框架,将T5-small编码器-解码器与关系图注意力网络(RGAT)结合,用于知识图谱链接预测。现有Seq2Seq模型仅依赖实体和关系的文本描述,最多将查询实体的邻域展平为线性序列,丢失了图结构信息。GA-S2S同时编码文本特征和查询实体周围的完整k跳子图拓扑,通过融合原始编码器输出与RGAT的关系感知嵌入,捕获更丰富的多跳关系模式和文本信息。在CoDEx数据集上的初步实验表明,GA-S2S相比竞争性的Seq2Seq基线模型,链接预测准确率最高提升19%。

arXiv cs.AIWe introduce Graph-Augmented Sequence-to-Sequence (GA-S2S), a novel framework that integrates a T5-small encoder-decoder with a Relational Graph Attention Network (RGAT) to improve link prediction in knowledge graphs. Wh