计算机与现代化 ›› 2021, Vol. 0 ›› Issue (09): 90-98.

• 人工智能 • 上一篇    下一篇

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

  

  1. (北京工业大学信息学部,北京100124)
  • 出版日期:2021-09-14 发布日期:2021-09-14
  • 作者简介:王钰蓥(1996—),女,天津人,硕士研究生,研究方向:深度学习,推荐系统,E-mail: wangyuying_425@163.com; 王勇(1974—),男,副教授,硕士生导师,研究方向:并行与分布式计算,E-mail: wangy@bjut.edu.cn。

Recommendation Algorithm Based on Knowledge Graph and Bi-LSTM

  1. (Department of Information Science, Beijing University of Technology, Beijing 100124, China)
  • Online:2021-09-14 Published:2021-09-14

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

关键词: 知识图谱, 双向循环神经网络, 注意力机制, 可解释性, 冷启动

Abstract: At present, most of the existing model-based recommendation algorithms input the score data into the deep learning model for training to get the recommendation results. Its defect is that it is unable to analyze the interpretability of the prediction results. In addition, the algorithm can not effectively solve the cold start problem. Therefore, this paper proposes a recommendation algorithm based on knowledge map and Bi-LSTM to effectively solve the problem of interpretability and cold start of the algorithm. Firstly, the data set is preprocessed to generate precoding vector. According to the connectivity of data aggregation points, the domain knowledge map is constructed. Secondly, the meta path extraction technology of knowledge map is used to obtain multiple user item path information, which is input into Bi-LSTM. A layer of attention mechanism is added to each node of the path, so that the model could effectively obtain the information of remote nodes. Finally, the training results of multiple paths are input into the average pooling layer to distinguish the importance of different paths. The cross-entropy loss function is used to train the model and the prediction results are obtained. The experimental results show that, compared with the traditional recommendation algorithm based on the cyclic neural network model, this algorithm can effectively improve the interpretability and prediction accuracy of the algorithm, and alleviate the cold start problem of the algorithm.

Key words: knowledge graph, bidirectional recurrent neural network, attention mechanism, interpretability, cold start