Traffic forecasting aims to capture complex spatial-temporal dependencies and non-linear dynamics, which plays an indispensable role in intelligent transportation systems and other domains like neuroscience, climate, etc. Most recent works rely on graph convolutional networks (GCN) to model the dependencies and the dynamics. However, the over-smoothing issue of GCN would produce indistinguishable features among nodes, leading to poor expressivity and weak capability of modeling complex dependencies and dynamics. To address this issue, we present a novel clustering spatial-temporal (ClusterST) unit, which incorporates unsupervised learning into GCN for extracting discriminative features. Specifically, we first exploit a neural network to learn a dynamic clustering, i.e., learning to partition the neighbors of each node into clusters at each time step. Two probabilistic losses are proposed to improve the separability of clusters. Then, the extracted features of different clusters can be distinguished. Based on the dynamically formed clusters, a vanilla GCN is applied to aggregate features within each cluster. By purely exploiting such a ClusterST unit, large improvements over the state-of-the-art are achieved. Furthermore, ClusterST units with a different number of clusters can be regarded as basic components to construct an inception-like ClusterST network for going deeper. We evaluate the framework on two real-world large-scale traffic datasets and observe an average improvement of 18.19% and 7.62% over state-of-the-art baselines, respectively. The code and models will be publicly available.


    Zugriff

    Zugriff prüfen

    Verfügbarkeit in meiner Bibliothek prüfen

    Bestellung bei Subito €


    Exportieren, teilen und zitieren



    Titel :

    ClusterST: Clustering Spatial–Temporal Network for Traffic Forecasting


    Beteiligte:
    Luo, Guiyang (Autor:in) / Zhang, Hui (Autor:in) / Yuan, Quan (Autor:in) / Li, Jinglin (Autor:in) / Wang, Fei-Yue (Autor:in)


    Erscheinungsdatum :

    2023-01-01


    Format / Umfang :

    20228740 byte




    Medientyp :

    Aufsatz (Zeitschrift)


    Format :

    Elektronische Ressource


    Sprache :

    Englisch




    Multi-Step Spatial-Temporal Fusion Network for Traffic Flow Forecasting*

    Dong, Honghui / Meng, Ziying / Wang, Yiming et al. | IEEE | 2021



    Spatial‐temporal correlation graph convolutional networks for traffic forecasting

    Ru Huang / Zijian Chen / Guangtao Zhai et al. | DOAJ | 2023

    Freier Zugriff

    Dynamic Spatial–Temporal Convolutional Networks for Traffic Flow Forecasting

    Zhang, Hong / Kan, Sunan / Zhang, XiJun et al. | Transportation Research Record | 2023