一种基于图注意力机制的对话历史建模方法

摘要

本发明公开了一种基于图注意力机制的对话历史建模方法，属于自然语言处理领域。该方法的基本思想是通过注意力机制对对话历史信息进行筛选。该方法的步骤包括：对对话历史，知识图谱和当前用户输入进行编码；基于用户输入计算知识图谱节点的注意力权重，结合节点表示向量加权求和获得知识表示向量；基于知识表示向量和用户输入表示向量计算对话历史中每轮对话的注意力权重，结合对话历史中的每轮对话的表示向量获得对话历史表示向量。利用本发明实施例，能够针对不同的用户输入结合知识图谱自适应的生成与当前轮对话信息最相关的对话历史表示向量。即获得噪声更少的对话历史表示向量，从而能够提升下游对话生成模型的效果，具有很大的实用价值。