一种基于强化状态建模的知识图谱多跳推理方法

专利申请号：CN202310749294.7

公 开 号：CN116757282A

发 明 人：张栗粽 田玲 郑旭 尹晓宇 黄嘉豪 杨筠 

代 理 人：重庆速腾专利代理事务所(普通合伙)李兴洲

代理机构：重庆速腾专利代理事务所(普通合伙)

专利类型：发明专利

申 请 日：20230915

公 开 日：20230621

专利主分类号：G06N5/025

关 键 词：嵌入 推理 图谱 强化学习 状态模型 权重 全局目标 预测性能 状态建模 多跳 全局信息 使用测试 图谱数据 指标计算 构建 链接 稀疏 排序 查询 代理 评估 

摘      要：本发明公开了一种基于强化状态建模的知识图谱多跳推理方法，包括以下步骤：利用一组嵌入模型对知识图谱数据进行预训练，得到实体和关系的嵌入表示，并使用测试集评估各嵌入模型的预测性能，得到一组MRR指标；根据MRR指标计算模型权重，并对各嵌入模型进行排序，得到带权重的嵌入模型；根据带权重的嵌入模型，对每个查询计算对应的全局目标嵌入表示；根据全局目标嵌入表示，构建强化学习状态模型；对强化学习状态模型进行训练，并使用训练完成的强化学习状态模型进行知识图谱的推理补全工作，完成基于强化状态建模的知识图谱多跳推理。本发明解决了针对强化代理全局信息表征不足，在稀疏图谱上的推理缺乏依据，从而限制了模型链接预测性能的问题。