Computer Science ›› 2022, Vol. 49 ›› Issue (11A): 211100106-8.doi: 10.11896/jsjkx.211100106

• Big Data & Data Science • Previous Articles     Next Articles

Movie Recommendation Model Based on Attribute Graph Attention Network

SUN Kai-wei, LIU Song, DU Yu-lu   

  1. School of Computer Science and Technology,Chongqing University of Posts and Telecommunications,Chongqing 400065,China
  • Online:2022-11-10 Published:2022-11-21
  • About author:SUN Kai-wei,born in 1987,Ph.D,associate professor,master tutor.His main research interests include machine lear-ning,data mining and big data analysis.
    DU Yu-lu,born in 1987,Ph.D,lecturer.His main research interests include personalized recommendation algorithms.
  • Supported by:
    National Natural Science Foundation of China(61806033) and Natural Science Foundation of Chongqing(cstc2019jcyj-msxmX0021).

Abstract: In recent years,graph network has been widely used in the field of recommendation and made a great progress.How-ever,the existing methods tend to focus on the interaction modeling of user projects,so the performance is limited by the problem of data sparsity.Therefore,this paper proposes a movie recommendation model based on graph attention network of attribute graph by using additional attribute information.Firstly,an attention-based GNN is proposed,which uses explicit feed-back to calculate the attention score between entities and attributes.Compared with the aggregation method using Laplace matrix,it can distinguish the influence of different attributes on entities more effectively,and the information aggregation between attributes and entities can be more effective.In addition,different entities are affected differently by attributes and behaviors,a fine-grained pre-ference fusion strategy is designed in this paper to calculate a set of preference fusion weights for each entity to make the embedding representation of entities more accurate and personalized.Experimental results on real data set show that the recommendation method that makes full use of attribute information contained in attribute graph can effectively alleviate the problem of data sparsity and is significantly better than other basic algorithm in terms of recall rate and nDCG,two evaluation indexes of movie recommendation.

Key words: Graph neural network, Attribute graph, Attention mechanism, Explicit feed-back, Movie recommendation

CLC Number: 

  • TP311
[1]LEE A R,SON S M,KIM K K.Information and communication technology overload and social networking service fatigue:A stress perspective[J].Computers in Human Behavior,2016,55(8):51-61.
[2]WANG C D,DENG Z H,LAI J H,et al.Serendipitous Recommendation in E-Commerce Using Innovator-Based Collaborative Filtering[J].IEEE Transactions on Cybernetics,2018,49(7):2678-2692.
[3]KOREN Y,BELL R,VOLINSKY C.Matrix Factorization Tech-niques for Recommender Systems[J].Computer,2009,42(8):30-37.
[4]XI W D,HUANG L,WANG C D,et al.BPAM:Recommendation Based on BP Neural Network with Attention Mechanism[C]//Twenty-Eighth International Joint Conference on Artificial Intelligence.California:IJCAI,2019:3905-3911.
[5]LEI W,ZHANG G,HE X,et al.Interactive path reasoning on graph for conversational recommendation[C]//Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining.Virtual Event CA USA:ACM,2020:2073-2083.
[6]WANG X,JIN H,ZHANG A,et al.Disentangled Graph Colla-borative Filtering[C]//Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval.Virtual Event China:ACM,2020:1001-1010.
[7]RAO N,YU H F,RAVIKUMAR P K,et al.Collaborative filtering with graph information:Consistency and scalable methods[C]//Proceedings of the 28th International Conference on Neural Information Processing Systems.Cambridge:MIT Press,2015:2107-2115.
[8]XU C,ZHAO P,LIU Y,et al.Graph Contextualized Self-Attention Network for Session-based Recommendation[C]//Procee-dings of the Twenty-Eighth International Joint Conference on Artificial Intelligence.Macao,China:DBLP,2019:3940-3946.
[9]CHEN T,ZHU Q,ZHOU M X,et al.Trust-Based Recommendation Algorithm in Social Network[J].Journal of Software,2017,28(3):721-731.
[10]WANG X,HE X,WANG M,et al.Neural graph collaborativefiltering[C]//Proceedings of the 42nd international ACM SIGIR Conference on Research and Development in Information Retrieval.Paris,France:ACM,2019:165-174.
[11]BERG R,KIPF T N,WELLING M.Graph convolutional matrix completion[C]//Proceedings ACM SIGKDD Int.Conf.Know-ledge Discovery and Data Mining(Deep Learning Day).London,UK,2018:1-9.
[12]HE X,DENG K,WANG X,et al.Lightgcn:Simplifying andpowering graph convolution network for recommendation[C]//Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval.Virtual Event China:ACM,2020:639-648.
[13]WU L,SUN P,HONG R,et al.A neural influence diffusion model for social recommendation[C]//Proceedings of the 42th International ACM SIGIR Conference on Research and Development in Information Retrieval.Paris,France:ACM,2019:2776-2783.
[14]GUO Z,WANG H.A deep graph neural network-based mechanism for social recommendations[J].IEEE Transactions on Industrial Informatics,2020,17(4):2776-2783.
[15]QIAN T,LIANG Y,LI Q,et al.Attribute Graph Neural Networks for Strict Cold Start Recommendation[J].IEEE Transactions on Knowledge and Data Engineering,2022,34(8):3597-3610.
[16]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,Montreal,QC,Canada:AUAI Press,2009:452-461.
[17]XUE H J,DAI X,ZHANG J,et al.Deep Matrix Factorization Models for Recommender Systems[C]//Twenty-Sixth International Joint Conference on Artificial Intelligence.Melbourne,Australia:AAAI Press,2017:3203-3209.
[18]HE X,LIAO L,ZHANG H,et al.Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web.Perth,Australia,2017:173-182.
[19]LEI W,HE X,MIAO Y,et al.Estimation-action-reflection:Towards deep interaction between conversational and recommender systems[C]//Proceedings of the 13th International Conference on Web Search and Data Mining.Houston,TX,USA:ACM,2020:304-312.
[20]CHEN J,ZHANG H,HE X,et al.Attentive collaborative filtering:Multimedia recommendation with item-and component-le-vel attention[C]//Proceedings of the 40th International ACM SIGIR Conference on Research and Development in Information Retrieval.Tokyo,Japan:ACM,2017:335-344.
[21]HU H,HE X.Sets2sets:Learning from sequential sets withneural networks[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mi-ning.Anchorage,AK,USA:ACM,2019:1491-1499.
[22]GUAN X,CHENG Z,HE X,et al.Attentive aspect modeling for review-aware recommendation[J].ACM Transactions on Information Systems(TOIS),2019,37(3):1-27.
[23]XIN X,HE X,ZHANG Y,et al.Relational collaborative filtering:Modeling multiple item relations for recommendation[C]//Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval.Paris,France:ACM,2019:125-134.
[24]LIU S,OUNIS I,MACDONALD C,et al.A HeterogeneousGraph Neural Model for Cold-Start Recommendation[C]//Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval.Virtual Event,China:ACM,2020:2029-2032.
[25]GE Y,CHEN S C.Graph Convolutional Network for Recom-mender Systems[J].Journal of Software,2020,31(4):1101-1112.
[1] RAO Zhi-shuang, JIA Zhen, ZHANG Fan, LI Tian-rui. Key-Value Relational Memory Networks for Question Answering over Knowledge Graph [J]. Computer Science, 2022, 49(9): 202-207.
[2] ZHOU Fang-quan, CHENG Wei-qing. Sequence Recommendation Based on Global Enhanced Graph Neural Network [J]. Computer Science, 2022, 49(9): 55-63.
[3] DAI Yu, XU Lin-feng. Cross-image Text Reading Method Based on Text Line Matching [J]. Computer Science, 2022, 49(9): 139-145.
[4] ZHOU Le-yuan, ZHANG Jian-hua, YUAN Tian-tian, CHEN Sheng-yong. Sequence-to-Sequence Chinese Continuous Sign Language Recognition and Translation with Multi- layer Attention Mechanism Fusion [J]. Computer Science, 2022, 49(9): 155-161.
[5] XIONG Li-qin, CAO Lei, LAI Jun, CHEN Xi-liang. Overview of Multi-agent Deep Reinforcement Learning Based on Value Factorization [J]. Computer Science, 2022, 49(9): 172-182.
[6] JIANG Meng-han, LI Shao-mei, ZHENG Hong-hao, ZHANG Jian-peng. Rumor Detection Model Based on Improved Position Embedding [J]. Computer Science, 2022, 49(8): 330-335.
[7] WANG Ming, PENG Jian, HUANG Fei-hu. Multi-time Scale Spatial-Temporal Graph Neural Network for Traffic Flow Prediction [J]. Computer Science, 2022, 49(8): 40-48.
[8] ZHU Cheng-zhang, HUANG Jia-er, XIAO Ya-long, WANG Han, ZOU Bei-ji. Deep Hash Retrieval Algorithm for Medical Images Based on Attention Mechanism [J]. Computer Science, 2022, 49(8): 113-119.
[9] SUN Qi, JI Gen-lin, ZHANG Jie. Non-local Attention Based Generative Adversarial Network for Video Abnormal Event Detection [J]. Computer Science, 2022, 49(8): 172-177.
[10] YAN Jia-dan, JIA Cai-yan. Text Classification Method Based on Information Fusion of Dual-graph Neural Network [J]. Computer Science, 2022, 49(8): 230-236.
[11] ZENG Zhi-xian, CAO Jian-jun, WENG Nian-feng, JIANG Guo-quan, XU Bin. Fine-grained Semantic Association Video-Text Cross-modal Entity Resolution Based on Attention Mechanism [J]. Computer Science, 2022, 49(7): 106-112.
[12] XU Ming-ke, ZHANG Fan. Head Fusion:A Method to Improve Accuracy and Robustness of Speech Emotion Recognition [J]. Computer Science, 2022, 49(7): 132-141.
[13] MENG Yue-bo, MU Si-rong, LIU Guang-hui, XU Sheng-jun, HAN Jiu-qiang. Person Re-identification Method Based on GoogLeNet-GMP Based on Vector Attention Mechanism [J]. Computer Science, 2022, 49(7): 142-147.
[14] JIN Fang-yan, WANG Xiu-li. Implicit Causality Extraction of Financial Events Integrating RACNN and BiLSTM [J]. Computer Science, 2022, 49(7): 179-186.
[15] XIONG Luo-geng, ZHENG Shang, ZOU Hai-tao, YU Hua-long, GAO Shang. Software Self-admitted Technical Debt Identification with Bidirectional Gate Recurrent Unit and Attention Mechanism [J]. Computer Science, 2022, 49(7): 212-219.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!