IDEAS home Printed from https://ideas.repec.org/a/wsi/ijmpcx/v32y2021i12ns0129183121501588.html
   My bibliography  Save this article

A combined traffic flow forecasting model based on graph convolutional network and attention mechanism

Author

Listed:
  • Hong Zhang

    (College of Computer & Communication, Lanzhou University of Technology, Lanzhou 730050, P. R. China)

  • Linlong Chen

    (College of Computer & Communication, Lanzhou University of Technology, Lanzhou 730050, P. R. China)

  • Jie Cao

    (College of Computer & Communication, Lanzhou University of Technology, Lanzhou 730050, P. R. China)

  • Xijun Zhang

    (College of Computer & Communication, Lanzhou University of Technology, Lanzhou 730050, P. R. China)

  • Sunan Kan

    (College of Computer & Communication, Lanzhou University of Technology, Lanzhou 730050, P. R. China)

Abstract

Accurate traffic flow forecasting is a prerequisite guarantee for the realization of intelligent transportation, but due to the complex spatiotemporal characteristics of traffic flow, its forecasting has always been difficult. Deep learning can learn the deep spatiotemporal characteristics of traffic flow from a large amount of data. Deep learning can learn the deep spatiotemporal characteristics of traffic flow from a large amount of data. This paper establishes a novel combination forecasting model GGCN-SA based on deep learning for traffic flow to effectively capture the spatiotemporal characteristics of traffic flow and improve forecasting accuracy. The model captures the spatial correlation of the road traffic network through the graph convolutional network (GCN), captures the time dependence of the traffic flow through the gated recursive unit (GRU), and further introduces the soft attention mechanism (Soft Attention) to aggregate different neighborhoods Spatio-temporal information within the range to enhance the model’s ability to characterize the temporal and spatial characteristics of traffic flow. A large number of experiments have been conducted on the METR-LA and SZ-taxi data sets. The experimental results show that the GGCN-SA model proposed in this paper has better forecasting performance compared with the baseline methods.

Suggested Citation

  • Hong Zhang & Linlong Chen & Jie Cao & Xijun Zhang & Sunan Kan, 2021. "A combined traffic flow forecasting model based on graph convolutional network and attention mechanism," International Journal of Modern Physics C (IJMPC), World Scientific Publishing Co. Pte. Ltd., vol. 32(12), pages 1-21, December.
  • Handle: RePEc:wsi:ijmpcx:v:32:y:2021:i:12:n:s0129183121501588
    DOI: 10.1142/S0129183121501588
    as

    Download full text from publisher

    File URL: http://www.worldscientific.com/doi/abs/10.1142/S0129183121501588
    Download Restriction: Access to full text is restricted to subscribers

    File URL: https://libkey.io/10.1142/S0129183121501588?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:wsi:ijmpcx:v:32:y:2021:i:12:n:s0129183121501588. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Tai Tone Lim (email available below). General contact details of provider: http://www.worldscinet.com/ijmpc/ijmpc.shtml .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.