mlsanityData to 14 Oct 2026

CoMemNet: A Continual Memory Network with Drift-Aware Sampling for Traffic Prediction

arXiv 2605.05738 ↗First posted 7 May 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Traffic sensor networks evolve as sensors are added and traffic distributions change, whereas most forecasting models assume a fixed node set and repeatedly retrain on all available data. We propose CoMemNet, a Continual Memory Network for efficient prediction over evolving traffic sensor networks. CoMemNet uses an Online branch to adapt to the current period and an exponential-moving-average Target branch as a stable feature reference. A Wasserstein-based Drift Sampler compares node-wise Online-Target feature distributions and selects a limited set of drift-sensitive nodes for updating. A lightweight Node-Adaptive Temporal Memory Replay Buffer (TMRB-N) retains compact temporal states without repeatedly traversing all historical training data. The prediction backbone does not consume an adjacency matrix; sensor adjacency is used only to construct data and optionally expand the selected update set to a limited neighborhood. Experiments on three multi-period PeMS datasets include three-seed evaluation, strong static retraining and continual baselines, controlled sampling strategies, continual-learning metrics, robustness tests, and resource accounting. The results show that CoMemNet maintains stable prediction accuracy and efficient adaptation under bounded shared-node selection, achieving a better balance between historical knowledge preservation and current-period prediction performance. Meanwhile, as the evolving network expands, CoMemNet shows clearer accuracy and cumulative training-time advantages over current-period retraining baselines. The code is available at:https://meiwu5.github.io/CoMemNet.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 45 references that are in the map. First posted on arXiv 7 May 2026.

Builds on

23 of 45 references.

References · 23

  1. 2016iCaRL: Incremental Classifier and Representation Learningcited by 2,333
  2. 2022A Time Series is Worth 64 Words: Long-term Forecasting with Transformerscited by 2,039
  3. 2017Continual Learning with Deep Generative Replaycited by 1,158
  4. 2018Experience Replay for Continual Learningcited by 1,095
  5. 2017PackNet: Adding Multiple Tasks to a Single Network by Iterative Pruningcited by 898
  6. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 872
  7. 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 829
  8. 2019On Tiny Episodic Memories in Continual Learningcited by 307
  9. 2018Task-Free Continual Learningcited by 221
  10. 2019Compacting, Picking and Growing for Unforgetting Continual Learningcited by 207
  11. 2022PromptCast: A New Prompt-based Learning Paradigm for Time Series Forecastingcited by 172
  12. 2022Spatial-Temporal Identity: A Simple yet Effective Baseline for Multivariate Time Series Forecastingcited by 151
11 more
  1. 2023PDFormer: Propagation Delay-Aware Dynamic Long-Range Transformer for Traffic Flow Predictioncited by 137
  2. 2024ST-LLM: Large Language Models Are Effective Temporal Learnerscited by 126
  3. 2022Decoupled Dynamic Spatial-Temporal Graph Neural Network for Traffic Forecastingcited by 123
  4. 2018Streaming Graph Neural Networkscited by 91
  5. 2022Spatio-Temporal Self-Supervised Learning for Traffic Flow Predictioncited by 77
  6. 2021TrafficStream: A Streaming Traffic Flow Forecasting Framework Based on Graph Neural Networks and Continual Learningcited by 42
  7. 2017Spatiotemporal Recurrent Convolutional Networks for Traffic Prediction in Transportation Networkscited by 35
  8. 2023GATGPT: A Pre-trained Large Language Model with Graph Attention Network for Spatiotemporal Imputationcited by 35
  9. 2023Spatial-Temporal-Decoupled Masked Pre-training for Spatiotemporal Forecastingcited by 27
  10. 2024Topology-aware Embedding Memory for Continual Learning on Expanding Networkscited by 14
  11. 2024Expand and Compress: Exploring Tuning Principles for Continual Spatio-Temporal Graph Forecastingcited by 10

Built on by

Not cited yet.