基于知识图谱和Bi-LSTM的推荐算法

• 作者: 王钰蓥 王勇
• 来源期刊: 计算机与现代化
• 关键词: 知识图谱 双向循环神经网络 注意力机制 可解释性 冷启动
• 年，卷（期）: 2021,（9）
• 所属期刊栏目: 人工智能|ARTIFICIAL INTELLIGENCE
• 页码范围: 90-98
• 页数: 9页
• 分类号: TP391
• 语种: 中文
• DOI: 10.3969/j.issn.1006-2475.2021.09.014

摘要

目前现有基于模型的推荐算法多是将评分数据输入到深度学习模型中进行训练,得出推荐结果.其缺陷在于无法对预测结果进行可解释性分析.除此之外,无法有效地解决算法的冷启动问题.因此,本文提出一种基于知识图谱和Bi-LSTM的推荐算法,来有效解决算法的可解释性和冷启动问题.首先将获取到的数据集进行预处理,生成预编码向量,根据数据集结点的连接性,构建专业领域知识图谱.其次利用知识图谱的元路径提取技术获取到多条用户-物品路径信息,将其输入到Bi-LSTM中,在路径经过的各结点处加入一层注意力机制,目的是为了模型能够有效地获取到较远结点的信息.最后将多条路径的训练结果输入到平均池化层中,用以区分不同路径的重要程度,利用交叉熵损失函数对模型进行训练,从而得出预测结果.实验结果表明,与传统基于循环神经网络模型的推荐算法相比,该算法可有效地提升算法的可解释性以及预测准确性,并缓解算法的冷启动问题.