基于图注意力神经网络的实体消歧方法

Entity disambiguation method based on graph attention networks

针对链接对象为存在半结构化数据的知识库,提出了一种基于图注意力神经网络的短文本实体指称消歧方法。通过信息抽取与融入关键词,将含有半结构化数据的知识库构建为全局知识图谱;同时基于Bert预训练模型对短文本中的实体指称项进行嵌入融合;使用图注意力神经网络对全局知识图谱中候选实体节点进行加权聚合表征,并计算实体指称项与各候选实体之间的相似度得分,实现实体消歧。在CCKS2019数据集上的实验结果表明,基于图注意力神网络的实体消歧模型有效提高了实体消歧效果。

We propose an entity disambiguation method based on graph attention networks for semi-structured knowledge base data.First,a global knowledge graph is constructed from the semi-structured knowledge base,and the entity reference items are embedded by Bert pre-trained model meanwhile.Next,graph attention networks which leverages masked self-attention layers is applyed on candidate entity nodes of global knowledge graph to fetch a vector of node level.Furtherly,we com pute similarity scores rank between the entity reference items and the candidate entity to complete the task of entity disambiguation.The experimental results on CCKS2019 dataset achieve state-of-the-art.