Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. STATNet: Spatial-temporal attention in the traffic prediction
 
conference paper not in proceedings

STATNet: Spatial-temporal attention in the traffic prediction

Moghadas, Seyed Mohamad
•
Gheibi, Amin
•
Alahi, Alexandre  
June 3, 2022
European Association for Research in Transportation (hEART)

Recent traffic flow prediction methods are lacking abilities to determine predictive features. Thus, they will propagate the error in the next timestamps. In this paper, first, we assess the role of spatial and temporal features on the traffic speed prediction task. Secondly, we propose an attention-based architecture to effectively leverage both cues. Our model mainly consists of two major building blocks to capture the spatial-temporal features in the data and dynamically calculate the attentive features. More specifically, the first block sequentially applies temporal convolution to produce time-based features and then employs graph convolution to capture spatial features. The second component determines the attention between spatial and temporal features. The combination of the component’s output will be calculated to generate the final prediction results. Experiments on two real-world large-scale road network traffic datasets (i.e., METR-LA and PEMS-BAY) demonstrate that the proposed STATNet (spatial-temporal attention traffic network) model outperforms the state-of-the-art baselines such as graph-wavenet and STGCN (spatial-temporal graph convolutional networks).

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

140.pdf

Type

N/a

Access type

openaccess

License Condition

copyright

Size

565.86 KB

Format

Adobe PDF

Checksum (MD5)

43c08f90af7cdac95f5adbd25a941556

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés