GraphSAGE与遗传算法:推荐系统中的聚类性能优化
GraphSAGE and Genetic Algorithms: Clustering Performance Optimization in Recommender Systems
摘要: 在信息过载的背景下,如何提高推荐系统的推荐质量和准确性成为了研究热点,而图神经网络作为一种新兴的技术,在这方面展现出了巨大的潜力。本研究首先将四种广泛使用的图神经网络模型GAT、GCN、GIN和GraphSAGE应用于推荐系统中,并通过比较发现GraphSAGE在多项指标上的表现最优。具体来说,GraphSAGE在推荐系统上实现了RMSE (0.6459)、MAE (0.4989)、AUC (0.8701)、Recall (0.6377)、Precision (0.8348)和F1-score (0.7231)的优异性能。进一步,本研究将GraphSAGE与传统的协同过滤算法在推荐系统上的表现进行比较,结果表明GraphSAGE在推荐系统上仍保持优势。在此基础上,本研究还探索了GraphSAGE与遗传算法优化K-means的结合算法——GraphKGA算法对于推荐系统性能的影响,实验证明,GraphKGA算法应用于推荐系统中的RMSE (0.6364)和MAE (0.4855)均取得优异结果,表明其能有效提高推荐系统的性能。
Abstract: In the context of information overload, how to improve the quality and accuracy of recommendation systems has become a research hotspot, and graph neural networks, as an emerging technology, have shown great potential in this regard. This study first applied four widely used graph neural network models, GAT, GCN, GIN, and GraphSAGE, to the recommendation system, and found through comparison that GraphSAGE performed best in multiple indicators. Specifically, GraphSAGE achieved excellent performance in RMSE (0.6459), MAE (0.4989), AUC (0.8701), Recall (0.6377), Precision (0.8348), and F1-score (0.7231) in the recommendation system. Furthermore, this study compared the performance of GraphSAGE with traditional collaborative filtering algorithms in the recommendation system, and the results showed that GraphSAGE still maintained its advantage in the recommendation system. On this basis, this study also explored the impact of the GraphKGA algorithm, which is a combination of GraphSAGE and genetic algorithm optimized K-means, on the performance of the recommendation system. Experiments have shown that the RMSE (0.6364) and MAE (0.4855) of the GraphKGA algorithm applied to the recommendation system have achieved excellent results, indicating that it can effectively improve the performance of the recommendation system.
文章引用:代梦飞. GraphSAGE与遗传算法:推荐系统中的聚类性能优化[J]. 运筹与模糊学, 2024, 14(5): 400-407. https://doi.org/10.12677/orf.2024.145481

参考文献

[1] Feng, J., Xia, Z., Feng, X. and Peng, J. (2021) RBPR: A Hybrid Model for the New User Cold Start Problem in Recommender Systems. Knowledge-Based Systems, 214, Article 106732. [Google Scholar] [CrossRef
[2] Wu, S., Sun, F., Zhang, W., Xie, X. and Cui, B. (2022) Graph Neural Networks in Recommender Systems: A Survey. ACM Computing Surveys, 55, 1-37. [Google Scholar] [CrossRef
[3] Berg, R.V.D., Kipf, T.N. and Welling, M. (2017) Graph Convolutional Matrix Completion.
https://www.kdd.org/kdd2018/files/deep-learning-day/DLDay18_paper_32.pdf
[4] 杨宇森. 融合图神经网络的协同过滤算法的改进研究[D]: [硕士学位论文]. 大连: 大连交通大学.
[5] 吕艳霞, 郝帅, 乔广通, 等. 一种基于图神经网络的社会化推荐算法[J]. 东北大学学报(自然科学版), 2024, 45(1): 10-17.
[6] 张安勤, 李然, 田秀霞. 基于图神经网络的社会感知顺序推荐模型[J]. 计算机应用与软件, 2024, 41(3): 246-252+282.
[7] 刘婷, 郭海湘, 诸克军, 等. 一种改进的遗传k-means聚类算法[J]. 数学的实践与认识, 2007, 37(8): 104-111.
[8] 王敞, 陈增强, 袁著祉. 基于遗传算法的K均值聚类分析[J]. 计算机科学, 2003, 30(2): 163-164.
[9] 吕强, 俞金寿. 基于混合遗传算法的K-Means最优聚类算法[J]. 华东理工大学学报(自然科学版), 2005, 31(2): 219-222.
[10] 戴文华, 焦翠珍, 何婷婷. 基于并行遗传算法的K-means聚类研究[J]. 计算机科学, 2008, 35(6): 171-174.
[11] 赖玉霞, 刘建平, 杨国兴. 基于遗传算法的K均值聚类分析[J]. 计算机工程, 2008, 34(20): 200-202.
[12] 刘天航, 杨晓雪, 周慧, 等. 基于图神经网络的协同过滤推荐算法综述[J]. 集成技术, 2024, 13(4): 1-15.
[13] 金玲, 刘晓丽, 李鹏飞, 等. 遗传算法综述[J]. 科学中国人, 2015(9X): 230.
[14] MacQueen, J. (1967) Some Methods for Classification and Analysis of Multivariate Observations. In: Berkeley Symposium on Mathematical Statistics and Probability, University of California Press, 1-15.