A Novel Clustering Model TEC for Station Classification

Abstract

Station classification plays an important role in urban intelligent transportation systems. Recently, most of studies usually choose different features to analyze the function of stations. Moreover, they usually choose K-means as the clustering method, which may lead to unstable clustering results. Therefore, a novel clustering model (Transformer Encoder Clustering, TEC) has been proposed for station classification by only using passenger flow data. Firstly, transformer encoder has been applied to deeply extract the features from original passenger flow. Then, an improved Canopy + has been proposed to determine the initial centers for K-means. Finally, a novel evaluation metric (NR) has been introduced to verify the stability and volatility of station classification results. Experiments are conducted on Xiamen bus rapid transit ridership dataset. The results show that the 44 stations have been classified into three categories, and the NRs in all stations are 0.75, which verifies the stability of results for station classification.