摘要
本发明涉及一种面向智能家居的注意力扩散知识图谱推理方法,属于知识图谱和人工智能技术领域。其包括以下步骤:从公开的WN18RR知识图谱数据集中提取实体、关系及三元组数据;设定推理过程的传播层数并初始化头实体集合,推理在层中循环如下过程:依据第l层头实体找相连关系及目标实体信息,得到相关三元组集合;用多头注意力机制捕获实体特征,生成消息传递;基于消息传递实现目标实体消息传递与聚合,生成目标实体的特征表示;基于语义相关性度量筛选top‑k个目标节点,扩散到下一层;循环完成后,将层消息扩散得到的最终目标实体的特征表示进行线性映射,得到推理结果。本发明能够提高智能家居场景中知识图谱推理的准确性和效率。