面向时序图神经网络的增量计算图缓存机制
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

福州市科技重大项目(2024-ZD-005)


Incremental Computational Graph Cache Mechanism for Temporal Graph Neural Network
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    时序图神经网络 (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.

    参考文献
    相似文献
    引证文献
引用本文

刘睿林,吕敏.面向时序图神经网络的增量计算图缓存机制.计算机系统应用,,():1-12

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2026-01-23
  • 最后修改日期:2026-02-14
  • 录用日期:
  • 在线发布日期: 2026-07-14
  • 出版日期:
文章二维码
您是第位访问者
版权所有:中国科学院软件研究所 京ICP备05046678号-3
地址:北京市海淀区中关村南四街4号,邮政编码:100190
电话:010-62661041 传真: Email:csa@iscas.ac.cn
技术支持:北京勤云科技发展有限公司

京公网安备 11040202500063号