LighterKGCN: A Recommender System Model based on Bi-layer Graph Convolutional Networks

LighterKGCN: A Recommender System Model based on Bi-layer Graph Convolutional Networks

Recommender systems have been extensively utilized to meet users’ personalized needs. Collaborative filtering is one of the most classic algorithms in the recommendation field. However, it has problems such as cold start and data sparsity. In that case, knowledge graphs and graph convolutional networks have been introduced by scholars into recommender systems to solve the above problems. However, the current graph convolutional networks fail to give full play to the advantages of graph convolution since they are employed either in the embedding representations of users and commodity entities, or in the embedding representations between entities of the knowledge graphs. Therefore, LighterKGCN, a recommender system model based on bi-layer graph convolutional networks was proposed in accordance with the KGCN model and the LightGCN model. In the first layer of GCN, the model first learned the embedding representations of users and commodity entities on the user-commodity entity interaction graph. Then, the attained user embedding and commodity embedding were used as the data source for the second layer of GCN. In the second layer, the entity v and its neighborhoods were calculated using the hybrid aggregation function proposed in this paper. The result was taken as the new entity v. According to tests on three public datasets and comparison results with the KGCN, LighterKGCN improved by 0.52% and 51.16% in terms of AUC and F1 performances, respectively on the dataset of MovieLens-20M; LighterKGCN improved by 0.67% and 45.0% in terms of AUC and F1 performances, respectively on the dataset of Yelp2018; and the number was 0.67% and 36.35% in AUC and F1 performances, respectively on the dataset of Last.FM.