AdaMove: Efficient Test-Time Adaptation for Human Mobility Prediction

Human mobility prediction is a fundamental technique for many urban applications, e.g., location-based recommendation, traffic scheduling, and travel demand prediction. Over the past decades, many methods, e.g., Markov Model, RNN, Transformer, have been leveraged to tackle the problem. However, existing approaches mainly train a supervised model based on an offline training dataset, which overlooks the phenomenon that the mobility behaviors of humans vary across time, and the trained models may not achieve ideal performance when applied to the testing data. To tackle this challenge, in this paper, we propose AdaMove, an efficient Test-Time Adaptive (TTA) model for human mobility prediction. AdaMove has a Preference-aware Test-Time Adaptation module called PTTA, which can adjust the parameters of a trained model based on the input test trajectory such that the model can generalize to the test distribution. In addition, to address the issue of reduced inference efficiency caused by parameter adjustment during the testing phase, AdaMove is equipped with a Lightweight human Mobility prediction model called LightMob, which only requires the recent trajectory as input to accelerate the inference. It is enhanced by historical trajectory knowledge via contrastive learning during the training time, so it has competitive performance compared with existing models. Extensive experiments on three real-world human mobility datasets demonstrate that AdaMove outperforms the best baseline by 9.3% on average in accuracy, and accelerates the inference speed by 28.5% on average compared with the original TTA - based inference.