Applied Sciences (Nov 2021)

A Knowledge Graph-Enhanced Attention Aggregation Network for Making Recommendations

  • Dehai Zhang,
  • Xiaobo Yang,
  • Linan Liu,
  • Qing Liu

DOI
https://doi.org/10.3390/app112110432
Journal volume & issue
Vol. 11, no. 21
p. 10432

Abstract

Read online

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.

Keywords