A Knowledge Graph-Enhanced Attention Aggregation Network for Making Recommendations

被引:5
作者
Zhang, Dehai [1 ]
Yang, Xiaobo [1 ]
Liu, Linan [1 ]
Liu, Qing [1 ]
机构
[1] Yunnan Univ, Coll Software, Kunming 650504, Yunnan, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2021年 / 11卷 / 21期
关键词
knowledge graph; personalized recommendation; attention aggregation;
D O I
10.3390/app112110432
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
In recent years, many researchers have devoted time to designing algorithms used to introduce external information from knowledge graphs, to solve the problems of data sparseness and the cold start, and thus improve the performance of recommendation systems. Inspired by these studies, we proposed KANR, a knowledge graph-enhanced attention aggregation network for making recommendations. This is an end-to-end deep learning model using knowledge graph embedding to enhance the attention aggregation network for making recommendations. It consists of three main parts. The first is the attention aggregation network, which collect the user's interaction history and captures the user's preference for each item. The second is the knowledge graph-embedded model, which aims to integrate the knowledge. The semantic information of the nodes and edges in the graph is mapped to the low-dimensional vector space. The final part is the information interaction unit, which is used for fusing the features of two vectors. Experiments showed that our model achieved a stable improvement compared to the baseline model in making recommendations for movies, books, and music.
引用
收藏
页数:12
相关论文
共 31 条
[1]   Deep Reinforcement Learning A brief survey [J].
Arulkumaran, Kai ;
Deisenroth, Marc Peter ;
Brundage, Miles ;
Bharath, Anil Anthony .
IEEE SIGNAL PROCESSING MAGAZINE, 2017, 34 (06) :26-38
[2]   DBpedia - A crystallization point for the Web of Data [J].
Bizer, Christian ;
Lehmann, Jens ;
Kobilarov, Georgi ;
Auer, Soeren ;
Becker, Christian ;
Cyganiak, Richard ;
Hellmann, Sebastian .
JOURNAL OF WEB SEMANTICS, 2009, 7 (03) :154-165
[3]  
Das D., 2017, INT J COMPUTER APPL, V160, P6, DOI DOI 10.5120/IJCA2017913081
[4]   Hybrid collaborative recommendation of co-embedded item attributes and graph features [J].
Dong, Bingbing ;
Zhu, Yi ;
Li, Lei ;
Wu, Xindong .
NEUROCOMPUTING, 2021, 442 :307-316
[5]   LSTM: A Search Space Odyssey [J].
Greff, Klaus ;
Srivastava, Rupesh K. ;
Koutnik, Jan ;
Steunebrink, Bas R. ;
Schmidhuber, Juergen .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2017, 28 (10) :2222-2232
[6]  
Haoyong Chen, 2016, 2016 IEEE Power and Energy Society General Meeting (PESGM), DOI [10.1109/iWEM.2016.7504980, 10.1109/PESGM.2016.7741231]
[7]  
Huang PS, 2013, PROCEEDINGS OF THE 22ND ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT (CIKM'13), P2333
[8]  
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
[9]  
Kipf TN, 2016, ARXIV
[10]  
Lei DR, 2020, PROCEEDINGS OF THE 2020 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING (EMNLP), P8541