Computer Science ›› 2021, Vol. 48 ›› Issue (11A): 244-250.doi: 10.11896/jsjkx.210100211

• Big Data & Data Science • Previous Articles     Next Articles

High-order Collaborative Filtering Recommendation System Based on Knowledge Graph Embedding

XU Bing1, YI Pei-yu1, WANG Jin-ce2, PENG Jian1   

  1. 1 College of Computer Science,Sichuan University,Chengdu 610065,China
    2 Department of Computer and Information Engineering,Shanxi Institute of Energy,Jinzhong,Shanxi 030600,China
  • Online:2021-11-10 Published:2021-11-12
  • About author:XU Bing,born in 1989,M.S.candidate.His main research interests include data mining and recommendation system.
    PENG Jian,born in 1970,Ph.D,professor,Ph.D supervisor,is a outstanding member of China Computer Federation.He main research interests include big data and wireless sensor network.
  • Supported by:
    Key R&D Program of Sichuan Province,China(2020YFG0089),University-Local Science and Technology Coo-peration Foundation of Sichuan University-Zigong(2018CDZG-15) and Natural Science Foundation for Young Scientists of Shanxi Province,China(201801D221176).

Abstract: For the problem of data sparsity in recommendation systems,traditional collaborative filtering methods fail to capture the correlations between auxiliary information,which decreases the accuracy of recommendation.In this paper,we propose the KGE-CF model by introducing the knowledge graph as auxiliary information,and utilizing the multi-source structured data in knowledge graph to effectively alleviate data sparsity.KGE-CF integrates multi-layer perceptron to capture the high-order nonli-near features to learn deeper interaction information between users and items,which can improve the quality of recommendation.Specifically,we first map the user's historical interaction items with the corresponding entities in the knowledge graph,and use the translation model of the knowledge graph for training,through that we can get the entity embedding vector and the relation embedding vector,moreover the model can learn a richer user vector by interest disseminating.Then,after concatenating the obtained user vector and the item vector,we send it into the multi-layer perceptron to capture the high-order feature information between the user and the item.Finally,a sigmoid function is used to obtain the user's preference probability for candidate items.The experimental results on the real-world datasets prove that the proposed KGE-CF model in this paper has achieved the best recommendation performance than state-of-art methods.

Key words: Collaborative filtering, Data sparsity, Knowledge graph, Multi-layer perceptron, Recommendation system

CLC Number: 

  • TP3
[1]MARTÍNEZ L,PÉREZ L G,BARRANCO M.A multigranular linguistic content-based recommendation model[J].International Journal of Intelligent Systems,2007,22(5):419-434.
[2]SU X,KHOSHGOFTAAR T M.A survey of collaborative filtering techniques[J/OL].Advances in Artificial Intelligence,2009.https://doi.org/10.1155/20091421425.
[3]NILASHI M,BIN IBRAHIM O,ITHNIN N.Hybrid recom-mendation approaches for multi-criteria collaborative filtering[J].Expert Systems with Applications,2014,41(8):3879-3900.
[4]HE X,LIAO L,ZHANG H,et al.Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web.2017:173-182.
[5]WANG H,ZHANG F,WANG J,et al.Ripplenet:Propagatinguser preferences on the knowledge graph for recommender systems[C]//Proceedings of the 27th ACM International Conference on Information and Knowledge Management.2018:417-426.
[6]CHENG H T,KOC L,HARMSEN J,et al.Wide & deep learning for recommender systems[C]//Proceedings of the 1st Workshop on Deep Learning for Recommender Systems.2016:7-10.
[7]GUO H,TANG R,YE Y,et al.DeepFM:a factorization-machine based neural network for CTR prediction[J].arXiv:1703.04247,2017.
[8]HE X,CHUA T S.Neural factorization machines for sparse predictive analytics[C]//Proceedings of the 40th International ACM SIGIR conference on Research and Development in Information Retrieval.2017:355-364.
[9]SEDHAIN S,MENON A K,SANNER S,et al.Autorec:Autoencoders meet collaborative filtering[C]//Proceedings of the 24th international conference on World Wide Web.2015:111-112.
[10]DEVOOGHT R,BERSINI H.Long and short-term recommendations with recurrent neural networks[C]//Proceedings of the 25th Conference on User Modeling,Adaptation and Personalization.2017:13-21.
[11]ZHANG F,YUAN N J,LIAN D,et al.Collaborative knowledge base embedding for recommender systems[C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.2016:353-362.
[12]LIN Y,LIU Z,SUN M,et al.Learning entity and relation embeddings for knowledge graph completion[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2015:2181-2187.
[13]WANG H,ZHANG F,XIE X,et al.DKN:Deep knowledge-aware network for news recommendation[C]//Proceedings of the 2018 World Wide Web Conference.2018:1835-1844.
[14]JI G,HE S,XU L,et al.Knowledge graph embedding via dynamic mapping matrix[C]//Proceedings of the 53rd Nnnual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing (volume 1:Long papers).2015:687-696.
[15]WANG H,ZHANG F,ZHAO M,et al.Multi-task feature learning for knowledge graph enhanced recommendation[C]//The World Wide Web Conference.2019:2000-2010.
[16]SUN Z,YANG J,ZHANG J,et al.Recurrent knowledge graph embedding for effective recommendation[C]//Proceedings of the 12th ACM Conference on Recommender Systems.2018:297-305.
[17]WANG X,WANG D,XU C,et al.Explainable reasoning overknowledge graphs for recommendation[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2019,33(1):5329-5336.
[18]CAO Y,WANG X,HE X,et al.Unifying knowledge graphlearning and recommendation:Towards a better understanding of user preferences[C]//The World Wide Web Conference.2019:151-161.
[19]ZHAO W X,HE G,YANG K,et al.Kb4rec:A data set for linking knowledge bases with recommender systems[J].Data Intelligence,2019,1(2):121-136.
[20]RENDLE S,FREUDENTHALER C,GANTNER Z,et al.BPR:Bayesian personalized ranking from implicit feedback[J].arXiv:1205.2618,2012.
[1] CHENG Zhang-tao, ZHONG Ting, ZHANG Sheng-ming, ZHOU Fan. Survey of Recommender Systems Based on Graph Learning [J]. Computer Science, 2022, 49(9): 1-13.
[2] WANG Guan-yu, ZHONG Ting, FENG Yu, ZHOU Fan. Collaborative Filtering Recommendation Method Based on Vector Quantization Coding [J]. Computer Science, 2022, 49(9): 48-54.
[3] XU Yong-xin, ZHAO Jun-feng, WANG Ya-sha, XIE Bing, YANG Kai. Temporal Knowledge Graph Representation Learning [J]. Computer Science, 2022, 49(9): 162-171.
[4] RAO Zhi-shuang, JIA Zhen, ZHANG Fan, LI Tian-rui. Key-Value Relational Memory Networks for Question Answering over Knowledge Graph [J]. Computer Science, 2022, 49(9): 202-207.
[5] WU Zi-yi, LI Shao-mei, JIANG Meng-han, ZHANG Jian-peng. Ontology Alignment Method Based on Self-attention [J]. Computer Science, 2022, 49(9): 215-220.
[6] KONG Shi-ming, FENG Yong, ZHANG Jia-yun. Multi-level Inheritance Influence Calculation and Generalization Based on Knowledge Graph [J]. Computer Science, 2022, 49(9): 221-227.
[7] QIN Qi-qi, ZHANG Yue-qin, WANG Run-ze, ZHANG Ze-hua. Hierarchical Granulation Recommendation Method Based on Knowledge Graph [J]. Computer Science, 2022, 49(8): 64-69.
[8] FANG Yi-qiu, ZHANG Zhen-kun, GE Jun-wei. Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning [J]. Computer Science, 2022, 49(8): 70-77.
[9] SHUAI Jian-bo, WANG Jin-ce, HUANG Fei-hu, PENG Jian. Click-Through Rate Prediction Model Based on Neural Architecture Search [J]. Computer Science, 2022, 49(7): 10-17.
[10] QI Xiu-xiu, WANG Jia-hao, LI Wen-xiong, ZHOU Fan. Fusion Algorithm for Matrix Completion Prediction Based on Probabilistic Meta-learning [J]. Computer Science, 2022, 49(7): 18-24.
[11] SUN Xiao-han, ZHANG Li. Collaborative Filtering Recommendation Algorithm Based on Rating Region Subspace [J]. Computer Science, 2022, 49(7): 50-56.
[12] WANG Jie, LI Xiao-nan, LI Guan-yu. Adaptive Attention-based Knowledge Graph Completion [J]. Computer Science, 2022, 49(7): 204-211.
[13] CAI Xiao-juan, TAN Wen-an. Improved Collaborative Filtering Algorithm Combining Similarity and Trust [J]. Computer Science, 2022, 49(6A): 238-241.
[14] WANG Wen-qiang, JIA Xing-xing, LI Peng. Adaptive Ensemble Ordering Algorithm [J]. Computer Science, 2022, 49(6A): 242-246.
[15] HE Yi-chen, MAO Yi-jun, XIE Xian-fen, GU Wan-rong. Matrix Transformation and Factorization Based on Graph Partitioning by Vertex Separator for Recommendation [J]. Computer Science, 2022, 49(6A): 272-279.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!