Spatio-Temporal Check-in Time Prediction with Recurrent Neural Network based Survival Analysis

Spatio-Temporal Check-in Time Prediction with Recurrent Neural Network based Survival Analysis

Guolei Yang, Ying Cai, Chandan K Reddy

Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence
Main track. Pages 2976-2983. https://doi.org/10.24963/ijcai.2018/413

We introduce a novel check-in time prediction problem. The goal is to predict the time a user will check-in to a given location. We formulate check-in prediction as a survival analysis problem and propose a Recurrent-Censored Regression (RCR) model. We address the key challenge of check-in data scarcity, which is due to the uneven distribution of check-ins among users/locations. Our idea is to enrich the check-in data with potential visitors, i.e., users who have not visited the location before but are likely to do so. RCR uses recurrent neural network to learn latent representations from historical check-ins of both actual and potential visitors, which is then incorporated with censored regression to make predictions. Experiments show RCR outperforms state-of-the-art event time prediction techniques on real-world datasets.
Keywords:
Machine Learning: New Problems
Machine Learning: Deep Learning
Machine Learning Applications: Applications of Supervised Learning