Self-supervised contrastive representation learning for large-scale trajectories

Shuzhe Li,Wei Chen,Bingqi Yan,Zhen Li,Shunzhi Zhu,Yanwei Yu
DOI: https://doi.org/10.1016/j.future.2023.05.033
IF: 7.307
2023-06-21
Future Generation Computer Systems
Abstract:Trajectory representation learning aims to embed trajectory sequences into fixed-length vector representations while preserving their original spatio-temporal feature proximity. Existing works either learn trajectory representations for specific mining tasks or fail to utilize large amounts of unlabeled trajectory data for representation learning. In this work, we propose a self-supervised Traj ectory representation learning based on R econstruction C ontrastive L earning called TrajRCL. To be specific, TrajRCL first obtains low-distortion and high-fidelity views of trajectories through trajectory augmentation. Then, TrajRCL leverages a Transformer based encoder–decoder network to reconstruct low-distortion view trajectories to approximate high-fidelity trajectories. Self-supervised contrastive learning is finally used to enhance the consistency of the two view's trajectory representations. Extensive experiments on two real-world demonstrate the superiority of our model over state-of-the-art baselines and significant efficiency on similarity trajectory search and k -NN query.
computer science, theory & methods
What problem does this paper attempt to address?