中文

GeoFormer:使用生成式预训练 Transformer(GPT)预测人类移动性

机器学习 2023-11-10 v1 计算机与社会

摘要

预测人类移动性具有重要的实用价值,其应用从改进灾害风险规划到模拟传染病传播不等。在本文中,我们提出 GeoFormer,一种改编自 GPT 架构的仅解码器 Transformer 模型,用于预测人类移动性。我们所提模型在 HuMob Challenge 2023 的背景下接受了严格测试——该竞赛旨在评估预测模型在标准化数据集上预测人类移动性的性能。该挑战使用了两个数据集,涵盖在 75 天纵向周期内 25,000 与 100,000 个体的城市尺度数据。GeoFormer 作为竞赛中的顶尖表现者脱颖而出,跻身前三名。其成功体现在竞赛所选两项性能指标——GEO-BLEU 与动态时间规整(DTW)度量——上均表现良好。GeoFormer 在 HuMob Challenge 2023 上的性能凸显了其在人类移动性预测领域作出实质贡献的潜力,对灾害准备、疫情控制等具有深远影响。

关键词

引用

@article{arxiv.2311.05092,
  title  = {GeoFormer: Predicting Human Mobility using Generative Pre-trained Transformer (GPT)},
  author = {Aivin V. Solatorio},
  journal= {arXiv preprint arXiv:2311.05092},
  year   = {2023}
}

备注

Accepted in the HuMob-Challenge '23: 1st International Workshop on the Human Mobility Prediction Challenge. Repository is available at https://github.com/avsolatorio/GeoFormer