融合知识图谱与协同过滤的推荐模型

doi:10.19678/j.issn.1000-3428.0056234

摘要/Abstract

摘要： 针对现有协同过滤推荐算法可解释性不高和基于内容推荐信息提取困难、推荐效率低等问题，提出一种融合知识图谱和协同过滤的混合推荐模型，其由知识图谱与深度学习结合模型RCKD和知识图谱与协同过滤结合模型RCKC构成。RCKD模型在获取知识图谱的推理路径后，利用TransE算法将路径嵌入为向量，并使用LSTM和soft attention机制捕获路径推理的语义，通过池化操作区分不同路径推理的重要性，经全连接层和sigmoid函数获得预测评分。RCKC模型根据知识图谱表示学习的语义相似性，利用协同过滤算法获得预测评分。按预测评分的准确度将两个模型相互融合，最终获得可解释的混合推荐模型。在MovieLens数据集上的实验结果表明，与RKGE、RippleN模型和经典协同过滤算法相比，该模型具有较好的推荐可解释性和较高的推荐准确率。

关键词: 知识图谱, 协同过滤, 深度学习, 混合推荐, 知识表示学习

Abstract: To address the problems of existing collaborative filtering recommendation algorithms,such as low interpretability and difficulty in information extraction based on content recommendation and low recommendation efficiency,this paper proposes a hybrid recommendation model fusing with knowledge graph and collaborative filtering.The model is composed of the RCKD model and the RCKC model,the former combining knowledge graph and deep learning,and the latter combining knowledge graph and collaborative filtering.After obtaining the inference path of knowledge graph,the RCKD model uses the TransE algorithm to embed the path into vector,and captures the semantics of path inference by using LSTM and the soft attention mechanism.Then the importance of different path inferences is distinguished through pooling operation,and the prediction score is obtained through the full connection layer and the sigmoid function.According to the semantic similarity of knowledge graph representation learning,the RCKC model uses the collaborative filtering algorithm to obtain the prediction score.The two models are fused with each other according to the accuracy of the prediction score,and finally the interpretable hybrid recommendation model is obtained.The experimental results on the MovieLens data set show that the proposed model has better recommendation interpretability and higher recommendation accuracy than the RKGE model,RippleN model and the classical collaborative filtering algorithms.

Key words: knowledge graph, collaborative filtering, deep learning, hybrid recommendation, knowledge representation learning

中图分类号:

TP18

康雁, 李涛, 李浩, 钟声, 张亚钏, 卜荣景. 融合知识图谱与协同过滤的推荐模型[J]. 计算机工程, 2020, 46(12): 73-79,87.

KANG Yan, LI Tao, LI Hao, ZHONG Sheng, ZHANG Yachuan, BU Rongjing. Recommendation Model Fusing with Knowledge Graph and Collaborative Filtering[J]. Computer Engineering, 2020, 46(12): 73-79,87.

https://www.ecice06.com/CN/Y2020/V46/I12/73

图/表 9

20201216140535

20201216140539

20201216140542

20201216140547

20201216140602

20201216140605

20201216140608

20201216140612

20201216140615

参考文献

[1] ZHU Wenyue,LIU Wei,LIU Zongtian.News personalized recommendation based on event ontology[J].Computer Engineering,2019,45(6):267-272,279.(in Chinese) 朱文跃,刘炜,刘宗田.基于事件本体的新闻个性化推荐[J].计算机工程,2019,45(6):267-272,279.
[2] RESNICK P,IACOVOU N,SUCHAK M,et al.GroupLens:an open architecture for collaborative filtering of netnews[C]//Proceedings of 1994 ACM Conference on Computer Supported Cooperative Work.New York,USA:ACM Press,1994:175-186.
[3] SARWAR B,KARYPIS G,KONSTAN J,et al.Item-based collaborative filtering recommendation algorithms[C]//Proceedings of International Conference on World Wide Web.New York,USA:ACM Press,2001:285-295.
[4] KOREN Y.Factorization meets the neighborhood:a multifaceted collaborative filtering model[C]//Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.New York,USA:ACM Press,2008:426-434.
[5] EKSTRAND M D.Evaluating collaborative filtering recommender systems[J].ACM Transactions on Information Systems,2007,22(1):5-53.
[6] ZHANG Yongfeng,CHEN Xu.Explainable recommendation:a survey and new perspectives[J].Foundations and Trends in Information Retrieval,2018,14(1):1-101.
[7] RICCI F,ROKACH L,SHAPIRA B,et al.Recommender systems handbook[M].New York,USA:Springer US,2011.
[8] PAZZANI M J.Content-based recommendation systems[M]//BRUSILOVSKY P,KOBSA A,NEJDL W.The adaptive Web.Berlin,Germany:Springer,2007:325-341.
[9] GUAN Saiping,JIN Xiaolong,JIA Yantao,et al.Knowledge reasoning over knowledge graph:a survey[J].Journal of Software,2018,29(10):2966-2994.(in Chinese) 官赛萍,靳小龙,贾岩涛,等.面向知识图谱的知识推理研究进展[J].软件学报,2018,29(10):2966-2994.
[10] SALAKHUTDINOV R,MNIH A,HINTON G.Restricted Boltzmann machines for collaborative filtering[C]//Proceedings of the 24th International Conference on Machine Learning.New York,USA:ACM Press,2007:791-798.
[11] HUANG Posen,HE Xiaodong,GAO Jianfeng,et al.Learning deep structured semantic models for Web search using clickthrough data[C]//Proceedings of the 22nd ACM International Conference on Information and Knowledge Management.New York,USA:ACM Press,2013:2333-2338.
[12] HE Xiangnan,LIAO Lizi,ZHANG Hanwang,et al.Neural collaborative filtering[EB/OL].[2019-08-10].https://arxiv.org/pdf/1708.05031.pdf.
[13] ZHANG Shuai,YAO Lina,SUN Aixin,et al.Deep learning based recommender system:a survey and new perspectives[J].ACM Computing Surveys,2019,52(1):1-38.
[14] WU Xiyu,CHEN Qimai,LIU Hai,et al.Collaborative filtering recommendation algorithm based on representation learning of knowledge graph[J].Computer Engineering,2018,44(2):226-232,263.(in Chinese)吴玺煜,陈启买,刘海,等.基于知识图谱表示学习的协同过滤推荐算法[J].计算机工程,2018,44(2):226-232,263.
[15] ZHAO Wei,CHAI Haixia,WANG Benyou,et al.Leveraging long and short-term information in content-aware movie recommendation[EB/OL].[2019-08-10].https://arxiv.org/pdf/1712.09059.pdf.
[16] SUN Zhu,YANG Jie,ZHANG Jie,et al.Recurrent knowledge graph embedding for effective recommendation[C]//Proceedings of the 12th ACM Conference on Recommender Systems.New York,USA:ACM Press,2018:297-305.
[17] WANG Hongwei,ZHANG Fuzheng,WANG Jialin,et al.RippleNet:propagating user preferences on the knowledge graph for recommender systems[C]//Proceedings of the 27th ACM International Conference.New York,USA:ACM Press,2018:417-426.
[18] SHI Y,LARSON M,HANJALIC A.Collaborative filtering beyond the user-item matrix:a survey of the state of the art and future challenges[J].ACM Computing Surveys,2014,47(1):1-45.
[19] SUN Yizhou,HAN Jiawei,YAN Xifeng,et al.PathSim:meta path-based top-k similarity search in heterogeneous information networks[J].Proceedings of the VLDB Endowment,2011,4(11):992-1003.
[20] LIU Zhiyuan,SUN Maosong,LIN Yankai,et al.Knowledge representation learning:a review[J].Journal of Computer Research and Development,2016,53(2):247-261.(in Chinese)刘知远,孙茂松,林衍凯,等.知识表示学习研究进展[J].计算机研究与发展,2016,53(2):247-261.
[21] BORDES A,USUNIER N,GARCÍADURÁN A,et al.Translating embeddings for modeling multi-relational data[C]//Proceedings of International Conference on Neural Information Processing Systems.[S.l]:Curran Associates,Inc.,2013:2787-2795.
[22] HOCHREITER S,SCHMIDHUBER J.Long short-term memory[J].Neural Computation,1997,9(8):1735-1780.
[23] XU K,BA J,KIROS R,et al.Show,attend and tell:neural image caption generation with visual attention[C]//Pro-ceedings of International Conference on Machine Learning.Lille,France:[s.n.],2015:2048-2057.

选择文件类型/文献管理软件名称

选择包含的内容