计算机科学 ›› 2021, Vol. 48 ›› Issue (7): 172-177.doi: 10.11896/jsjkx.200600077

• 数据库&大数据&数据科学 • 上一篇    下一篇

基于对抗性学习的协同过滤推荐算法

詹皖江1, 洪植林1, 方路平1, 吴哲夫1, 吕跃华2   

  1. 1 浙江工业大学信息学院 杭州310023
    2 浙江省科技信息研究院 杭州310006
  • 收稿日期:2020-06-12 修回日期:2020-11-26 出版日期:2021-07-15 发布日期:2021-07-02
  • 通讯作者: 吕跃华(lyh@zjinfo.gov.cn)
  • 基金资助:
    浙江省自然科学基金(LY18F010025)

Collaborative Filtering Recommendation Algorithm Based on Adversarial Learning

ZHAN Wan-jiang1, HONG Zhi-lin1, FANG Lu-ping1, WU Zhe-fu1, LYU Yue-hua2   

  1. 1 College of Information Engineering,Zhejiang University of Technology,Hangzhou 310023,China
    2 Zhejiang Institute of Science and Technology Information,Hangzhou 310006,China
  • Received:2020-06-12 Revised:2020-11-26 Online:2021-07-15 Published:2021-07-02
  • About author:ZHAN Wan-jiang,born in 1994,postgraduate.His main research interests include recommendation system and data mining.(wzf@zjut.edu.cn)
    LYU Yue-hua,born in 1978,master.His main research interests include data mining and artificial intelligence.
  • Supported by:
    Natural Science Foundation of Zhejiang Province(LY18F010025).

摘要: 推荐系统能够根据用户的兴趣特点和购买行为,向用户推荐感兴趣的信息和商品。随着用户生成内容UGC逐渐成为当前Web应用的主流,基于UGC的推荐也得到了广泛关注。区别于传统推荐中用户与物品的二元交互,有的UGC推荐采用协同过滤方法,提出了消费者、物品和生产者的三元交互,进而提高了推荐准确度,但大多算法都集中在推荐的性能而忽略了对鲁棒性的研究。因此,通过结合对抗性学习和协同过滤的思想,提出了一种基于对抗性学习的协同过滤推荐算法。首先在三元关系模型参数上加入对抗性扰动,使模型的性能降至最差,与此同时使用对抗性学习的方法训练模型,以达到提高推荐模型鲁棒性的目的;其次设计了一种高效的算法用于求解模型所需的参数;最后在Reddit和Pinterest两个公共数据集上进行测试。实验结果表明:1)在相同参数设置下,与现有算法相比,所提方法的AUC,Precision和Recall指标均有明显的提高,验证了其可行性与有效性;2)该算法不仅增强了推荐性能,还提高了模型的鲁棒性。

关键词: UGC, 对抗性学习, 矩阵分解, 推荐系统, 协同过滤

Abstract: The recommendation system can recommend relevant information and commodities to the user according to the user’s hobbies and purchase behavior.As user-generated content UGC gradually becomes the mainstream of current Web applications,recommendations based on UGC have also received widespread attention.Different from the binary interaction between user and item in traditional recommendation,the existing UGC recommendation adopts collaborative filtering method to propose a ternary interaction between consumer,item and producer,thereby improving the accuracy of recommendation,but most of the algorithms focus on the recommended performance and ignore the research on robustness.Therefore,by combining the ideas of adversarial learning and collaborative filtering,a collaborative filtering recommendation algorithm based on adversarial learning is proposed.First,the adversarial disturbance is added to the ternary relationship model parameters to make the performance of the model the worst.At the same time,the adversarial learning method is used to train the model to achieve the purpose of improving the robustness of the recommendation model.Secondly,an efficient algorithm is designed used to transform the parameters required by the model.Finally,it is tested on two public data sets generated by Reddit and Pinterest.The experimental results show that under the same parameter settings,compared with the existing algorithms,the AUC,Precision and Recall indicators of the proposed algorithm have been significantly improved,verifying its feasibility and effectiveness.The algorithm not only enhances the recommendation performance,but also improves the robustness of the model.

Key words: Adversarial learning, Collaborative filtering, Matrix factorization, Recommendation system, User-generated content

中图分类号: 

  • TP301.6
[1]SMITH B,LINDEN G.Two Decades of Recommender Systems at Amazon.com[J].IEEE Internet Computing,2017,21(3):12-18.
[2]SCHAFER J B,FRANKOWSKI D,HERLOCKER J,et al.Collaborative Filtering Recommender Systems[M]//The Adaptive Web.Berlin:Springer,2007:291-324.
[3]MA W K,LI G,LI Z Y,et al.A Top-N Personalized Recommendation Algorithm Based on Tags[J].Computer Science,2019,46(S2):224-229.
[4]KANG W C,MCAULEY J.Learning Consumer and ProducerEmbeddings for User-Generated Content Recommendation[C]//Proceedings of the 12th ACM Conference on Recommender Systems.2018:407-411.
[5]SZEGEDY C,ZAREMBA W,SUTSKEVER I,et al.Intriguing Properties of Neural Networks[J/OL].Computer Science,2013.http://arxiv.org/pdf/1312.6199.pdf.
[6]ZHANG Z P,GUO X L.Optimized Collaborative Filtering Re-commendation Algorithm Based on Item Rating Prediction[J].Computer Application Research,2008,25(9):2658-2660.
[7]Sarwar B,Karypis G,Konstan J,et al.Item Based Collaborative Filtering Recommendation Algorithms[C]//Proceedings of the 10th International Conference on World Wide Web.2001:285-295.
[8]HE X,ZHANG H,KAN M Y,et al.Fast Matrix Factorization for Online Recommendation with Implicit Feedback[C]//Proceedings of the 39th International ACM SIGIR Conference on Research and Development in Information Retrieval.2016:549-558.
[9]HU G N,DAI X Y,QIU F Y,et al.Collaborative Filtering withTopic and Social Latent Factors Incorporating Implicit Feedback[J].ACM Transactions on Knowledge Discovery from Data(TKDD),2018,12(2):1-30.
[10]WANG C,ZHU H,ZHU C,et al.SetRank:A Setwise Bayesian Approach for Collaborative Ranking from Implicit Feedback[J/OL].2020.http://arxiv.org/pdf/2002.09841.pdf.
[11]YU W H,ZHANG H D,HE X N,et al.Aesthetic-based Clo-thing Recommendation[C]//the 2018 World Wide Web Confe-rence.2018:649-658.
[12]LI X,JIANG M,HONG H,et al.A Time-aware PersonalizedPoint-of-Interest Recommendation via High-order Tensor Factorization[J].ACM Transactions on Information Systems(TOIS),2017,35(4):1-23.
[13]WU Y,BAMMAN D,RUSSELL S.Adversarial Training forRelation Extraction[C]//Proceedings of the 2017 Conference on Empirical Methods in Natural Language Processing.2017:1778-1783.
[14]LEE D,KIM J,MOON W J,et al.CollaGAN:CollaborativeGAN for Missing Image Data Imputation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2019:2487-2496.
[15]WANG J,YU L,ZHANG W,et al.IRGAN:A Minimax Game for Unifying Generative and Discriminative Information Retrie-val Models[C]//Proceedings of the 40th International ACM SIGIR Conference on Research and Development in Information Retrieval.2017:515-524.
[16]WANG H,JIA W,WANG J,et al.GraphGAN:Graph Repre-sentation Learning with Generative Adversarial Nets[J].IEEE Transactions on Knowledge and Data Engineering,2017(99):2508-2515.
[17]CHAE D K,KANG J S,KIM S W,et al.CfGAN:A Generic Collaborative Filtering Framework Based on Generative Adversarial Networks[C]//Proceedings of the 27th ACM Internatio-nal Conference on Information and Knowledge Management.2018:137-146.
[18]KOREN Y.Factorization Meets the Neighborhood:A Multifa-ceted Collaborative Filtering Model[C]//Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.2008:426-434.
[19]RENDLE S,FREUDENTHALER C,GANTNER Z,et al.BPR:Bayesian Personalized Ranking from Implicit Feedback[C]//Proceedings of the 25th Conference on Uncertainty in Artificial Intelligence.2009:452-461.
[20]HE X N,HE Z,DU X,et al.Adversarial Personalized Ranking for Recommendation[C]//The 41st International ACM SIGIR Conference on Research & Development in Information Retrie-val.2018:355-364.
[21]GOODFELLOW I J,SHLENS J,SZEGEDY C.Explaining and Harnessing Adversarial Examples[J].arXiv:1412.6572,2014.
[22]ZHONG C T,SHAH S,SASTRY N,et al.Sharing the Loves:Understanding the How and Why of Online Content Curation[C]//Proceedings of Seventh International AAAI Conference on Weblogs and Social Media Sharing.2013:659-667.
[23]RENDLE S.Factorization Machines[C]//2010 IEEE International Conference on Data Mining.IEEE,2010:995-1000.
[1] 程章桃, 钟婷, 张晟铭, 周帆.
基于图学习的推荐系统研究综述
Survey of Recommender Systems Based on Graph Learning
计算机科学, 2022, 49(9): 1-13. https://doi.org/10.11896/jsjkx.210900072
[2] 王冠宇, 钟婷, 冯宇, 周帆.
基于矢量量化编码的协同过滤推荐方法
Collaborative Filtering Recommendation Method Based on Vector Quantization Coding
计算机科学, 2022, 49(9): 48-54. https://doi.org/10.11896/jsjkx.210700109
[3] 秦琪琦, 张月琴, 王润泽, 张泽华.
基于知识图谱的层次粒化推荐方法
Hierarchical Granulation Recommendation Method Based on Knowledge Graph
计算机科学, 2022, 49(8): 64-69. https://doi.org/10.11896/jsjkx.210600111
[4] 方义秋, 张震坤, 葛君伟.
基于自注意力机制和迁移学习的跨领域推荐算法
Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning
计算机科学, 2022, 49(8): 70-77. https://doi.org/10.11896/jsjkx.210600011
[5] 帅剑波, 王金策, 黄飞虎, 彭舰.
基于神经架构搜索的点击率预测模型
Click-Through Rate Prediction Model Based on Neural Architecture Search
计算机科学, 2022, 49(7): 10-17. https://doi.org/10.11896/jsjkx.210600009
[6] 齐秀秀, 王佳昊, 李文雄, 周帆.
基于概率元学习的矩阵补全预测融合算法
Fusion Algorithm for Matrix Completion Prediction Based on Probabilistic Meta-learning
计算机科学, 2022, 49(7): 18-24. https://doi.org/10.11896/jsjkx.210600126
[7] 孙晓寒, 张莉.
基于评分区域子空间的协同过滤推荐算法
Collaborative Filtering Recommendation Algorithm Based on Rating Region Subspace
计算机科学, 2022, 49(7): 50-56. https://doi.org/10.11896/jsjkx.210600062
[8] 蔡晓娟, 谭文安.
一种改进的融合相似度和信任度的协同过滤算法
Improved Collaborative Filtering Algorithm Combining Similarity and Trust
计算机科学, 2022, 49(6A): 238-241. https://doi.org/10.11896/jsjkx.210400088
[9] 何亦琛, 毛宜军, 谢贤芬, 古万荣.
基于点割集图分割的矩阵变换与分解的推荐算法
Matrix Transformation and Factorization Based on Graph Partitioning by Vertex Separator for Recommendation
计算机科学, 2022, 49(6A): 272-279. https://doi.org/10.11896/jsjkx.210600159
[10] 郭亮, 杨兴耀, 于炯, 韩晨, 黄仲浩.
基于注意力机制和门控网络相结合的混合推荐系统
Hybrid Recommender System Based on Attention Mechanisms and Gating Network
计算机科学, 2022, 49(6): 158-164. https://doi.org/10.11896/jsjkx.210500013
[11] 熊中敏, 舒贵文, 郭怀宇.
融合用户偏好的图神经网络推荐模型
Graph Neural Network Recommendation Model Integrating User Preferences
计算机科学, 2022, 49(6): 165-171. https://doi.org/10.11896/jsjkx.210400276
[12] 洪志理, 赖俊, 曹雷, 陈希亮, 徐志雄.
基于遗憾探索的竞争网络强化学习智能推荐方法研究
Study on Intelligent Recommendation Method of Dueling Network Reinforcement Learning Based on Regret Exploration
计算机科学, 2022, 49(6): 149-157. https://doi.org/10.11896/jsjkx.210600226
[13] 余皑欣, 冯秀芳, 孙静宇.
结合物品相似性的社交信任推荐算法
Social Trust Recommendation Algorithm Combining Item Similarity
计算机科学, 2022, 49(5): 144-151. https://doi.org/10.11896/jsjkx.210300217
[14] 陈壮, 邹海涛, 郑尚, 于化龙, 高尚.
基于用户覆盖及评分差异的多样性推荐算法
Diversity Recommendation Algorithm Based on User Coverage and Rating Differences
计算机科学, 2022, 49(5): 159-164. https://doi.org/10.11896/jsjkx.210300263
[15] 陈晋鹏, 胡哈蕾, 张帆, 曹源, 孙鹏飞.
融合时间特性和用户偏好的卷积序列化推荐
Convolutional Sequential Recommendation with Temporal Feature and User Preference
计算机科学, 2022, 49(1): 115-120. https://doi.org/10.11896/jsjkx.201200192
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!