Tucker factorization-based tensor completion for robust traffic data imputation

Cheng Lyu, Qing Long Lu, Xinhua Wu, Constantinos Antoniou

Research output: Contribution to journalArticlepeer-review

2 Scopus citations

Abstract

Missing values are prevalent in spatio-temporal traffic data, undermining the quality of data-driven analysis. While prior works have demonstrated the promise of tensor completion methods for imputation, their performance remains limited for complicated composite missing patterns. This paper proposes a novel imputation framework combining tensor factorization and rank minimization, which is effective in capturing key traffic dynamics and eliminates the need for exhaustive rank tuning. The framework is further supplemented with time series decomposition to account for trends, spatio-temporal correlations, and outliers, with the intention of improving the robustness of imputation results. A Bregman ADMM algorithm is designed to solve the resulting multi-block nonconvex optimization efficiently. Experiments on four real-world traffic state datasets suggest that the proposed framework outperforms state-of-the-art imputation methods, including the context of complex missing patterns with high missing rates, while maintaining reasonable computation efficiency. Furthermore, the robustness of our model in extreme missing data scenarios, as well as under perturbation in hyperparameters, has been validated. These results also underscore the potential benefits of incorporating temporal modeling for more reliable imputation.

Original languageEnglish
Article number104502
JournalTransportation Research Part C: Emerging Technologies
Volume160
DOIs
StatePublished - Mar 2024

Keywords

  • Bregman ADMM
  • Changepoint detection
  • Data imputation
  • Tensor completion
  • Tucker factorization

Fingerprint

Dive into the research topics of 'Tucker factorization-based tensor completion for robust traffic data imputation'. Together they form a unique fingerprint.

Cite this