Computer Engineering and Applications ›› 2022, Vol. 58 ›› Issue (17): 199-205.DOI: 10.3778/j.issn.1002-8331.2101-0324

• Pattern Recognition and Artificial Intelligence • Previous Articles     Next Articles

Collaborative Filtering Recommendation Model Based on Improved Graph Convolutional Network

ZHAI Zhengli, FENG Shu, LI Penghui   

  1. School of Information and Control Engineering, Qingdao University of Technology, Qingdao, Shandong 266520, China
  • Online:2022-09-01 Published:2022-09-01

改进图卷积网络的协同过滤推荐模型

翟正利,冯舒,李鹏辉   

  1. 青岛理工大学 信息与控制工程学院,山东 青岛 266520

Abstract: graph convolutional networks(GCN) have been applied in various fields with great success, including recommendation systems. The core goal of the recommendation system is to infer user preferences so as to actively provide users with valuable and targeted information. Collaborative filtering is a classic and widely used recommendation method.However, GCN-based recommendation methods still have many problems. For example, GCN recursively merges messages from different order neighborhoods, mixing different node messages indistinguishably leads to training difficulties, and problems such as over-smoothing have great constraints on the recommendation model. Different from the current GCN-based method, in order to solve the above problems, it is proposed to use a simple GCN model to separately aggregate neighbor messages in different orders for collaborative filtering, and then aggregate them in a hierarchical manner without introducing other model parameters. After that, the Dropout idea is transferred to the model, and the influence of over-smoothing is reduced by randomly discarding neighbor messages at each layer, which prevents over-fitting and improves the performance of the model.Experimental results on three data sets prove the effectiveness of the proposed model.

Key words: graph data, recommendation systems, collaborative filtering, graph convolutional networks

摘要: 图卷积网络(GCN)已应用于各领域并取得巨大成功,其中包括推荐系统。推荐系统的核心目标是推测用户偏好从而主动为用户提供有价值有针对性的消息,协同过滤是经典且广泛应用的一种推荐方法。但基于GCN的推荐方法仍存在诸多问题,如GCN递归地合并来自不同阶邻域的消息,难以区分地混合不同的节点消息导致训练困难,以及过平滑等问题对推荐模型产生了很大的约束。与目前基于GCN的方法不同,针对以上问题,提出使用简单GCN模型分别汇总不同顺序的邻域消息用于协同过滤,然后以分层方式将它们聚合,无需引入其他模型参数。之后,将Dropout思想迁移至模型中,通过在每一层随机丢弃邻居消息来减轻过平滑的影响,很好地防止了过拟合并提升了模型性能。在三个数据集上进行的实验结果证明了所提模型的有效性。

关键词: 图数据, 推荐系统, 协同过滤, 图卷积网络