3 个月前

ST-MoE-BERT:一种用于长期跨城市出行预测的时空混合专家框架

ST-MoE-BERT:一种用于长期跨城市出行预测的时空混合专家框架

摘要

跨多个城市的人员流动预测面临重大挑战,这主要源于不同城市环境所固有的复杂且多样的时空动态特性。在本研究中,我们提出了一种稳健的人员流动模式预测方法,称为ST-MoE-BERT。相较于现有方法,本方法将预测任务建模为一个时空分类问题。所提出的方法结合了Mixture-of-Experts(专家混合)架构与BERT模型,以捕捉复杂的人员流动动态,并完成下游的人员流动预测任务。此外,通过引入迁移学习,有效缓解了跨城市预测中数据稀缺的问题。我们在GEO-BLEU和DTW两个评估指标上验证了所提模型的有效性,并与多种前沿方法进行了对比。结果表明,ST-MoE-BERT在平均性能上实现了8.29%的显著提升。

代码仓库

he-h/HuMob
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
trajectory-prediction-on-yjmob100k-bST-MoE-BERT
Accuracy: 28.7
BLEU: 29.7
DTW R@1: 29.3
trajectory-prediction-on-yjmob100k-cST-MoE-BERT
Accuracy: 28.9
BLEU: 29.7
DTW R@1: 19.7
trajectory-prediction-on-yjmob100k-dST-MoE-BERT
Accuracy: 26.5
BLEU: 30
DTW R@1: 48.1

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
ST-MoE-BERT:一种用于长期跨城市出行预测的时空混合专家框架 | 论文 | HyperAI超神经