Computer Science ›› 2022, Vol. 49 ›› Issue (6): 165-171.doi: 10.11896/jsjkx.210400276

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

Graph Neural Network Recommendation Model Integrating User Preferences

XIONG Zhong-min, SHU Gui-wen, GUO Huai-yu   

  1. College of Information,Shanghai Ocean University,Shanghai 201306,China
  • Received:2021-04-26 Revised:2021-05-25 Online:2022-06-15 Published:2022-06-08
  • About author:XIONG Zhong-min,born in 1971,Ph.D,postdoctor,associate professor,is a member of China Computer Federation.His main research interests include database theory and application,data warehouse and data mining,network analysis technology and information recommendation.
  • Supported by:
    National Natural Science Foundation of China(41501419) and Shanghai Local College Capacity Building Project(19050502100).

Abstract: Aiming at the problem that knowledge graph-driven graph neural network recommendation algorithm cannot learn the user and item representations at the same time,a graph neural network recommendation model that integrates user preferences is proposed.The model learns user and item representations from user’s perspective and entity’s perspective respectively.Firstly,the user’s perspective spreads user preferences in the knowledge graph based on user historical interaction records and enhances user representation.Secondly,the entity perspective gathers neighbor information of candidate entities through graph convolu-tional network to enrich the representation of the entity.At the same time,a hybrid layer is designed to capture high-level connectivity and hybrid hierarchical information from both the width and depth aspects to enhance the item representation.The enhanced user representation vector and item representation vector are input to the prediction function to predict the interaction probability.Finally,the fixed-size sampling method and phased training strategy are used to optimize the model.The click-through rate prediction experiment is conducted on the MovieLens-1M data set,and the results show that,compared with the benchmark methods RippleNet and KGCN,its AUC increases by 1.7% and 2.3% respectively.

Key words: Graph neural network, Knowledge graph, Personalized recommendation, Preference propagation, Recommendation system

CLC Number: 

  • TP391.3
[1] HE X N,LIAO L Z,ZHANG H W,et al.Neural CollaborativeFiltering[C]//Proceedings of the 26th International Conference on World Wide Web.New York,USA:ACM,2017:173-182.
[2] JAMALI M,ESTER M.A matrix factorization technique with trust propagation for recommendation in social networks[C]//Proceedings of the 4th ACM Conference on Recommender Systems.New York,USA:ACM,2010:135-142.
[3] ZHANG F,YUAN N,LIAN D,et al.Collaborative knowledge base embedding for recommender systems[C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.San Francisco,USA:ACM,2016:353-362.
[4] SUN Y,YUAN N,XIE X,et al.Collaborative Intent Prediction with Real-Time Contextual Data[J].ACM Transactions on Information Systems,2017:35(4):30:1-30:33.
[5] QIN C,ZHU H S,ZHUANG F Z,et al.Research review ofrecommendation system based on knowledge graph[J].Science in China:Information Science,2020,50(7):937-956.
[6] YU X,REN X,GU Q,et al.Personalized entity recommendation:A heterogeneous information network approach[C]//Proceedings of the 7th ACM International Conference on Web Search and Data Mining.New York,USA:ACM,2014:283-292.
[7] HUANG Z,ZHENG Y,CHENG R,et al.Meta structure:Computing relevance in large heterogeneous information networks[C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.San Francisco,USA:ACM,2016:1595-1604.
[8] HU B,SHI C,ZHAO W,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 & Data Mining.London,UK,2018:1531-1540.
[9] HUANG J,ZHAO W,DOU H,et al.Improving sequentialrecommendation with knowledge-enhanced memory networks[C]//The 41st International ACM SIGIR Conference on Research & Development in Information Retrieval.New York,USA,2018:505-514.
[10] WANG H W,ZHANG F Z,XIE X,et al.DKN:Deep know-ledge-aware network for news recommendation[C]//Procee-dings of the 2018 World Wide Web Conference.Lyon,France,2018:1835-1844.
[11] BORDES A,USUNIER N,GARCIA-DURAN A,et al.Translating embeddings for modeling multi-relational data[J/OL].Advances in Neural Information Processing Systems.https://dblp.uni-trier.de/rec/conf/nips/BordesUGWY13.html.
[12] WANG Z,ZHANG J,FENG J,et al.Knowledge graph embedding by translating on hyperplanes[C]//Proceedings of the 28th AAAI Conference on Artificial Intelligence.Quebec,Canada,2014:1112-1119.
[13] LIN Y,LIN Z,SUN M,et al.Learning entity and relation embeddings for knowledge graph completion[C]//Proceedings of the 29th AAAI Conference on Artificial Intelligence.Texas,USA,2015:2181-2187.
[14] JI G,HE S,XU L,et al.Knowledge graph embedding via dynamic mapping matrix[C]//Proceedings of the 53rd Annual Meeting of the Association for Computational Linguistics and the 7th International Joint Conference on Natural Language Processing.Beijing,China,2015:687-696.
[15] SOCHER R,CHEN D,MANNING C D,et al,Reasoning with neural tensor networks for knowledge base completion[C]//Annual Conference on Neural Information Processing Systems.2013:926-934.
[16] GUAN N,SONG D,LIAO L,Knowledge graph embedding with concepts[J].Knowledge-Based Systems,2019,164:38-44.
[17] WESTON J,BORDES A,YAKHNENKO O,et al.Connecting language and knowledge bases with embedding models for relation extraction[C]//Conference on Empirical Methods in Natural Language Processing.Washington,USA,2013:1366-1371.
[18] WANG H W,ZHANG F Z,WANG J,et al.Ripplenet:Propagating user preferences on the knowledge graph for recommender systems[C]//Proceedings of the 27th ACM International Conference on Information and Knowledge Management.Torino,Italy,2018:417-426.
[19] WANG H W,ZHAO M,XIE X,et al.Knowledge graph convolutional networks for recommender systems[C]//The World Wide Web Conference.San Francisco,USA,2019:3307-3313.
[20] JIA Z H,BIN C Z,GU T L,et al.Personalized attraction recommendation based on knowledge graph and long-term and short-term preferences of users[J].Journal of Intelligent Systems,2020,15(5):990-997.
[21] LIU Q,CHEN S P,HUO H.An entity recommendation model
based on user preference dissemination of knowledge graph[J].Application Research of Computers,2020,37(10):2926-2931.
[22] WANG X,HE X N,CAO Y X,et al.KGAT:Knowledge Graph Attention Network for Recommendation[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining(KDD ’19).New York:ACM,2019:950-958.
[23] ABU-EI-HAIJA S,PEROZZI B,KAPOOR A,et al.MixHop:Higher-Order Graph Convolutional Architectures via Sparsified Neighborhood Mixing[J].arXiv:1905.00067,2019.
[24] MIKOLOV T,SUTSKEVER I,CHEN K,et al.DistributedRepresentations of Words and Phrases and their Compositio-nality[J].arXiv:1310.4546,2013.
[25] YING R,HE R,CHEN K,et al.Graph Convolutional Neural Networks for Web-Scale Recommender Systems[J].arXiv:1806.01973,2018.
[26] TAICH Y,WU M R,CHU Y W,et al.GraphSW:a training protocol based on stage-wise training for GNN-based Recommender Model[J].arXiv:1908.05611,2019.
[27] TAI C Y,WU M R,CHU Y W,et al.MVIN:Learning Multi-view Items for Recommendation[C]//43rd International ACM SIGIR Conference on Research and Development in Information Retrieval.New York,2020:99-108.
[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] WU Zi-yi, LI Shao-mei, JIANG Meng-han, ZHANG Jian-peng. Ontology Alignment Method Based on Self-attention [J]. Computer Science, 2022, 49(9): 215-220.
[3] KONG Shi-ming, FENG Yong, ZHANG Jia-yun. Multi-level Inheritance Influence Calculation and Generalization Based on Knowledge Graph [J]. Computer Science, 2022, 49(9): 221-227.
[4] ZHOU Fang-quan, CHENG Wei-qing. Sequence Recommendation Based on Global Enhanced Graph Neural Network [J]. Computer Science, 2022, 49(9): 55-63.
[5] XU Yong-xin, ZHAO Jun-feng, WANG Ya-sha, XIE Bing, YANG Kai. Temporal Knowledge Graph Representation Learning [J]. Computer Science, 2022, 49(9): 162-171.
[6] 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.
[7] 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.
[8] 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.
[9] WANG Jie, LI Xiao-nan, LI Guan-yu. Adaptive Attention-based Knowledge Graph Completion [J]. Computer Science, 2022, 49(7): 204-211.
[10] 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.
[11] 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.
[12] YANG Bing-xin, GUO Yan-rong, HAO Shi-jie, Hong Ri-chang. Application of Graph Neural Network Based on Data Augmentation and Model Ensemble in Depression Recognition [J]. Computer Science, 2022, 49(7): 57-63.
[13] MA Rui-xin, LI Ze-yang, CHEN Zhi-kui, ZHAO Liang. Review of Reasoning on Knowledge Graph [J]. Computer Science, 2022, 49(6A): 74-85.
[14] DENG Kai, YANG Pin, LI Yi-zhou, YANG Xing, ZENG Fan-rui, ZHANG Zhen-yu. Fast and Transmissible Domain Knowledge Graph Construction Method [J]. Computer Science, 2022, 49(6A): 100-108.
[15] DU Xiao-ming, YUAN Qing-bo, YANG Fan, YAO Yi, JIANG Xiang. Construction of Named Entity Recognition Corpus in Field of Military Command and Control Support [J]. Computer Science, 2022, 49(6A): 133-139.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!