arXiv:2605.05738cs.LGcs.AI2026-05

提出CoMemNet框架,解决交通预测中持续学习的遗忘问题。

CoMemNet: Contrastive Sampling with Memory Replay Network for Continual Traffic Prediction

论文配图:CoMemNet: Contrastive Sampling with Memory Replay Network for Continual Traffic Prediction
图 1 · 摘自论文原文
  • 双分支架构:在线分支快速预测,目标分支用Wasserstein距离提取历史特征
  • 动态对比采样器选出变化显著节点,减少灾难性遗忘
  • 轻量记忆缓冲区防止内存爆炸,适合实时交通流预测场景

近年来,将非拓扑空间建模与时间学习方法结合已成为捕捉非欧几里得图中时空信息的有效途径。然而,现有方法多依赖静态底层图结构,难以应对流式交通网络中持续扩展和演变的模式。为此,本文提出一种简单高效的双分支持续学习框架CoMemNet,用于交通预测。快速收敛的在线分支承担主要预测任务,而动量更新的目标分支利用Wasserstein距离特征提取历史信息,构建动态对比采样器(DC Sampler),选择具有显著动态网络特征变化的节点集进行训练,有效缓解灾难性遗忘。此外,骨干网络引入轻量级节点自适应时间记忆缓冲区(TMRB-N),通过记忆重放巩固旧知识,降低内存爆炸风险。最后,我们构建了两个新的开源数据集。实验结果表明,CoMemNet在三个大规模真实世界数据集上均达到当前最优性能。代码已公开于https://github.com/meiwu5/CoMemNet。

原文摘要 · Abstract (English)

In recent years, the integration of non-topological space modeling with temporal learning methods has emerged as an effective approach for capturing spatio-temporal information in non-Euclidean graphs. However, most existing methods rely on static underlying graph structures, which are inadequate for capturing the continuously expanding and evolving patterns in streaming traffic networks. To address this challenge, we propose a simple yet efficient dual-branch continual learning framework for traffic prediction, named CoMemNet. The fast-converging Online branch undertakes the primary prediction tasks, while the momentum-updated Target branch extracts historical information using Wasserstein Distance features to create a Dynamic Contrastive Sampler (DC Sampler). This sampler selects a node set with significant dynamic network feature changes for training, effectively mitigating the issue of catastrophic forgetting. Additionally, the backbone incorporates a lightweight Node-Adaptive Temporal Memory Buffer (TMRB-N) to consolidate old knowledge through memory replay and address the risk of memory explosion. Finally, we provide two newly curated open-source datasets. Experimental results demonstrate that CoMemNet achieves state-of-the-art (SOTA) performance across all three large-scale real-world datasets. The code is available at: https://github.com/meiwu5/CoMemNet.

交通预测持续学习记忆重放图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。