ProDiff: Prototype-Guided Diffusion for Minimal Information Trajectory Imputation
Tianci Bu, Le Zhou, Wenchuan Yang, Jianhong Mou, Kang Yang, Suoyi Tan, Feng Yao, Jingyuan Wang, Xin Lu
Abstract
Trajectory data is crucial for various applications but often suffers from incompleteness due to device limitations and diverse collection scenarios. Existing imputation methods rely on sparse trajectory or travel information, such as velocity, to infer missing points. However, these approaches assume that sparse trajectories retain essential behavioral patterns, which place significant demands on data acquisition and overlook the potential of large-scale human trajectory embeddings. To address this, we propose ProDiff, a trajectory imputation framework that uses only two endpoints as minimal information. It integrates prototype learning to embed human movement patterns and a denoising diffusion probabilistic model for robust spatiotemporal reconstruction. Joint training with a tailored loss function ensures effective imputation. ProDiff outperforms stateof-the-art methods, improving accuracy by 6.28% on FourSquare and 2.52% on WuXi. Further analysis shows a 0.927 correlation between generated and real trajectories, demonstrating the effectiveness of our approach.
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.
Builds on17
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Transformers are RNNs: Fast Autoregressive Transformers with Linear AttentionAngelos Katharopoulos, Apoorv Vyas, Nikolaos Pappas, François FleuretICML 2020 · 2,665 citations
- PDFormer: Propagation Delay-Aware Dynamic Long-Range Transformer for Traffic Flow PredictionJiawei Jiang, Chengkai Han, Wayne Xin Zhao, Jingyuan WangAAAI 2023 · 542 citations
- TimesNet: Temporal 2D-Variation Modeling for General Time Series AnalysisHaixu Wu, Tengge Hu, Yong Liu, Hang Zhou et al.ICLR 2023 · 423 citations
Related papers
- Generative Human Trajectory Recovery via Embedding-Space Conditional DiffusionKaijun Liu, Sijie Ruan, Liang Zhang, Cheng Long et al.ICML 2025
- TRACE: Trajectory Recovery with State Propagation Diffusion for Urban MobilityJinming Wang, Hai Wang, Hongkai Wen, Geyong Min et al.WWW 2026
- TERI: An Effective Framework for Trajectory Recovery with Irregular Time IntervalsYile Chen, Gao Cong, Cuauhtemoc AndaVLDB 2024 · 22 citations
- DiffTraj: Generating GPS Trajectory with Diffusion Probabilistic ModelYuanshao Zhu, Yongchao Ye, Shiyao Zhang, Xiangyu Zhao et al.NeurIPS 2023 · 134 citations
- RDPI: A Refine Diffusion Probability Generation Method for Spatiotemporal Data ImputationZijin Liu, Xiang Zhao, You SongAAAI 2025
