Knowledge Expansion and Consolidation for Continual Traffic Prediction With Expanding Graphs

Binwu Wang,Yudong Zhang,Jiahao Shi,Pengkun Wang,Xu Wang,Lei Bai,Yang Wang
DOI: https://doi.org/10.1109/tits.2023.3263904
IF: 8.5
2023-01-01
IEEE Transactions on Intelligent Transportation Systems
Abstract:Accurate traffic prediction plays a vital role in intelligent transport managements and applications. However, in the vast majority of existing works, the focus is mainly on modeling spatiotemporal correlations in static traffic networks. Thus, the continuous expansion and evolution of traffic networks are ignored. In this work, we study the problem of traffic prediction with expanding road network structures under the continual learning paradigm. Considering the model prediction performance, efficiency, and data accessibility, a SpatioTemporal Knowledge Expansion and Consolidation (STKEC) framework is proposed. This framework contains an influence-based knowledge expansion strategy to help the spatiotemporal learning model integrate new spatiotemporal traffic patterns and a memory-augmented knowledge consolidation mechanism to preserve the learned spatiotemporal patterns without accessing the data in previous graphs. Extensive experiments are conducted on a large-scale dataset and verify the superior performance of STKEC in continual traffic prediction.
engineering, electrical & electronic,transportation science & technology, civil
What problem does this paper attempt to address?