UniTraj: Learning a Universal Trajectory Foundation Model from Billion-Scale Worldwide Traces
Yuanshao Zhu, James Jianqiao Yu, Xiangyu Zhao, Xun Zhou, Liang Han, Xuetao Wei, Yuxuan Liang
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. The implementation codes and full dataset are available at https://github.com/Yasoz/UniTraj.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 46e6cba4-212b-430a-8ec8-6c45ada79848Cited by top-tier papers6
- TrajAgent: An LLM-Agent Framework for Trajectory Modeling via Large-and-Small Model CollaborationYuwei Du, Jie Feng, Jie Zhao, Yong LiNeurIPS 2025 · 8 citations
- Learning Generalized and Flexible Trajectory Models from Omni-Semantic SupervisionYuanshao Zhu, James Jianqiao Yu, Xiangyu Zhao, Xiao Han et al.KDD 2025 · 2 citations
- SMARTraj2: A Stable Multi-City Adaptive Method for Multi-View Spatio-Temporal Trajectory Representation LearningTangwen Qian, Junhe Li, Yile Chen, Gao Cong et al.NeurIPS 2025 · 1 citation
- Unified Multi-Agent Trajectory Modeling with Masked Trajectory DiffusionSongru Yang, Zhenwei Shi, Zhengxia ZouICCV 2025 · 1 citation
- Interaction-Merged Motion Planning: Effectively Leveraging Diverse Motion Datasets for Robust PlanningGiwon Lee, Wooseong Jeong, Daehee Park, Jaewoo Jeong et al.ICCV 2025 · 1 citation
Builds on13
- Language Models are Few-Shot LearnersTom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah et al.NeurIPS 2020 · 64,255 citations
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- A decoder-only foundation model for time-series forecastingAbhimanyu Das, Weihao Kong, Rajat Sen, Yichen ZhouICML 2024 · 601 citations
- Unified Training of Universal Time Series Forecasting TransformersGerald Woo, Chenghao Liu, Akshat Kumar, Caiming Xiong et al.ICML 2024 · 513 citations
- ClimaX: A foundation model for weather and climateTung Nguyen, Johannes Brandstetter, Ashish Kapoor, Jayesh K. Gupta et al.ICML 2023 · 426 citations
Related papers
- Trajectory World Models for Heterogeneous EnvironmentsShaofeng Yin, Jialong Wu, Siqiao Huang, Xingjian Su et al.ICML 2025
- TransferTraj: A Vehicle Trajectory Learning Model for Region and Task TransferabilityTonglong Wei, Yan Lin, Zeyu Zhou, Haomin Wen et al.NeurIPS 2025 · 6 citations
- TrajFlow: Nation-wide Pseudo GPS Trajectory Generation with Flow Matching ModelsPeiran Li, Jiawei Wang, Haoran Zhang, Xiaodan Shi et al.ICLR 2026 · 4 citations
- SingularTrajectory: Universal Trajectory Predictor Using Diffusion ModelInhwan Bae, Young-Jae Park, Hae-Gon JeonCVPR 2024
- Sports-Traj: A Unified Trajectory Generation Model for Multi-Agent Movement in SportsYi Xu, Yun FuICLR 2025
