Parameter-Transferred Irreducible LSTM for Traffic Data Imputation

We propose an imputation algorithm for missing spatiotemporal data based on long short-term memory (LSTM) model factorization in a traffic environment where the roadside units (RSUs) collect traffic speed data. We considered a scenario where data collection by RSUs occurs for each road segment, but...

Full description

Saved in:
Bibliographic Details
Published in:IEEE sensors journal Vol. 24; no. 14; pp. 22178 - 22188
Main Authors: Kwon, Jungmin, Park, Hyunggon
Format: Journal Article
Language:English
Published: New York IEEE 15.07.2024
The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
Subjects:
Accuracy
Algorithms
Complexity theory
Data collection
Data imputation
Data models
Datasets
Error analysis
extrapolation
Factorization
Interpolation
Long short term memory
long short-term memory (LSTM)
Missing data
parameter transfer
Parameters
Roads
Roads & highways
Roadsides
Root-mean-square errors
Segments
Sensors
spatial interpolation
Spatiotemporal data
Traffic congestion
Traffic information
Traffic speed
Training
ISSN:1530-437X, 1558-1748
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:We propose an imputation algorithm for missing spatiotemporal data based on long short-term memory (LSTM) model factorization in a traffic environment where the roadside units (RSUs) collect traffic speed data. We considered a scenario where data collection by RSUs occurs for each road segment, but the absence of RSUs results in an incomplete dataset. To enhance imputation accuracy, we mitigate the risk of error propagation of model training on the entire dataset and take into account the spatiotemporal correlation of the dataset. The proposed algorithm can reduce the dimensionality of the input dataset by employing an adjacency matrix to identify data both highly correlated and connected to the target road segment, subsequently transforming parallel datasets into a serial format. Then, we extrapolate the missing data using an irreducible LSTM model, which is a factorization of a standard LSTM model. To enhance imputation performance, we also adopt spatial interpolation on extrapolated data across multiple paths that lead to the target road segment. Extensive experiment results using synthetic and real-world datasets confirm that the proposed algorithm outperforms other imputation algorithms in terms of imputation accuracy measured by the root mean square error (RMSE) and the mean absolute error (MAE) as well as the space complexity measured by the number of model parameters. In particular, the experiments with real-world datasets show that the proposed algorithm consistently achieves high imputation accuracy across a wide range of traffic scenarios, including actual traffic congestion and rapid traffic fluctuations.
Bibliography:ObjectType-Article-1
SourceType-Scholarly Journals-1
ObjectType-Feature-2
content type line 14
ISSN:1530-437X
1558-1748
DOI:10.1109/JSEN.2024.3392938