Neural Collaborative Recommendation Algorithm Based on Attention Mechanism and Knowledge Graph

被引:0
作者
Zhang, Chuang [1 ,2 ]
Wang, Wei [1 ,2 ,3 ]
Du, Yuxuan [1 ,2 ,3 ]
Zheng, Xiaoli [1 ,2 ]
He, Tingting [1 ,2 ]
机构
[1] School of Information & Electrical Engineering, Hebei University of Engineering, Hebei, Handan
[2] Hebei Key Laboratory of Security & Protection Information Sensing & Processing, Hebei University of Engineering, Hebei, Handan
[3] School of Internet of Things Engineering, Jiangnan University, Jiangsu, Wuxi
关键词
attention mechanism; collaborative filtering; knowledge graph; neural network;
D O I
10.3778/j.issn.1002-8331.2207-0314
中图分类号
学科分类号
摘要
At present, the traditional recommendation algorithm based on collaborative filtering has poor performance in the face of sparse data and cold start. However, the recommendation system assisted by knowledge graph can effectively alleviate this problem. Supplemented by attention mechanism, a neural collaborative recommendation algorithm combining attention mechanism and knowledge graph is designed. Then the attention mechanism is used to learn and aggregate the higher-order potential relationship information in the knowledge graph. At the same time, the user’s final preference is obtained through gated recurrent neural network training based on the user’s long-term and short-term interest preferences. Finally, the collaborative filtering method is used to generate the recommendation list. Through experiments on MovieLens-1M and Amazon-Book datasets, the recommended recall rate, accuracy rate, hit rate and NDCG evaluation indicators are all improved, which validates the effectiveness of the proposed method. © 2023 The Author(s).
引用
收藏
页码:111 / 120
页数:9
相关论文
共 27 条
[1]  
HU B, SHI C, ZHAO W X, Et al., Leveraging meta-path based context for top-n recommendation with a neural coattention model[C], Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 1531-1540, (2018)
[2]  
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, pp. 353-362, (2016)
[3]  
JU C, WANG J, ZHOU G., The commodity recommendation method for online shopping based on data mining[J], Multimedia Tools and Applications, 78, 21, pp. 30097-30110, (2019)
[4]  
WU X, CHENG B, CHEN J., Collaborative filtering service recommendation based on a novel similarity computation method[J], IEEE Transactions on Services Computing, 10, 3, pp. 352-365, (2015)
[5]  
XUE F,, HE X, WANG X, Et al., Deep item-based collaborative filtering for top-n recommendation[J], ACM Transactions on Information Systems, 37, 3, (2019)
[6]  
WANG X, WANG D, Et al., Explainable reasoning over knowledge graphs for recommendation[C], Proceedings of the 33rd AAAI Conference on Artificial Intelligence, pp. 5329-5336, (2019)
[7]  
RENDLE S, Et al., Fast contextaware recommendations with factorization machines[C], Proceedings of the 34th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 635-644, (2011)
[8]  
XIANG H, TAT C., Neural factorization machines for sparse predictive analytics[C], Proceedings of the 40th International ACM SIGIR Conference on Research and Development in Information Retrieval, pp. 355-364, (2017)
[9]  
CHENG H T, KOC L, Et al., Wide & deep learning for recommender systems[C], Proceedings of the 1st Workshop on Deep Learning for Recommender Systems, pp. 7-10, (2016)
[10]  
JIAN L, XIAO Z, FU Z, Et al., xDeepFM:combining explicit and implicit feature interactions for recommender systems[C], Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 1754-1763, (2018)