一种测试时知识图谱增强大语言模型的方法

本发明提供一种测试时知识图谱增强大语言模型方法。该方法提出一种新型基于知识图谱的注意力机制，以一种双向信息流聚合方式来提升知识图谱增强大语言模型的效果，包括以下步骤：根据文本输入序列从知识图谱中检索三元组作为知识图谱三元组输入序列；将文本输入序列和知识图谱三元组输入序列转换为嵌入向量，并计算查询矩阵、键矩阵和值矩阵；计算混合注意力；通过前馈神经网络得到更丰富特征表示；依据上述特征表示生成模型预测概率分布，从而预测下一个单词。