Abstract:As a crucial component in intelligent transportation systems, traffic flow prediction has recently attracted widespread research interest in the field of artificial intelligence (AI) with the increasing availability of massive traffic mobility data. Its key challenge lies in how to integrate diverse factors (such as temporal rules and spatial dependencies) to infer the evolution trend of traffic flow. To address this problem, we propose a unified neural network called Attentive Traffic Flow Machine (ATFM), which can effectively learn the spatial-temporal feature representations of traffic flow with an attention mechanism. In particular, our ATFM is composed of two progressive Convolutional Long Short-Term Memory (ConvLSTM \cite{xingjian2015convolutional}) units connected with a convolutional layer. Specifically, the first ConvLSTM unit takes normal traffic flow features as input and generates a hidden state at each time-step, which is further fed into the connected convolutional layer for spatial attention map inference. The second ConvLSTM unit aims at learning the dynamic spatial-temporal representations from the attentionally weighted traffic flow features. Further, we develop two deep learning frameworks based on ATFM to predict citywide short-term/long-term traffic flow by adaptively incorporating the sequential and periodic data as well as other external influences. Extensive experiments on two standard benchmarks well demonstrate the superiority of the proposed method for traffic flow prediction. Moreover, to verify the generalization of our method, we also apply the customized framework to forecast the passenger pickup/dropoff demands in traffic prediction and show its superior performance. Our code and data are available at {\color{blue}\url{<a class="link-external link-https" href="https://github.com/liulingbo918/ATFM" rel="external noopener nofollow">this https URL</a>}}.

Attention Based Convolutional Networks for Traffic Flow Prediction

Attention Based Graph Covolution Networks for Intelligent Traffic Flow Analysis

Urban Road Traffic Flow Prediction with Attention-Based Convolutional Bidirectional Long Short-Term Memory Networks

Forecasting traffic flow with spatial–temporal convolutional graph attention networks

Deep spatio-temporal neural network based on interactive attention for traffic flow prediction

Spatial-Temporal Convolutional Graph Attention Networks for Citywide Traffic Flow Forecasting

A Hybrid Deep Learning Model With Attention-Based Conv-LSTM Networks for Short-Term Traffic Flow Prediction

Traffic Flow Prediction Based on Spatio-temporal Attention Block

A Short-term Traffic Flow Forecasting Model Based on Spatial-temporal Attention Neural Network

STA-GCN: Spatial-Temporal Self-Attention Graph Convolutional Networks for Traffic-Flow Prediction

A Deep Learning Framework Based on Spatio-Temporal Attention Mechanism for Traffic Prediction.

Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting

Attention-Based Spatial–Temporal Convolution Gated Recurrent Unit for Traffic Flow Forecasting

An effective spatial-temporal attention based neural network for traffic flow prediction

Dynamic Graph Convolutional Network with Attention Fusion for Traffic Flow Prediction

Dynamic Spatial-Temporal Representation Learning for Traffic Flow Prediction

Attention-Based Gated Recurrent Graph Convolutional Network for Short-Term Traffic Flow Forecasting

Dynamic multi-scale spatial-temporal graph convolutional network for traffic flow prediction

A time-dependent attention convolutional LSTM method for traffic flow prediction

The Deep 3D Convolutional Multi-Branching Spatial-Temporal-Based Unit Predicting Citywide Traffic Flow

Urban Traffic Flow Prediction with Deep Neural Network