3D channel-wise attention network for spatio-temporal traffic raster flow prediction
CSTR:
Author:
Affiliation:

(1. School of Computer and Information Technology, Beijing Jiaotong University, Beijing 100044, China;2. TravelSky Technology Limited, Beijing 101318, China)

Clc Number:

TP391

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Urban traffic flow forecasting is of great significance for traffic management and public safety. However, the correlations of traffic raster flow change with time. There are global spatio-temporal correlations in the city, and the contributions of channel-wise features vary on each city region. To tackle these challenges and make more accurate prediction, a novel spatio-temporal neural network model, named 3D-CANet (three-dimensional channel-wise attention network), was designed. A 3D-InnerCA (three-dimensional inner-channel attention) unit was proposed to dynamically capture the global spatio-temporal correlations for different channel-wise features. Meanwhile, an InterCA (inter-channel attention) unit was designed to adaptively recalibrate the contributions of different channel-wise features on each region. The experimental results on three real-world traffic raster flow datasets demonstrate that the predictive performance of the 3D-CANet model was better than the others,which proved the validity of the model proposed.

    Reference
    Related
    Cited by
Get Citation

TONG Kainan, LIN Youfang, LIU Jun, GUO Shengnan, WAN Huaiyu.3D channel-wise attention network for spatio-temporal traffic raster flow prediction[J]. Journal of National University of Defense Technology,2022,44(3):41-49.

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:June 16,2021
  • Revised:
  • Adopted:
  • Online: June 02,2022
  • Published: June 28,2020
Article QR Code