# Tensor Dirichlet Process Multinomial Mixture Model for Passenger Trajectory Clustering

arXiv: https://arxiv.org/abs/2306.13794 · HTML: https://mlsanity.com/paper/2306.13794/

## Key facts

- Cited by 4 papers in the map (data through 14 Oct 2026).
- Builds on 4 of its 49 references that are in the map.
- First posted on arXiv 23 Jun 2023.

## Builds on (most cited first, 4 of 4)

- [MM-DAG: Multi-task DAG Learning for Multi-modal Data -- with Application for Traffic Congestion Analysis](https://mlsanity.com/paper/2306.02831/) (2023)
- [Dynamic Causal Graph Convolutional Network for Traffic Prediction](https://mlsanity.com/paper/2306.07019/) (2023)
- [Correlated Time Series Self-Supervised Representation Learning via Spatiotemporal Bootstrapping](https://mlsanity.com/paper/2306.06994/) (2023)
- [Adaptive Hierarchical SpatioTemporal Network for Traffic Forecasting](https://mlsanity.com/paper/2306.09386/) (2023)

## Built on by (most cited first, 4 of 4)

- [VisionTraj: A Noise-Robust Trajectory Recovery Framework based on Large-scale Camera Network](https://mlsanity.com/paper/2312.06428/) (2023)
- [Skin Cancer Detection utilizing Deep Learning: Classification of Skin Lesion Images using a Vision Transformer](https://mlsanity.com/paper/2407.18554/) (2024)
- [Deep Causal Learning to Explain and Quantify The Geo-Tension's Impact on Natural Gas Market](https://mlsanity.com/paper/2407.10878/) (2024)
- [Exploring Genre and Success Classification through Song Lyrics using DistilBERT: A Fun NLP Venture](https://mlsanity.com/paper/2407.21068/) (2024)

## Abstract (arXiv)

Passenger clustering based on travel records is essential for transportation operators. However, existing methods cannot easily cluster the passengers due to the hierarchical structure of the passenger trip information, namely: each passenger has multiple trips, and each trip contains multi-dimensional multi-mode information. Furthermore, existing approaches rely on an accurate specification of the clustering number to start, which is difficult when millions of commuters are using the transport systems on a daily basis. In this paper, we propose a novel Tensor Dirichlet Process Multinomial Mixture model (Tensor-DPMM), which is designed to preserve the multi-mode and hierarchical structure of the multi-dimensional trip information via tensor, and cluster them in a unified one-step manner. The model also has the ability to determine the number of clusters automatically by using the Dirichlet Process to decide the probabilities for a passenger to be either assigned in an existing cluster or to create a new cluster: This allows our model to grow the clusters as needed in a dynamic manner. Finally, existing methods do not consider spatial semantic graphs such as geographical proximity and functional similarity between the locations, which may cause inaccurate clustering. To this end, we further propose a variant of our model, namely the Tensor-DPMM with Graph. For the algorithm, we propose a tensor Collapsed Gibbs Sampling method, with an innovative step of "disband and relocating", which disbands clusters with too small amount of members and relocates them to the remaining clustering. This avoids uncontrollable growing amounts of clusters. A case study based on Hong Kong metro passenger data is conducted to demonstrate the automatic process of learning the number of clusters, and the learned clusters are better in within-cluster compactness and cross-cluster separateness.