统一模型预测多城市人流,准确率提升超10.2%
UniMove: A Unified Model for Multi-city Human Mobility Prediction
- 双塔架构分离位置与轨迹建模,实现跨城市通用编码
- 引入MoE Transformer自适应处理不同城市移动模式
- 支持多城联合训练,适合城市规划与交通优化研究
人类出行预测对城市规划、交通优化和个性化服务至关重要。然而,出行行为具有内在随机性、非均匀时间间隔及复杂模式,加之城市结构、基础设施和人口密度差异带来的异质性,给建模带来巨大挑战。现有方法通常需为每个城市单独训练模型,因空间表示和地理覆盖范围各异。本文提出UniMove,一种统一的多城市出行预测模型,解决两大难题:(1) 构建通用空间表征以实现跨城市令牌共享;(2) 建模由城市特征差异引起的异构出行模式。设计轨迹-位置双塔架构,位置塔用于通用空间编码,轨迹塔用于序列出行建模。引入MoE Transformer块,动态选择专家以应对多样化移动模式。在多个来自不同城市的数据集上进行大量实验表明,UniMove真正实现了统一模型的本质。通过多城市数据联合训练并相互增强,预测准确率显著提升超过10.2%。UniMove代表了迈向真正统一架构的人类出行基础模型的关键进展。代码已开源:https://github.com/tsinghua-fib-lab/UniMove/
原文摘要 · Abstract (English)
Human mobility prediction is vital for urban planning, transportation optimization, and personalized services. However, the inherent randomness, non-uniform time intervals, and complex patterns of human mobility, compounded by the heterogeneity introduced by varying city structures, infrastructure, and population densities, present significant challenges in modeling. Existing solutions often require training separate models for each city due to distinct spatial representations and geographic coverage. In this paper, we propose UniMove, a unified model for multi-city human mobility prediction, addressing two challenges: (1) constructing universal spatial representations for effective token sharing across cities, and (2) modeling heterogeneous mobility patterns from varying city characteristics. We propose a trajectory-location dual-tower architecture, with a location tower for universal spatial encoding and a trajectory tower for sequential mobility modeling. We also design MoE Transformer blocks to adaptively select experts to handle diverse movement patterns. Extensive experiments across multiple datasets from diverse cities demonstrate that UniMove truly embodies the essence of a unified model. By enabling joint training on multi-city data with mutual data enhancement, it significantly improves mobility prediction accuracy by over 10.2\%. UniMove represents a key advancement toward realizing a true foundational model with a unified architecture for human mobility. We release the implementation at https://github.com/tsinghua-fib-lab/UniMove/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。