本文已被:浏览 15次 下载 36次
Received:January 23, 2026 Revised:February 14, 2026
Received:January 23, 2026 Revised:February 14, 2026
中文摘要: 时序图神经网络 (temporal graph neural network, TGNN) 在动态图建模学习中表现出优异效果, 广泛应用于时序预测等任务中. TGNN对时间序列的建模依赖多层时序采样, 深层模型虽具备提升表示能力的潜力, 但其发展受限于随模型层数和采样邻居数量呈指数级增长的复杂度. 现有研究主要从数据传输角度优化, 而针对采样阶段的优化仍存在较大探索空间. 本文以TGNN中普遍采用的最近邻采样方法为研究对象, 分析其增量特性, 据此设计支持增量更新的计算图缓存机制IncCGCache, 通过复用历史采样结果并增量更新, 避免重复采样, 在缓存命中时将复杂度降低一个数量级. 本文利用向量化索引实现缓存操作的并行处理, 将缓存更新与模型计算解耦, 使缓存更新可以异步执行, 从而进一步提升加速比. 此外, 为缓解常用随机负采样对缓存的不利影响, 本文提出面向缓存的负采样策略, 在基本不影响模型性能的前提下改善缓存命中率. 基于TGN、TIGER和SEAN等代表性模型和常用数据集的实验结果表明, IncCGCache能够对采样阶段提供达3.68倍的加速效果.
Abstract:Temporal graph neural network (TGNN) has demonstrated strong performance in dynamic graph representation learning and are widely used in temporal prediction tasks. TGNNs rely on multi-layer temporal sampling to capture temporal sequences. Although deeper models have the potential to improve representation capability, their development is constrained by sampling complexity, which grows exponentially with the number of layers and sampled neighbors. Existing studies mainly focus on optimizing data transfer, while the sampling stage remains under explored. This study investigates the widely used most-recent neighbor sampling strategy in TGNNs, analyzes its incremental property, and proposes an incremental computational graph caching scheme, IncCGCache, which reuses historical sampling results and updates them incrementally to eliminate redundant sampling, reducing the complexity by an order of magnitude under cache hits. Vectorized indexing is employed to enable parallel cache operations, while cache updates are decoupled from model computation and executed asynchronously, further improving the speedup. In addition, to mitigate the negative impact of commonly used random negative sampling on cache effectiveness, a cache-aware negative sampling strategy is proposed, which improves the cache hit rate with negligible impact on model performance. Experimental results on representative models such as TGN, TIGER, and SEAN and commonly used datasets demonstrate that IncCGCache provides up to 3.68× speedup for the sampling stage.
文章编号: 中图分类号: 文献标志码:
基金项目:福州市科技重大项目(2024-ZD-005)
引用文本:
刘睿林,吕敏.面向时序图神经网络的增量计算图缓存机制.计算机系统应用,,():1-12
LIU Rui-Lin,LYU Min.Incremental Computational Graph Cache Mechanism for Temporal Graph Neural Network.COMPUTER SYSTEMS APPLICATIONS,,():1-12
刘睿林,吕敏.面向时序图神经网络的增量计算图缓存机制.计算机系统应用,,():1-12
LIU Rui-Lin,LYU Min.Incremental Computational Graph Cache Mechanism for Temporal Graph Neural Network.COMPUTER SYSTEMS APPLICATIONS,,():1-12

