Traffic flow prediction, one of the essential problems in traffic control and guidance systems, is still drawing increasing attention in recent years with the new methods tipped by the success of AI. In this paper, we propose a novel model, namely self-attention generative adversarial networks for time-series prediction (SATP-GAN). The SATP-GAN method is based on self-attention and generative adversarial networks (GAN) mechanisms, which are composed of the GAN module and reinforcement learning (RL) module. In the GAN module, we apply the self-attention layer to capture the pattern of time-series data instead of RNNs (recurrent neural networks). In the RL module, we apply the RL algorithm to tune the parameters of our SATP-GAN model. We evaluate the framework on the real-world traffic dataset and obtain a consistent improvement of 6.5% over baseline methods. The SATP-GAN framework proves the GAN mechanism is also available for time-series prediction after fine-tuning the parameters.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    SATP-GAN: self-attention based generative adversarial network for traffic flow prediction


    Contributors:
    Zhang, Liang (author) / Wu, Jianqing (author) / Shen, Jun (author) / Chen, Ming (author) / Wang, Rui (author) / Zhou, Xinliang (author) / Xu, Cankun (author) / Yao, Quankai (author) / Wu, Qiang (author)

    Published in:

    Publication date :

    2021-01-01


    Size :

    17 pages




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    Unknown




    Traffic flow prediction method based on generative adversarial network

    XU DONGWEI / LIN ZHENQIAN / YANG HAO et al. | European Patent Office | 2021

    Free access

    Urban traffic flow prediction method based on generative adversarial network

    ZHU NINGBO / PU BIN / FAN XINXIN et al. | European Patent Office | 2020

    Free access


    Road traffic network state prediction based on a generative adversarial network

    Xu, Dongwei / Peng, Peng / Wei, Chenchen et al. | IET | 2020

    Free access

    Urban traffic flow prediction method, system and equipment based on generative adversarial network

    WEI ZHICHENG / ZHANG TAOYI | European Patent Office | 2023

    Free access