融合社交关系和知识图谱的推荐算法

doi:10.3778/j.issn.1673-9418.2112088

摘要/Abstract

摘要： 推荐系统可以帮助用户快速发现有用信息，有效提高用户的检索效率，然而推荐系统存在数据稀疏性、冷启动等问题，现有的融合了社交关系的推荐算法大多忽略了社交关系数据的稀疏性，且同时融合社交关系和物品属性数据的推荐算法较少。为解决这方面的问题，提出了一种融合社交关系和知识图谱的推荐算法（MSAKR）。首先，该算法通过图卷积神经网络提取用户的社交关系得到用户的特征向量，采用图中心性筛选邻居，采用word2vec模型思想生成虚拟邻居，从而缓解社交数据的稀疏性，采用注意力机制来聚集邻居；其次，采用多任务学习和基于语义的匹配模型来提取物品属性知识图谱信息得到物品的特征向量；最后，根据得到的用户和物品特征向量向用户综合推荐。为验证提出算法的性能，在真实数据集豆瓣和Yelp上进行实验验证，分别使用点击率预测和[Top-K]推荐来评估模型性能，实验结果表明，提出的模型优于其他的基准模型。

Abstract: Recommendation system can help users quickly find useful information and improve the retrieval efficiency of users effectively. However, the recommendation system has problems such as data sparsity and cold start, most of the existing recommendation algorithms that integrate social relations ignore the sparsity of social relations data, and there are few recommendation algorithms that integrate social relations and item attribute data at the same time. This paper proposes a recommendation model that is multi-task feature learning approach for social relationship and knowledge graph enhanced recommendation (MSAKR) in response to solve the above problems. Firstly, the algorithm extracts the user’s social relations through the graph convolutional neural network to get the user’s feature vector, then selects the neighbor by the graph centrality, and generates the virtual neighbor by the word2vec model, so as to alleviate the sparsity of the social data. This paper uses the attention mechanism to gather the neighbors. Secondly, multi-task learning and semantic-based matching model are used to extract the information of attribute knowledge graph to obtain the feature vector of the item. Finally, comprehensive recommendation is made to the user based on the obtained user and item feature vectors. In order to assess the performance of the recommendation algorithm, experiments are carried out on real datasets Douban and Yelp. Click-through rate predi-ction and Top-K recommendation are used to evaluate the performance of the model respectively. Experimental results show that the proposed model is superior to other benchmark models.

Key words: recommendation algorithm, graph convolutional neural network, social network, knowledge graph, multi-task learning

高仰, 刘渊. 融合社交关系和知识图谱的推荐算法[J]. 计算机科学与探索, 2023, 17(1): 238-250.

GAO Yang, LIU Yuan. Recommendation Algorithm Combining Social Relationship and Knowledge Graph[J]. Journal of Frontiers of Computer Science and Technology, 2023, 17(1): 238-250.

参考文献

[1] HUANG L W, JIANG B T, LV S Y, et al. Survey on deep learning based recommender systems[J]. Chinese Journal of Computers, 2018, 41(7): 1619-1647.
[2] ZHOU W Z, CAO D, XU Y F, et al. A survey of recom-mendation systems[J]. Journal of Hebei University of Sci-ence and Technology, 2020, 41(1): 76-87.
[3] WANG J Z, KONG L W, HUANG Z C, et al. The survey of graph neural network[J]. Computer Engineering, 2021, 47(4): 1-12.
[4] DEFFERRARD M, BRESSON X, VANDERGHEYNST P, et al. Convolutional neural networks on graphs with fast loca-lized spectral filtering[C]//Advances in Neural Information Processing Systems 29, Barcelona, Dec 5-10, 2016: 3844-3852.
[5] MICHELI A. Neural network for graphs: a contextual cons-tructive approach[J]. IEEE Transactions on Neural Networks, 2009, 20(3): 498-511.
[6] ATWOOD J, TOWSLEY D. Diffusion-convolutional neural networks[C]//Advances in Neural Information Processing Systems 29, Barcelona, Dec 5-10, 2016: 1993-2001.
[7] HAMILTON W L, YING R, LESKOVEC J. Inductive rep-resentation learning on large graphs[C]//Advances in Neural Information Processing Systems 30, Long Beach, Dec 4-9, 2017: 1024-1034.
[8] VELICKOVIC P, GUILLEM C, ARANTXA C, et al. Graph attention networks[J]. arXiv:1710.10903, 2017.
[9] YU J L, YIN H Z, LI J D, et al. Self-supervised multi-channel hypergraph convolutional network for social recommen-dation[C]//Proceedings of the Web Conference 2021, Ljubl-jana, Apr 19-23, 2021. New York: ACM, 2021: 413-424.
[10] QIN C, ZHU H S, ZHUANG F Z, et al. A survey on knowledge graph-based recommender system[J]. Scientia Sinica (Informationis), 2020, 50(7): 937-956.
[11] BORDES A, USUNIER N, GARCíA-DURáN A, et al. Translating embeddings for modeling multi-relational data[C]//Advances in Neural Information Processing Systems 26, Lake Tahoe, Dec 5-8, 2013: 2787-2795.
[12] LUO A G, GAO S, XU Y J. Deep semantic match model for entity linking using knowledge graph and text[J]. Pro-cedia Computer Science, 2018, 129: 110-114.
[13] WANG H W, ZHAO M, XIE X, et al. Knowledge graph convolutional networks for recommender systems[C]//Pro-ceedings of the World Wide Web Conference, San Fran-cisco, May 13-17, 2019. New York: ACM, 2019: 3307-3313.
[14] CHOWDHURY G, SRILAKSHMI M, CHAIN M, et al. Neural factorization for offer recommendation using know-ledge graph embeddings[C]//Proceedings of the SIGIR 2019 Workshop on eCommerce, Co-located with the 42nd Interna-tional ACM SIGIR Conference on Research and Develop-ment in Information Retrieval, Paris, Jul 25, 2019: 1-6.
[15] HU B B, SHI C, ZHAO W Y, et al. Leveraging meta-path based context for top-n recommendation with a neural co-attention model[C]//Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, London, Aug 19-23, 2018. New York: ACM, 2018: 1531-1540.
[16] MOONEY R J, ROY L. Content-based book recommen-ding using learning for text categorization[C]//Proceedings of the 5th ACM Conference on Digital Libraries, San Anto-nio, Jun 2-7, 2000. New York: ACM, 2000: 195-204.
[17] GOLDBERG D, NICHOLS D, OKI B M, et al. Using colla-borative filtering to weave an information tapestry[J]. Com-munications of the ACM, 1992, 35(12): 61-70.
[18] MARKO B, YOAV S. Fab: content-based, collaborative reco-mmendation[J]. Communications of the ACM, 1997, 40(3): 66-72.
[19] ZHANG S, YAO L N, SUN A X, et al. Deep learning based recommender system: a survey and new perspectives[J]. ACM Computing Surveys, 2019, 52(1): 1-38.
[20] KIM D, PARK C, OH J, et al. Convolutional matrix facto-rization for document context-aware recommendation[C]//Proceedings of the 10th ACM Conference on Recommen-der Systems, Boston, Sep 15-19, 2016. New York: ACM, 2016: 233-240.
[21] BANSAL T，BELANGER D, MCCALLUM A. Ask the GRU: multi-task learning for deep text recommendations[C]//Proceedings of the 10th ACM Conference on Recom-mender Systems, Boston, Sep 7, 2016. New York: ACM, 2016: 107-114.
[22] WANG H W, ZHANG F Z, ZHAO M, et al. Multi-task feature learning for knowledge graph enhanced recommen-dation[C]//Proceedings of The World Wide Web Conference, San Francisco, May 13, 2019. New York: ACM, 2019: 2000-2010.
[23] JAMALI M, ESTER M. A matrix factorization technique with trust propagation for recommendation in social net-works[C]//Proceedings of the 4th ACM Conference on Reco-mmender Systems, Barcelona, Sep 23-26, 2010. New York: ACM, 2010: 135-142.
[24] WU L, SUN P J, FU Y J, et al. A neural influence diffusion model for social recommendation[C]//Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval, Paris, Jul 18, 2019. New York: ACM, 2019: 235-244.
[25] FAN W Q, MA Y, LI Q, et al. A graph neural network framework for social recommendations[J]. IEEE Transac-tions on Knowledge and Data Engineering, 2022, 34(5): 2033-2047.
[26] MA H, YANG H X, LYU M R, et al. SoRec: social recom-mendation using probabilistic matrix factorization[C]//Procee-dings of the 17th ACM Conference on Information and Knowledge Management, Napa Valley, Oct 26, 2008. New York: ACM, 2008: 931-940.
[27] GUO G B, ZHANG J, YORKE-SMITH N. TrustSVD: collaborative filtering with both the explicit and implicit influence of user trust and of item ratings[C]//Proceedings of the 29th AAAI Conference on Artificial Intelligence, Austin, Jan 25-30, 2015. Menlo Park: AAAI, 2015: 123-129.
[28] FU B R, ZHANG W M, HU G N, et al. Dual side deep context-aware modulation for social recommendation[C]//Proceedings of the Web Conference 2021, Ljubljana, Apr 19-23, 2021. New York: ACM, 2021: 2524-2534.
[29] SONG W P, XIAO Z P, WANG Y F, et al. Session-based social recommendation via dynamic graph attention net-works[C]//Proceedings of the 12th ACM International Con-ference on Web Search and Data Mining, Melbourne, Feb 11-15, 2019. New York: ACM, 2019: 555-563.
[30] CHEN C, ZHANG M, WANG C Y, et al. An efficient adap-tive transfer neural network for social-aware recommenda-tion[C]//Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval, Paris, Jul 21-25, 2019. New York: ACM, 2019: 225-234.
[31] KOREN Y, BELL R, VOLINSKY C. Matrix factorization techniques for recommender systems[J]. Computer, 2009, 42(8): 30-37.