Journal of Jilin University(Engineering and Technology Edition) ›› 2021, Vol. 51 ›› Issue (1): 278-284.doi: 10.13229/j.cnki.jdxbgxb20200068

Previous Articles    

Recommending activity to users via deep graph neural network

Xiao-hui WEI1,2(),Bing-yi SUN1,2,Jia-xu CUI1,2   

  1. 1.College of Computer Science and Technology,Jilin University,Changchun 130012,China
    2.Key Laboratory of Symbol Computation and Knowledge Engineering of Ministry of Education,Jilin University,Changchun 130012,China
  • Received:2020-02-10 Online:2021-01-01 Published:2021-01-20

Abstract:

In event-based social network, users and activities construct heterogeneous network naturally. We propose an end-to-end recommendation model using high-order connectivity in heterogeneous network and non-linear matching relations to increase hit ratio. First, we extract the effect of high-order connectivity of each node, and get the embedding of nodes by graph neural network. Then, the embedding of user-activity pair is input of the multilayer perceptron to gain the current matching probability. Finally, by repeating this process the final matching probability is obtained. Experiments show that, the recommendation method is stable. Comparing with existing algorithms, our algorithm increases hit ratio more than 10%, and normalized discounted cumulative gain about 10%. Comparing with using multilayer perceptron only, both hit ratio and normalized discounted cumulative gain rise when considering high-order connectivity information.

Key words: computer software and theory, recommendation system, event-based social network, heterogeneous graph, graph neural network

CLC Number: 

  • TP301.6

Fig.1

Comparison between the first-order connectivity and the higher-order connectivity"

Fig.2

Process of DGNN-AR"

Table 1

Statistics of train datasets"

城市交互数用户数活动数稀疏度
洛杉矶112 0737 58810 2730.001 54
休斯敦10 4697741 4650.009 91
纽约443 19929 05931 0030.000 50

Table 2

Parameters of DGNN-AR"

参数名参数值参数名参数值
迭代次数35批尺寸512
邻居层数3学习率0.001
嵌入式维度64惩罚率0.0001
MLP维度[128,64,32]

Fig.3

Stability of DGNN-AR"

Table 3

Performance comparison among different algorithms"

对比算法洛杉矶休斯敦纽约

HR

@10

NDCG@10

HR

@10

NDCG@10

HR

@10

NDCG@10
BPR[8]0.10760.04780.10100.05020.000530.00017
HereRS[4]0.60420.51210.45920.42270.52250.4884
GC?MC[13]0.37350.21130.60290.35940.25830.1103
NGCN[14]0.60540.41110.70770.43640.30260.2009
DGNN?AR0.82490.60620.73790.44420.76510.5472

Table 4

Effect of neighbor layers"

情况洛杉矶休斯敦纽约

HR

@10

NDCG@10

HR

@10

NDCG@10

HR

@10

NDCG@10
DGNN?AR?00.74570.50830.50460.24710.72420.4936
DGNN?AR?10.80080.56030.63700.36480.74520.5257
DGNN?AR?20.81840.58590.69990.41760.75790.5284
DGNN?AR?30.82490.60620.73790.44420.76510.5472
DGNN?AR?40.80290.56160.73650.42830.75360.5291
1 Liu X, He Q, Tian Y, et al. Event-based social networks: linking the online and offline social worlds[C]∥Proceedings of the 18th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, Beijing, China, 2012: 1032-1040.
2 Qiao Z, Zhang P, Cao Y, et al. Combining heterogenous social and geographical information for event recommendation[C]∥Proceedings of the Twenty-Eighth AAAI Conference on Artificial Intelligence, Quebec, Canada, 2014: 145-151.
3 Macedo A Q, Marinho L B, Santos R L T. Context-aware event recommendation in event-based social networks[C]∥Proceedings of the 9th ACM Conference on Recommender Systems, Vienna, Austria, 2015: 123-130.
4 Pham T, Li X, Cong G, et al. A general graph-based model for recommendation in event-based social networks[C]∥IEEE 31st International Conference on Data Engineering, Seoul, South Korea, 2015: 567-578.
5 Wu X, Dong Y, Shi B, et al. Who will attend this event together? event attendance prediction via deep LSTM networks[C]∥ Proceedings of the 2018 SIAM International Conference on Data Mining, San Diego, CA, USA, 2018: 180-188.
6 董立岩, 王越群, 贺嘉楠, 等. 基于时间衰减的协同过滤推荐算法[J]. 吉林大学学报:工学版, 2017, 47(4): 1268-1272.
Dong Li-yan, Wang Yue-qun, He Jia-nan, et al. Collaborative filtering recommendation algorithm based on time decay[J]. Journal of Jilin University (Engineering and Technology Edition), 2017, 47(4): 1268-1272.
7 Hamilton W L, Ying R, Leskovec J. Representation learning on graphs: methods and applications[J]. arXiv:1709.05584, 2017: 1-24.
8 Rendle S, Freudenthaler C, Gantner Z, et al. BPR: bayesian personalized ranking from implicit feedback[C]∥ Proceedings of the twenty-fifth Conference on Uncertainty in Artificial Intelligence, Montreal, QC, Canada, 2009: 452-461.
9 Wang H, Terrovitis M, Mamoulis N. Location recommendation in location-based social networks using user check-in data[C]∥Proceedings of the 21st ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems, Orlando, Florida, USA, 2013: 364-373.
10 Jiang X, Sun X, Zhuge H. Towards an effective and unbiased ranking of scientific literature through mutual reinforcement[C]∥Proceedings of the 21st ACM International Conference on Information and Knowledge Management, Maui, HI, USA, 2012: 714-723.
11 Thomas N K, Max W. Semi-Supervised classification with graph convolutional networks[C]∥The 5th International Conference on Learning Representations, Toulon, France, 2017.
12 Hamilton W, Ying R, Leskovec J. Inductive representation learning on large graphs[C]∥Advances in Neural Information Processing Systems, Long Beach, CA, USA, 2017: 1024-1034.
13 Berg R V D, Kipf T, Welling M. Graph convolutional matrix completion[C]∥Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, London, UK, 2018.
14 Wang X, He X, Wang M, et al. Neural graph collaborative filtering[C]∥Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval, Paris, France, 2019: 165-174.
15 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.
16 Kingma D, Ba J. Adam: a method for stochastic optimization[C]∥The 3rd International Conference for Learning Representations, San Diego, USA, 2015.
17 Bayer I, He X, Kanagal B, et al. A generic coordinate descent framework for learning from implicit feedback[C]∥Proceedings of the 26th International Conference on World Wide Web, Perth, Australia, 2017: 1341-1350.
[1] LI Jun, LI Xiong-fei, DONG Yuan-fang, ZHAO Hai-ying. New performance evaluation method for classifier [J]. 吉林大学学报(工学版), 2012, 42(02): 463-468.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!