Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

ST-MoE-BERT: A Spatial-Temporal Mixture-of-Experts Framework for Long-Term Cross-City Mobility Prediction

About

Predicting human mobility across multiple cities presents significant challenges due to the complex and diverse spatial-temporal dynamics inherent in different urban environments. In this study, we propose a robust approach to predict human mobility patterns called ST-MoE-BERT. Compared to existing methods, our approach frames the prediction task as a spatial-temporal classification problem. Our methodology integrates the Mixture-of-Experts architecture with BERT model to capture complex mobility dynamics and perform the downstream human mobility prediction task. Additionally, transfer learning is integrated to solve the challenge of data scarcity in cross-city prediction. We demonstrate the effectiveness of the proposed model on GEO-BLEU and DTW, comparing it to several state-of-the-art methods. Notably, ST-MoE-BERT achieves an average improvement of 8.29%.

Haoyu He, Haozheng Luo, Qi R. Wang• 2024

Related benchmarks

TaskDatasetResultRank
Mobility PredictionSapporo
DTW3.80e+3
13
Mobility PredictionHiroshima
DTW4.89e+3
13
Human mobility predictionKumamoto
Acc@128.62
13
Human mobility predictionSapporo
Acc@128.69
13
Human mobility predictionHiroshima
Acc@128.39
13
Mobility PredictionKumamoto
DTW4.69e+3
13
Human mobility predictionCity B (test)
GEO-BLEU0.297
4
Human mobility predictionCity C (test)
GEO-BLEU0.297
4
Human mobility predictionCity D (test)
GEO-BLEU0.3
4
Human mobility predictionCity A (test)
GEO-BLEU0.286
3
Showing 10 of 10 rows

Other info

Code

Follow for update