>> Nature Journal >> 2024 >> Issue 1 >> 正文
A neural network recommender algorithm with bi-directional knowledge graph attention
Author: ZHANG Qiuling,WANG Yingxi,WANG Jianfang,NING Hui,WANG Rongsheng Time: 2024-01-25 Counts:

doi10.16186/j.cnki.1673-9787.2021050117

Received:2021/05/26

Revised:2022/07/18

Published:2024/01/25

A neural network recommender algorithm with bi-directional knowledge graph attention

ZHANG Qiuling, WANG Yingxi, WANG Jianfang, NING Hui, WANG Rongsheng

School of Computer Science and TechnologyHenan Polytechnic UniversityJiaozuo 454000HenanChina

Abstract: Objective Recommendation systemone of the most successful application of e-commerce and online services with the main goal of analyzing a user’s history behavior and then predicting items which are of most interest to usershas become a cornerstone of today’s information dissemination.Comparing with the traditional neural networkthe neural network based on knowledge graphKG took the building graph as the input in the recommendation systemwhich could combine the node information and topology for predictionand had demonstrated good results in terms of recommendation accuracy.Howeverthe existing methods rarely consider the symmetric relationship in the graph structure and the problem of gradient vanishing in information aggregation. Methods A bi-directional graph attention neural network recommendation algorithm BGANR was proposed based on the combination of knowledge graph and neural network.Firstly the graph neural network and the symmetric attention mechanism were combined.Thenwithout adding additional dataset dimensionsthe higher-order relationships between users-items were obtained through a bidirectional symmetric embedded translation modelwhich aimd at embedding representations of the features of user-item information in the Knowledge Graphso that the relationships were considered by the attention mechanism in the decision-making weights more comprehensively.The graph-based neural network was used to correct different higher-order relationships by using multi-channel activation functions during the training process of node and neighbor informationso as to increase the amount of feedback information and avoid the over-fitting in the training process. Results The Recall and NDCG metrics in Last-FM data were improved by 2.56% and 1.96% respectivelycompared with the best results of state-of-the-art model. Conclusion The extensive empirical results demonstrated that BGANR could not only explore the higher-order connectivity in bi-directionsbut also realize the efficient transmission of information while capturing effective collaborative signals.

Key words:bi-directional embedding;attention mechanism;knowledge graph;graph neural network

Lastest