Abstract:Spatial temporal imputation is an important part of intelligent transportation system, and spatial temporal data missing also largely affects the performance of its downstream tasks. Spatial temporal data are affected by spatial, temporal, and unexpected conditions, and has a complex change pattern. Existing researches mostly focus on time series and often ignore the modeling of spatial dependency. Moreover, the imputation performance is poor under high missing rate or block missing scenarios, needing to enhance the model robustness. Therefore, we proposed a spatio-temporal generative adversarial imputation network (named STGAIN) to handle with the complex spatial temporal dependency and the diversity of missing conditions on spatio-temporal data. It includes a spatio-temporal generator and a spatio-temporal discriminator, which uses a graph convolutional neural network as a spatial aggregator and a one-dimensional deformable convolutional neural network as a temporal extractor. The generator observes the unmissing part data and the spatial relationships to impute the missing parts. The discriminator then takes the completed data as input and attempts to determine which components were actually observed and which were imputed. The spatio-temporal modeling ability was improved in the competition with each other. In order to make the generator in GAN distinguish the observed part from the missing part, a fusion loss function (including reconstruction loss and BCE loss) was designed for the generator training. Finally, experiments were conducted on two real highway datasets for different missing rates. And we test the model robustness under different size of dataset and noise interference. The results show that STGAIN significantly outperforms the existing methods.

Modified GAN Model for Traffic Missing Data Imputation

A Bipartite Graph Based Method for Traffic Continuous Data Imputation

AGNP: Network-wide Short-Term Probabilistic Traffic Speed Prediction and Imputation

A Missing Traffic Data Imputation Method Based on a Diffusion Convolutional Neural Network–Generative Adversarial Network

Spatial-temporal traffic data imputation based on dynamic multi-level generative adversarial networks for urban governance

Missing Data Repairs for Traffic Flow With Self-Attention Generative Adversarial Imputation Net

An Experimental Evaluation of Imputation Models for Spatial-Temporal Traffic Data

Traffic Volume Imputation Using Attention-based Spatiotemporal Generative Adversarial Imputation Network

A Latent-Factor-Model-Based Approach for Traffic Data Imputation with Road Network Information

Missing Traffic Data: Comparison of Imputation Methods

A hybrid model for missing traffic flow data imputation based on clustering and attention mechanism optimizing LSTM and AdaBoost

Bidirectional Spatial-Temporal Traffic Data Imputation via Graph Attention Recurrent Neural Network

Traffic Accident Data Generation Based on Improved Generative Adversarial Networks

Traffic Prediction With Missing Data: A Multi-Task Learning Approach

Graph Markov Network for Traffic Forecasting with Missing Data

IGANI: Iterative Generative Adversarial Networks for Imputation with Application to Traffic Data

MagiNet: Mask-Aware Graph Imputation Network for Incomplete Traffic Data

Efficient Generative Adversarial Networks for Imbalanced Traffic Collision Datasets

Robust spatial temporal imputation based on spatio-temporal generative adversarial nets

Improved generative adversarial imputation networks for missing data

A novel reinforced dynamic graph convolutional network model with data imputation for network-wide traffic flow prediction