A Bipartite Graph Based Method for Traffic Continuous Data Imputation

Peng Jiaheng,Guan Tong,Liang Jun
DOI: https://doi.org/10.1109/cac59555.2023.10451480
2023-01-01
Abstract:Traffic information missing conditions are common in Intelligent Transportation System and would cause huge challenges for sequential traffic tasks, such as traffic speed prediction. Rare models are designed to tackle the long-range continuous data missing problems. Thus, in this paper, we propose a Bipartite Graph-based message passing Imputation Network (BGIN) to reconstruct traffic features data of nodes. In BGIN, the nodes are divided into two types: known nodes that contain traffic information and imputed nodes without data. First, a known information depth extraction module is given to achieve the messages passing among known nodes. Then, we construct a bipartite graph to represent the correlations between known and imputed nodes. Bipartite graph propagation module propagates the known information to imputed nodes. Finally, a imputation output module aggregates the messages to complete the imputation task. To evaluate the performance of our method, extensive experiments are conducted on a newly traffic imputation dataset called PeMS-impute. Experiment results demonstrate that BGIN achieves the best imputation effects than some graph based traffic prediction models and imputation models.
What problem does this paper attempt to address?