Hybrid Recommendation System with Graph Neural Collaborative Filtering and Local Self-attention Mechanism.
Ao Zhang,Yifei Sun,Shi Cheng,Jie Yang,Xin Sun,Zhuo Liu,Yifei Cao,Jiale Ju,Wenya Shi
DOI: https://doi.org/10.1007/978-981-99-5847-4_14
2023-01-01
Abstract:As the problem of information overload becomes increasingly serious, traditional recommendation algorithms are difficult to efficiently provide assistance to users. In this context, more and more scholars have applied deep learning methods to recommendation systems, such as deep reinforcement learning, self-attention mechanisms, graph neural collaborative filtering and so on. Experimental results have shown that these methods significantly improve the performance of systems. However, most studies directly apply existing models to recommendation algorithms, wasting a lot of potential features. In recent years, the latest studies have optimized model performance through method fusion. In this paper, we propose a hybrid model NGCF-A that combines graph neural collaborative filtering and self-attention mechanisms. In a system based on graph neural collaborative filtering, local self-attention layers are added to embedding propagation layer, so that we can capturing high level associations between users and items. In addition, we add an implicit feedback transformation method and a dataset splitting strategy that conforms to change in interest, aiming to make better use of features. The experimental results show that this model achieves better performance than traditional graph neural collaborative filtering model, and the new implicit feedback transformation method further optimizes the model.