Computer Science ›› 2021, Vol. 48 ›› Issue (7): 172-177.doi: 10.11896/jsjkx.200600077

• Database & Big Data & Data Science • Previous Articles     Next Articles

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).

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

CLC Number: 

  • 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] CHENG Zhang-tao, ZHONG Ting, ZHANG Sheng-ming, ZHOU Fan. Survey of Recommender Systems Based on Graph Learning [J]. Computer Science, 2022, 49(9): 1-13.
[2] WANG Guan-yu, ZHONG Ting, FENG Yu, ZHOU Fan. Collaborative Filtering Recommendation Method Based on Vector Quantization Coding [J]. Computer Science, 2022, 49(9): 48-54.
[3] CAO Xiao-wen, LIANG Mei-yu, LU Kang-kang. Fine-grained Semantic Reasoning Based Cross-media Dual-way Adversarial Hashing Learning Model [J]. Computer Science, 2022, 49(9): 123-131.
[4] QIN Qi-qi, ZHANG Yue-qin, WANG Run-ze, ZHANG Ze-hua. Hierarchical Granulation Recommendation Method Based on Knowledge Graph [J]. Computer Science, 2022, 49(8): 64-69.
[5] FANG Yi-qiu, ZHANG Zhen-kun, GE Jun-wei. Cross-domain Recommendation Algorithm Based on Self-attention Mechanism and Transfer Learning [J]. Computer Science, 2022, 49(8): 70-77.
[6] SHUAI Jian-bo, WANG Jin-ce, HUANG Fei-hu, PENG Jian. Click-Through Rate Prediction Model Based on Neural Architecture Search [J]. Computer Science, 2022, 49(7): 10-17.
[7] QI Xiu-xiu, WANG Jia-hao, LI Wen-xiong, ZHOU Fan. Fusion Algorithm for Matrix Completion Prediction Based on Probabilistic Meta-learning [J]. Computer Science, 2022, 49(7): 18-24.
[8] SUN Xiao-han, ZHANG Li. Collaborative Filtering Recommendation Algorithm Based on Rating Region Subspace [J]. Computer Science, 2022, 49(7): 50-56.
[9] CAI Xiao-juan, TAN Wen-an. Improved Collaborative Filtering Algorithm Combining Similarity and Trust [J]. Computer Science, 2022, 49(6A): 238-241.
[10] HE Yi-chen, MAO Yi-jun, XIE Xian-fen, GU Wan-rong. Matrix Transformation and Factorization Based on Graph Partitioning by Vertex Separator for Recommendation [J]. Computer Science, 2022, 49(6A): 272-279.
[11] HONG Zhi-li, LAI Jun, CAO Lei, CHEN Xi-liang, XU Zhi-xiong. Study on Intelligent Recommendation Method of Dueling Network Reinforcement Learning Based on Regret Exploration [J]. Computer Science, 2022, 49(6): 149-157.
[12] GUO Liang, YANG Xing-yao, YU Jiong, HAN Chen, HUANG Zhong-hao. Hybrid Recommender System Based on Attention Mechanisms and Gating Network [J]. Computer Science, 2022, 49(6): 158-164.
[13] XIONG Zhong-min, SHU Gui-wen, GUO Huai-yu. Graph Neural Network Recommendation Model Integrating User Preferences [J]. Computer Science, 2022, 49(6): 165-171.
[14] YU Ai-xin, FENG Xiu-fang, SUN Jing-yu. Social Trust Recommendation Algorithm Combining Item Similarity [J]. Computer Science, 2022, 49(5): 144-151.
[15] HOU Hong-xu, SUN Shuo, WU Nier. Survey of Mongolian-Chinese Neural Machine Translation [J]. Computer Science, 2022, 49(1): 31-40.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!