UniTraj: Learning a Universal Trajectory Foundation Model from Billion-Scale Worldwide Traces

Yuanshao Zhu, James Jianqiao Yu, Xiangyu Zhao, Xun Zhou, Liang Han +2 more
11/6/2024
cs.ETcs.AIcs.LG

No official code found. Best implementation: Yasoz/UniTraj (36 stars, Apache-2.0, last commit 2025-12-11).

Abstract

Building a universal trajectory foundation model is a promising solution to address the limitations of existing trajectory modeling approaches, such as task specificity, regional dependency, and data sensitivity. Despite its potential, data preparation, pre-training strategy development, and architectural design present significant challenges in constructing this model. Therefore, we introduce UniTraj, a Universal Trajectory foundation model that aims to address these limitations through three key innovations. First, we construct WorldTrace, an unprecedented dataset of 2.45 million trajectories with billions of GPS points spanning 70 countries, providing the diverse geographic coverage essential for region-independent modeling. Second, we develop novel pre-training strategies--Adaptive Trajectory Resampling and Self-supervised Trajectory Masking--that enable robust learning from heterogeneous trajectory data with varying sampling rates and quality. Finally, we tailor a flexible model architecture to accommodate a variety of trajectory tasks, effectively capturing complex movement patterns to support broad applicability. Extensive experiments across multiple tasks and real-world datasets demonstrate that UniTraj consistently outperforms existing methods, exhibiting superior scalability, adaptability, and generalization, with WorldTrace serving as an ideal yet non-exclusive training resource.

View on arXivView PDF

Code Implementations(1)

Yasoz/UniTrajCommunity72%
Apache-2.0

Implementation for NeurIPS 2025 paper: UniTraj: Learning a Universal Trajectory Foundation Model from Billion-Scale Worldwide Traces

365Python, Jupyter NotebookSep 21, 202510 months agoApache-2.0

Similar Papers

Cite this paper

@article{zhu2024unitraj,
  title  = {UniTraj: Learning a Universal Trajectory Foundation Model from Billion-Scale Worldwide Traces},
  author = {Yuanshao Zhu and James Jianqiao Yu and Xiangyu Zhao and Xun Zhou and Liang Han and Xuetao Wei and Yuxuan Liang},
  year   = {2024},
  eprint = {2411.03859},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2411.03859v3},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2411.03859.svg)](https://codeofpaper.com/paper/2411.03859)

Discussion