HST-LSTM: A Hierarchical Spatial-Temporal Long-Short Term Memory Network for Location Prediction

Dejiang Kong, Fei Wu

Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence

Main track. Pages 2341-2347.

https://doi.org/10.24963/ijcai.2018/324

PDF BibTeX

Abstract

The widely use of positioning technology has made mining the movements of people feasible and plenty of trajectory data have been accumulated. How to efficiently leverage these data for location prediction has become an increasingly popular research topic as it is fundamental to location-based services (LBS). The existing methods often focus either on long time (days or months) visit prediction (i.e., the recommendation of point of interest) or on real time location prediction (i.e., trajectory prediction). In this paper, we are interested in the location prediction problem in a weak real time condition and aim to predict users' movement in next minutes or hours. We propose a Spatial-Temporal Long-Short Term Memory (ST-LSTM) model which naturally combines spatial-temporal influence into LSTM to mitigate the problem of data sparsity. Further, we employ a hierarchical extension of the proposed ST-LSTM (HST-LSTM) in an encoder-decoder manner which models the contextual historic visit information in order to boost the prediction performance. The proposed HST-LSTM is evaluated on a real world trajectory data set and the experimental results demonstrate the effectiveness of the proposed model.

Keywords

  • Machine Learning: Machine Learning
  • Machine Learning: Neural Networks
  • Machine Learning: Learning Preferences or Rankings
  • Machine Learning: Deep Learning
  • Machine Learning: Recommender Systems
Welcome Future Conferences AI Journal Division IJCAI Proceedings Travel Grant Program IJCAI Awards