Temporal-geographical attention-based transformer for point-of-interest recommendation

Abstract

Point-of-Interest (POI) recommendation is one of the most important tasks in the field of social network analysis. Many efforts have been proposed to enhance the model performance for the POI recommendation task in recent years. Existing studies have revealed that the temporal factor and geographical factor are two crucial contextual factors which influence user decisions. However, they only learn representations of POIs and users from the single contextual factor and fuse the learned representations in the final stage, which ignores the interactions of different contextual factors, leading to learning suboptimal representations of POIs and users. To overcome this gap, we propose a novel Temporal-Geographical Attention-based Transformer (TGAT) for the POI recommendation task. Specifically, TGAT develops a hybrid sequence sampling strategy that samples the sequence of POIs from the different contextual factor POI graphs generated by the users’ check-in records. In this way, the interactions of different contextual factors can be care-fully pre-served. Then TGAT conducts a Transformer-based neural network backbone to learn representations of POIs from the sampling sequences. In addition, a weighted aggregation strategy is proposed to fuse the representations learned from different context factors. The extensive experimental results on real-world datasets have demonstrated the effectiveness of TGAT.