An attention-based category-aware GRU model for the next POI recommendation

被引:128
作者
Liu, Yuwen [1 ]
Pei, Aixiang [2 ]
Wang, Fan [1 ]
Yang, Yihong [1 ]
Zhang, Xuyun [3 ]
Wang, Hao [4 ]
Dai, Hongning [5 ]
Qi, Lianyong [1 ]
Ma, Rui [6 ]
机构
[1] Qufu Normal Univ, Sch Comp Sci, Rizhao 276800, Peoples R China
[2] Weifang Univ Sci & Technol, Weifang Key Lab Blockchain Agr Vegetables, Shouguang, Peoples R China
[3] Macquarie Univ, Dept Comp, Sydney, NSW, Australia
[4] Norwegian Univ Sci & Technol, Dept Comp Sci, Gjovik, Norway
[5] Macau Univ Sci & Technol, Fac Informat Technol, Macau, Peoples R China
[6] Shandong First Med Univ, Shandong Acad Med Sci, Gen Educ Dept, Tai An, Shandong, Peoples R China
基金
中国国家自然科学基金;
关键词
attention; category‐ aware; embedding; gated recurrent unit; next POI recommendation;
D O I
10.1002/int.22412
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
With the continuous accumulation of users' check-in data, we can gradually capture users' behavior patterns and mine users' preferences. Based on this, the next point-of-interest (POI) recommendation has attracted considerable attention. Its main purpose is to simulate users' behavior habits of check-in behavior. Then, different types of context information are used to construct a personalized recommendation model. However, the users' check-in data are extremely sparse, which leads to low performance in personalized model training using recurrent neural network. Therefore, we propose a category-aware gated recurrent unit (GRU) model to mitigate the negative impact of sparse check-in data, capture long-range dependence between user check-ins and get better recommendation results of POI category. We combine the spatiotemporal information of check-in data and take the POI category as users' preference to train the model. Also, we develop an attention-based category-aware GRU (ATCA-GRU) model for the next POI category recommendation. The ATCA-GRU model can selectively utilize the attention mechanism to pay attention to the relevant historical check-in trajectories in the check-in sequence. We evaluate ATCA-GRU using a real-world data set, named Foursquare. The experimental results indicate that our ATCA-GRU model outperforms the existing similar methods for next POI recommendation.
引用
收藏
页码:3174 / 3189
页数:16
相关论文
共 34 条
[1]  
[Anonymous], 2015, PROC NEURIPS, P577, DOI DOI 10.1016/0167-739X(94)90007-8
[2]   A Differential-Private Framework for Urban Traffic Flows Estimation via Taxi Companies [J].
Cai, Zhipeng ;
Zheng, Xu ;
Yu, Jiguo .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2019, 15 (12) :6492-6499
[3]   Trust-aware generative adversarial network with recurrent neural network for recommender systems [J].
Chen, Honglong ;
Wang, Shuai ;
Jiang, Nan ;
Li, Zhe ;
Yan, Na ;
Shi, Leyi .
INTERNATIONAL JOURNAL OF INTELLIGENT SYSTEMS, 2021, 36 (02) :778-795
[4]   Sequential Recommendation with User Memory Networks [J].
Chen, Xu ;
Xu, Hongteng ;
Zhang, Yongfeng ;
Tang, Jiaxi ;
Cao, Yixin ;
Qin, Zheng ;
Zha, Hongyuan .
WSDM'18: PROCEEDINGS OF THE ELEVENTH ACM INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING, 2018, :108-116
[5]   Mutual-Preference Driven Truthful Auction Mechanism in Mobile Crowdsensing [J].
Duan, Zhuojun ;
Li, Wei ;
Zheng, Xu ;
Cai, Zhipeng .
2019 39TH IEEE INTERNATIONAL CONFERENCE ON DISTRIBUTED COMPUTING SYSTEMS (ICDCS 2019), 2019, :1233-1242
[6]  
Firat Orhan, 2016, P 2016 C N AM CHAPT, P866
[7]  
He J, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1837
[8]   An Introductory Survey on Attention Mechanisms in NLP Problems [J].
Hu, Dichao .
INTELLIGENT SYSTEMS AND APPLICATIONS, VOL 2, 2020, 1038 :432-448
[9]   Location based services: ongoing evolution and research agenda [J].
Huang, Haosheng ;
Gartner, Georg ;
Krisp, Jukka M. ;
Raubal, Martin ;
Van de Weghe, Nico .
JOURNAL OF LOCATION BASED SERVICES, 2018, 12 (02) :63-93
[10]   An Attention-Based Spatiotemporal LSTM Network for Next POI Recommendation [J].
Huang, Liwei ;
Ma, Yutao ;
Wang, Shibo ;
Liu, Yanbo .
IEEE TRANSACTIONS ON SERVICES COMPUTING, 2021, 14 (06) :1585-1597