Cross attention fusion for knowledge graph optimized recommendation

被引:19
作者
Huang, Weijian [1 ]
Wu, Jianhua [1 ]
Song, Weihu [1 ]
Wang, Zehua [1 ]
机构
[1] Hebei Univ Engn, Sch Informat & Elect Engn, Handan 056038, Peoples R China
关键词
Recommendation systems; Knowledge graph; Multi-task learning; Feature cross;
D O I
10.1007/s10489-021-02930-1
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Knowledge Graph has attracted a wide range of attention in the field of recommendation, which is usually applied as auxiliary information to solve the problem of data sparsity. However, most recommendation models cannot effectively mine the associations between the items to be recommended and the entities in the Knowledge Graph. In this paper, we propose CAKR, a knowledge graph recommendation method based on the cross attention unit, which is similar to MKR, a multi-task feature learning general framework that uses knowledge graph embedding tasks to assist recommendation tasks. Specifically, we design a new method to optimize the feature interaction between the items and the corresponding entities in the Knowledge Graph and propose a feature cross-unit combined with the attention mechanism to enhance the recommendation effect. Through extensive experiments on the public datasets of movies, books, and music, we prove that CAKR is better than MKR and other knowledge graph recommendation methods so that the new feature cross-unit designed in this paper is effective in improving the accuracy of the recommendation system.
引用
收藏
页码:10297 / 10306
页数:10
相关论文
共 27 条
[1]  
Bordes A., 2013, NIPS'13, P1
[2]   Unifying Knowledge Graph Learning and Recommendation: Towards a Better Understanding of User Preferences [J].
Cao, Yixin ;
Wang, Xiang ;
He, Xiangnan ;
Hu, Zikun ;
Chua, Tat-Seng .
WEB CONFERENCE 2019: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2019), 2019, :151-161
[3]   A Survey on Knowledge Graph-Based Recommender Systems [J].
Guo, Qingyu ;
Zhuang, Fuzhen ;
Qin, Chuan ;
Zhu, Hengshu ;
Xie, Xing ;
Xiong, Hui ;
He, Qing .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2022, 34 (08) :3549-3568
[4]  
Ji GL, 2015, PROCEEDINGS OF THE 53RD ANNUAL MEETING OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS AND THE 7TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING, VOL 1, P687
[5]  
Lin YK, 2015, AAAI CONF ARTIF INTE, P2181
[6]   Learning social representations with deep autoencoder for recommender system [J].
Pan, Yiteng ;
He, Fazhi ;
Yu, Haiping .
WORLD WIDE WEB-INTERNET AND WEB INFORMATION SYSTEMS, 2020, 23 (04) :2259-2279
[7]   A correlative denoising autoencoder to model social influence for top-N recommender system [J].
Pan, Yiteng ;
He, Fazhi ;
Yu, Haiping .
FRONTIERS OF COMPUTER SCIENCE, 2020, 14 (03)
[8]   Modeling Relational Data with Graph Convolutional Networks [J].
Schlichtkrull, Michael ;
Kipf, Thomas N. ;
Bloem, Peter ;
van den Berg, Rianne ;
Titov, Ivan ;
Welling, Max .
SEMANTIC WEB (ESWC 2018), 2018, 10843 :593-607
[9]  
Srivastava N, 2014, J MACH LEARN RES, V15, P1929
[10]  
Sun Z., 2019, 7 INT C LEARN REPR O