Enhanced Sub-graph Reconstruction Graph Neural Network for Recommendation

Zhe Liu,Xiaojun Lou,Jian Li,Guanjun Liu
DOI: https://doi.org/10.1080/08839514.2024.2355425
IF: 2.777
2024-06-14
Applied Artificial Intelligence
Abstract:Personalized recommendation can recommend items of interest to different users and is widely used in the real world. Among them, graph collaborative filtering is a method of personalized recommendation. It can enrich the connection between users and items on the basis of collaborative filtering, to learn the embedded representation of nodes more accurately. Since graph collaborative filtering is based on bipartite graphs, few exciting graph collaborative methods consider the relationships between users (or items), the message between homogeneous nodes are diluted or ignored. Predicting and constructing the relationship between users (or items) has become a challenging. To solve this problem, we propose an enhanced sub-graph reconstruction graph neural network for recommendation (SRCF), using a heterogeneous graph neural network based encoder-decoder learn potential relationships between users (or items), and reconstruct sub-graphs based on those relationships. In the proposed model, the information of user and item sub-graphs is merged with the network of graph collaborative filtering, which enhances effective information transfer between homogeneous nodes, thereby improving the model performance. We have selected a number of data sets of different scenarios and different scales to comprehensively evaluate the performance of the model, and the experimental results confirmed the superiority of our model.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?
The problem that this paper attempts to solve is how to more effectively capture the relationships between users and between items in the personalized recommendation system to improve the performance of the recommendation system. Traditional graph - collaborative filtering methods mainly focus on the interaction relationships between users and items, but often overlook the relationships of homogeneous nodes (i.e., between users or between items), resulting in important information in these relationships being diluted or ignored during the information transfer process. Therefore, the paper proposes an enhanced sub - graph reconstruction graph neural network (SRCF), which enhances the information transfer between homogeneous nodes by constructing weighted user sub - graphs and item sub - graphs, thereby improving the model performance. Specifically, the main contributions of the paper include: 1. **Latent Relationship Mining**: Proposed an encoder based on graph neural networks for learning the interaction structure between users and items, mining the latent relationships between users (or items), and reconstructing weighted user sub - graphs and item sub - graphs through these latent relationships, enabling the model to accurately learn the information transfer between homogeneous nodes. 2. **High - Order Relationship Representation**: Utilized the characteristic that graph neural networks can aggregate high - order connection relationships to obtain high - order relationship representations between users (or items) from weighted user sub - graphs and item sub - graphs, and combined them with the CF signals in the user - item interaction graph. Compared with traditional graph - collaborative filtering models, it can mine more useful information from the data set and represent users and items more comprehensively and accurately. 3. **Experimental Verification**: Experiments were carried out on data sets in multiple different scenarios and scales, and the experimental results show that the proposed model is effective. Through these improvements, the paper aims to enhance the performance of the personalized recommendation system, especially its performance when dealing with large - scale and sparse data.