ST-MoE-BERT: A Spatial-Temporal Mixture-of-Experts Framework for Long-Term Cross-City Mobility Prediction
This work addresses cross-city mobility prediction, which is important for urban planning and transportation, but it appears incremental as it combines existing techniques like MoE and BERT for a specific domain.
The paper tackles the problem of predicting human mobility across multiple cities by proposing ST-MoE-BERT, a framework that integrates Mixture-of-Experts with BERT for spatial-temporal classification, achieving an average improvement of 8.29% over state-of-the-art methods on metrics like GEO-BLEU and DTW.
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%.