Neighbor-Augmented Knowledge Graph Attention Network for Recommendation

Qi Wang,Hao Cui,Jiapeng Zhang,Yan Du,Yuan Zhou,Xiaojun Lu
DOI: https://doi.org/10.1007/s11063-023-11310-4
IF: 2.565
2023-07-03
Neural Processing Letters
Abstract:Previous knowledge graph-based recommendation models use Graph Neural Networks to aggregate information from neighbors. However, entities of the same hop should have different importance in aggregating, but this is often ignored. To overcome this problem, we try to utilize more helpful information to enhance the recommender system. This paper presents a new recommendation model: Neural Attention Network for Neighborhood Augmented Knowledge Graph (NKGAT). The model is developed based on knowledge graph and graph neural network. In the proposed model, we explore higher-order relationships using embeddings of nodes. Moreover, a new attention module (Neighbor-Augmented Attention (NAA)) is proposed. It assigns proper importance to different entities of neighbors at the same hop in the graph neural network. The experiments are carried out on Movie-Lens, Last-FM, Book-Crossing and Yelp2018 datasets. The experimental results show that NKGAT achieves state-of-the-art performance. It demonstrates that the proposed method integrates rich higher-order information with more proper attention to entities.
computer science, artificial intelligence
What problem does this paper attempt to address?