Computer Science ›› 2021, Vol. 48 ›› Issue (8): 72-79.doi: 10.11896/jsjkx.200800226

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

Recommendation Algorithm Based on Heterogeneous Information Network Embedding and Attention Neural Network

ZHAO Jin-long, ZHAO Zhong-ying   

  1. School of Computer Science and Engineering,Shandong University of Science and Technology,Qingdao,Shandong 266590,China
  • Received:2020-08-31 Revised:2020-11-06 Published:2021-08-10
  • About author:ZHAO Jin-long,born in 1995,postgra-duate.His main research interests include network representation learning and recommendation system.(zhaojinlongchn@foxmail.com)ZHAO Zhong-ying,born in 1983,asso-ciate professor,is a senior member of China Computer Federation.Her main research interests include social network analysis and data mining.
  • Supported by:
    National Natural Science Foundation of China(62072288, 61702306) and Natural Science Foundation of Shandong Province(ZR2018BF013).

Abstract: Recommendation system,as a very effective technique to solve the information overload,has received a great deal of attention from researchers.However,the real application of recommending systems can be modeled as heterogeneous networks with multi-typed nodes and relations.Thus,heterogeneous network embedding based recommendation becomes a very hot research topic in recent years.However,most of the existing studies do not fully explore the auxiliary information and complex relations which are valuable for enhancing recommending performance.To address the above problems,a recommendation algorithm based on heterogeneous information network embedding and attention neural network is proposed.First,this paper proposes a heterogeneous information network embedding method that maintains semantic relationship and topological structure simultaneously.Then,it designs a meta-path based random walk strategy to extract node sequences from heterogeneous information networks.All the sequences are filtered and then employed to learn the embeddings for each user and item in different meta-paths.At last,this paper presents a recommendation algorithm based on attention neural network with the above embeddings as input.The attention network composed of attention layers and hidden layers is able to explore the complex relationships and hence enhance the performance of recommendation.To verify the effectiveness of the proposed method,this paper conducts experiments on two kinds of real-world datasets and makes a comparison with three competitive algorithms.The results show that the proposed algorithm improves the recommending performance in terms of MAE and RMSE,with a maximum increase of 8.9%.

Key words: Attention neural network, Heterogeneous information networks, Meta-path, Recommendation algorithm, Representation learning

CLC Number: 

  • TP181
[1]SARWAR B,KARYPIS G,KONSTAN J,et al.Analysis of re-commendation algorithms for e-commerce[C]//Proceedings of the 2nd ACM Conference on Electronic Commerce.Association for Computing Machinery.2000:158-167.
[2]SCHAFER J B,KONSTAN J A,RIEDL J.E-commerce recommendation applications[J].Data Mining and Knowledge Disco-very,2001,5(1/2):115-153.
[3]YE M,YIN P,LEE W C.Location recommendation for location-based social networks[C]//Proceedings of the 18th SIGSPATIAL International Conference on Advances in Geographic Information Systems.Association for Computing Machinery,2010:458-461.
[4]SONG Y,ZHANG L,GILES C L.Automatic tag recommendation algorithms for social recommender systems[J].ACM Transactions on the Web,2011,5(1):1-31.
[5]DAVIDSON J,LIEBALD B,LIU J,et al.The YouTube video recommendation system[C]//Proceedings of the fourth ACM Conference on Recommender Systems.Association for Computing Machinery,2010:293-296.
[6]ZHOU C H,SHEN J J,LI Y,et al.Review of classical recommendation algorithms[J].Computer Science and Application,2019,9(9):1803-1817.
[7]PAZZANI M J,BILLSUS D.Content-based recommendationsystems[M].The Adaptive Web.Berlin:Springer,2007:325-341.
[8]LINDEN G,SMITH B,YORK J.Amazon.com recommenda-tions:Item-to-item collaborative filtering[J].IEEE Internet Computing,2003,7(1):76-80.
[9]CHOI K,YOO D,KIM G,et al.A hybrid online-product recommendation system:Combining implicit rating-based collaborative filtering and sequential pattern analysis[J].Electronic Commerce Research and Applications,2012,11(4):309-317.
[10]HAO R F,ZHANG G M,CHENG Y Q.Socialized Matrix Factorization Recommendation Algorithm with User RatingPrefe-rence Confidence[J].Journal of Chongqing University of Technology (Natural Science),2020,34(11):138-146.
[11]PAZZANI M J.A framework for collaborative,content-basedand demographic filtering[J].Artificial Intelligence Review,1999,13(5/6):393-408.
[12]LUO H,NIU C,SHEN R,et al.A collaborative filtering framework based on both local user similarity and global user similarity[J].Machine Learning,2008,72(3):231-245.
[13]SARWAR B,KARYPIS G,KONSTAN J,et al.Item-based collaborative filtering recommendation algorithms[C]//Procee-dings of the 10th international conference on World Wide Web.Association for Computing Machinery,2001:285-295.
[14]SHI C,LI Y,ZHANG J,et al.A survey of heterogeneous information network analysis[J].IEEE Transactions on Knowledge and Data Engineering,2016,29(1):17-37.
[15]SHI C,ZHOU C,KONG X,et al.Heterecom:a semantic-based recommendation system in heterogeneous networks[C]//Proceedings of the 18th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.Association for Computing Machinery,2012:1552-1555.
[16]JAMALI M,LAKSHMANAN L.HeteroMF:recommendation in heterogeneous information networks using context dependent factor models[C]//Proceedings of the 22nd International Conference on World Wide Web.Association for Computing Machinery,2013:643-654.
[17]YAN B,ZHANG L,GUO L,et al.Personalized Recommendation Based on Tag Semantics in the Heterogeneous Information Network[C]//China Conference on Wireless Sensor Networks.Springer,2019:224-235.
[18]LUO C,PANG W,WANG Z,et al.Hete-CF:Social-based collaborative filtering recommendation using heterogeneous relations[C]//2014 IEEE International Conference on Data Mi-ning.IEEE,2014:917-922.
[19]NIU Y Q,MENG Y Y,NIU Q F.Text Recommendation Based on Heterogeneous Attention Recurrent Neural Network[J].Computer Engineering,2020,46(10):52-59.
[20]CAO B,LIU N N,YANG Q.Transfer learning for collective link prediction in multiple heterogenous domains[C]//Procee-dings of the 27th International Conference on Machine Lear-ning.Omnipress,2010:159-166.
[21]JACOB Y,DENOYER L,GALLINARI P.Learning latent representations of nodes for classifying in heterogeneous social networks[C]//Proceedings of the 7th ACM International Confe-rence on Web Search and Data Mining.Association for Computing Machinery,2014:373-382.
[22]TANG J,QU M,MEI Q.Pte:Predictive text embeddingthrough large-scale heterogeneous text networks[C]//Procee-dings of the 21th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.Association for Computing Machinery,2015:1165-1174.
[23]TANG J,QU M,WANG M,et al.Line:Large-scale information network embedding[C]//Proceedings of the 24th International Conference on World Wide Web.International World Wide Web Conferences Steering Committee,2015:1067-1077.
[24]SUN Y,HAN J.Mining heterogeneous information networks:principles and methodologies[J].Synthesis Lectures on Data Mining and Knowledge Discovery,2012,3(2):1-159.
[25]SUN Y,HAN J,YAN X,et al.Pathsim:Meta path-based top-k similarity search in heterogeneous information networks[J].Proceedings of the VLDB Endowment,2011,4(11):992-1003.
[26]CHEN Y,WANG C.HINE:Heterogeneous information net-work embedding[C]//Proceedings of the International Confe-rence on Database Systems for Advanced Applications.Sprin-ger,2017:180-195.
[27]DONG Y,CHAWLA N V,SWAMI A.metapath2vec:Scalable representation learning for heterogeneous networks[C]//Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.Association for Computing Machinery,2017:135-144.
[28]PEROZZI B,AL-RFOU R,SKIENA S.Deepwalk:Online lear-ning of social representations[C]//Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Disco-very and Data Mining.Association for Computing Machinery,2014:701-710.
[29]GUTHRIE D,ALLISON B,LIU W,et al.A closer look at skip-gram modelling[C]//Proceedings of the 5th International Conference on Language Resources and Evaluation,2006:1222-1225.
[30]CHANG S,HAN W,TANG J,et al.Heterogeneous networkembedding via deep architectures[C]//Proceedings of the 21th ACM SIGKDD International Conference on Knowledge Disco-very and Data Mining.Association for Computing Machine-ry,2015:119-128.
[31]FU T,LEE W C,LEI Z.Hin2vec:Explore meta-paths in heterogeneous information networks for representation learning[C]//Proceedings of the 2017 ACM on Conference on Information and Knowledge Management.Association for Computing Machine-ry,2017:1797-1806.
[32]YU X,REN X,SUN Y,et al.Recommendation in heterogeneous information networks with implicit user feedback[C]//Procee-dings of the 7th ACM Conference on Recommender Systems.Association for Computing Machinery,2013:347-350.
[33]HU B,SHI C,ZHAO W X,et al.Local and global information fusion for top-n recommendation in heterogeneous information network[C]//Proceedings of the 27th ACM International Conference on Information and Knowledge Management.Association for Computing Machinery,2018:1683-1686.
[34]SHI C,ZHANG Z,JI Y,et al.SemRec:a personalized semantic recommendation method based on weighted heterogeneous information networks[J].World Wide Web,2019,22(1):153-184.
[35]ZHAO H,YAO Q,LI J,et al.Meta-graph based recommendation fusion over heterogeneous information networks[C]//Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.Association for Computing Machinery,2017:635-644.
[36]SHI C,HU B,ZHAO W X,et al.Heterogeneous information network embedding for recommendation[J].IEEE Transactions on Knowledge and Data Engineering,2018,31(2):357-370.
[37]GROVER A,LESKOVEC J.Node2vec:Scalable feature learning for networks[C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mi-ning.Association for Computing Machinery,2016:855-864.
[38]BOTTOU L.Large-scale machine learning with stochastic gra-dient descent[C]//Proceedings of COMPSTAT'2010.2010:177-186.
[39]WILLMOTT C J,MATSUURA K.Advantages of the mean absolute error (MAE) over the root mean square error (RMSE) in assessing average model performance[J].Climate Research,2005,30(1):79-82.
[40]MNIH A,SALAKHUTDINOV R R.Probabilistic matrix fac-torization[C]//Proceedings of the Twenty-First Annual Confe-rence on Neural Information Processing Systems.Curran Asso-ciates,2007:1257-1264.
[41]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.
[42]HE X,LIAO L,ZHANG H,et al.Neural collaborative filtering[C]//Proceedings of the 26th International Conference on World Wide Web.International World Wide Web Conferences Steering Committee,2017:173-182.
[1] SONG Jie, LIANG Mei-yu, XUE Zhe, DU Jun-ping, KOU Fei-fei. Scientific Paper Heterogeneous Graph Node Representation Learning Method Based onUnsupervised Clustering Level [J]. Computer Science, 2022, 49(9): 64-69.
[2] HUANG Li, ZHU Yan, LI Chun-ping. Author’s Academic Behavior Prediction Based on Heterogeneous Network Representation Learning [J]. Computer Science, 2022, 49(9): 76-82.
[3] LYU Xiao-feng, ZHAO Shu-liang, GAO Heng-da, WU Yong-liang, ZHANG Bao-qi. Short Texts Feautre Enrichment Method Based on Heterogeneous Information Network [J]. Computer Science, 2022, 49(9): 92-100.
[4] 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.
[5] LI Zong-min, ZHANG Yu-peng, LIU Yu-jie, LI Hua. Deformable Graph Convolutional Networks Based Point Cloud Representation Learning [J]. Computer Science, 2022, 49(8): 273-278.
[6] HUANG Pu, DU Xu-ran, SHEN Yang-yang, YANG Zhang-jing. Face Recognition Based on Locality Regularized Double Linear Reconstruction Representation [J]. Computer Science, 2022, 49(6A): 407-411.
[7] PU Qian-qian, LEI Hang, LI Zhen-hao, LI Xiao-yu. Personalized News Recommendation Algorithm with Enhanced List Information and User Interests [J]. Computer Science, 2022, 49(6): 142-148.
[8] WANG Mei-ling, LIU Xiao-nan, YIN Mei-juan, QIAO Meng, JING Li-na. Deep Learning Recommendation Algorithm Based on Reviews and Item Descriptions [J]. Computer Science, 2022, 49(3): 99-104.
[9] JIANG Zong-li, FAN Ke, ZHANG Jin-li. Generative Adversarial Network and Meta-path Based Heterogeneous Network Representation Learning [J]. Computer Science, 2022, 49(1): 133-139.
[10] DONG Xiao-mei, WANG Rui, ZOU Xin-kai. Survey on Privacy Protection Solutions for Recommended Applications [J]. Computer Science, 2021, 48(9): 21-35.
[11] WANG Ying-li, JIANG Cong-cong, FENG Xiao-nian, QIAN Tie-yun. Time Aware Point-of-interest Recommendation [J]. Computer Science, 2021, 48(9): 43-49.
[12] YE Hong-liang, ZHU Wan-ning, HONG Lei. Music Style Transfer Method with Human Voice Based on CQT and Mel-spectrum [J]. Computer Science, 2021, 48(6A): 326-330.
[13] YANG Ru-han, DAI Yi-ru, WANG Jian, DONG Jin. Humans-Cyber-Physical Ontology Fusion of Industry Based on Representation Learning [J]. Computer Science, 2021, 48(5): 190-196.
[14] XIONG Xu-dong, DU Sheng-dong, XIA Wan-jun, LI Tian-rui. Recommendation Algorithm Based on Bipartite Graph Convolution Representation [J]. Computer Science, 2021, 48(4): 78-84.
[15] QIAN Sheng-sheng, ZHANG Tian-zhu, XU Chang-sheng. Survey of Multimedia Social Events Analysis [J]. Computer Science, 2021, 48(3): 97-112.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!