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

Yuwen LIU, Aixiang PEI, Fan WANG, Yihong YANG, Xuyun ZHANG, Hao WANG, Hongning DAI, Lianyong QI*, Rui MA

*Corresponding author for this work

Research output: Journal PublicationsJournal Article (refereed)peer-review

5 Citations (Scopus)

Abstract

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.

Original languageEnglish
Pages (from-to)3174-3189
Number of pages16
JournalInternational Journal of Intelligent Systems
Volume36
Issue number7
Early online date25 Mar 2021
DOIs
Publication statusPublished - Jul 2021
Externally publishedYes

Bibliographical note

Funding Information:
This study was supported by the National Natural Science Foundation of China (No. 61872219), the Natural Science Foundation of Shandong Province (ZR2019MF001), the Open Project of State Key Laboratory for Novel Software Technology (No. KFKT2020B08), the Macao Science and Technology Development Fund under Macao Funding Scheme for Key R&D Projects (0025/2019/AKP) and the Fundamental Research Funds for the Central Universities under Grant (No. 30919011282).

Publisher Copyright:
© 2021 Wiley Periodicals LLC

Keywords

  • attention
  • category-aware
  • embedding
  • gated recurrent unit
  • next POI recommendation

Fingerprint

Dive into the research topics of 'An attention-based category-aware GRU model for the next POI recommendation'. Together they form a unique fingerprint.

Cite this