arXiv:2410.14970cs.SIcs.AI2024-10NeurIPS被引 20

解决用户出行预测中的冷门地点难预测问题

Taming the Long Tail in Human Mobility Prediction

  • 构建图结构并动态调整长尾节点权重
  • 在真实数据集上提升冷门地点预测准确率
  • 适合做出行推荐与城市规划的研究者

随着基于位置服务的普及,人类出行预测在个性化导航、推荐系统优化和城市交通规划中扮演关键角色。该任务旨在根据用户历史访问记录预测其下一个将访问的地点(POI)。然而,由于时空分布不均导致的长尾问题——即多数地点访问频率极低——使得现有AI模型难以准确预测这些冷门地点。为此,本文提出长尾调整型下一地点预测框架LoTNext,包含长尾图调整模块以降低稀疏节点的影响,以及基于对数几率和样本权重调整的新型损失函数。同时引入辅助预测任务以增强模型泛化能力与精度。在两个真实轨迹数据集上的实验表明,LoTNext显著优于现有最先进方法。

原文摘要 · Abstract (English)

With the popularity of location-based services, human mobility prediction plays a key role in enhancing personalized navigation, optimizing recommendation systems, and facilitating urban mobility and planning. This involves predicting a user's next POI (point-of-interest) visit using their past visit history. However, the uneven distribution of visitations over time and space, namely the long-tail problem in spatial distribution, makes it difficult for AI models to predict those POIs that are less visited by humans. In light of this issue, we propose the Long-Tail Adjusted Next POI Prediction (LoTNext) framework for mobility prediction, combining a Long-Tailed Graph Adjustment module to reduce the impact of the long-tailed nodes in the user-POI interaction graph and a novel Long-Tailed Loss Adjustment module to adjust loss by logit score and sample weight adjustment strategy. Also, we employ the auxiliary prediction task to enhance generalization and accuracy. Our experiments with two real-world trajectory datasets demonstrate that LoTNext significantly surpasses existing state-of-the-art works.

出行预测长尾问题图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。