arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~

用于大规模动态图的可扩展高效联合脉冲嵌入预测架构

Scalable and Efficient Joint Spiking Embedding Predictive Architecture for Large-Scale Dynamic Graphs

Huizhe Zhang, Yuchang Zhu, Huazhen Zhong, Liang Chen, Zibin Zheng

arXiv 2607.18412首次发表:更新:

发表机构

Sun Yat-sen University(中山大学)

机构由 AI 辅助整理,请以论文原文为准。

AI 中文总结

针对现实世界动态图标记数据稀缺问题,提出SG-JEPA架构,通过沿时间维度划分节点并利用脉冲神经元编码,学习相互预测的嵌入,在节点分类任务中性能优于判别基线,且训练效率和内存可扩展性高。

AI 中文摘要

动态图学习旨在捕捉现实世界系统中不断演变的结构和语义模式,如欺诈检测和推荐系统。由于现实世界动态图中标记数据稀缺,近期研究引入生成或对比范式生成与任务无关的图嵌入,但这些方法存在计算开销大的问题。本文提出用于大规模动态图的联合脉冲嵌入预测架构SG-JEPA,它沿时间维度将节点分为上下文和目标集,通过额外时空信息学习相互预测的嵌入。此外,脉冲神经元将序列输入编码为从粗到细的脉冲计数嵌入,使SG-JEPA能适应下游任务的不同计算约束。实验表明,SG-JEPA在节点分类上比判别基线有竞争力甚至更优,能有效扩展到含1300万条边的动态图,训练效率和内存可扩展性优于先前自监督动态图基线。

英文摘要

Dynamic graph learning aims to capture evolving structural and semantic patterns in real-world systems, such as fraud detection and recommender systems. Due to the scarcity of labeled data in real-world dynamic graphs, recent studies have introduced generative or contrastive paradigms (e.g., masked graph autoencoders or graph contrastive learning) to generate task-agnostic graph embeddings. However, these methods typically rely on complex edge-level reconstruction objectives and tailored graph augmentation strategies. This incurs substantial computational overhead when scaling to large-scale dynamic graphs. In this paper, we propose SG-JEPA, a joint spiking embedding predictive architecture for large-scale dynamic graphs. In contrast to existing self-supervised methods, SG-JEPA partitions nodes into context and target sets along the temporal dimension to learn embeddings that are predictive of each other via additional spatial-temporal information. Furthermore, through encoding sequential inputs into coarse-to-fine spike count embeddings, spiking neurons enable SG-JEPA to adapt to the varying computational constraints of downstream tasks. Extensive experiments demonstrate that SG-JEPA achieves competitive or even superior performance over discriminative baselines on node classification, while effectively scaling to the dynamic graph with 13 million edges. SG-JEPA avoids the complex machinery (negative sampling, graph augmentations, edge-level reconstruction, etc.), resulting in superior training efficiency and memory scalability compared with prior self-supervised dynamic graph baselines.

论文原文

arXiv 摘要页 · PDF 原文 · HTML 原文

↑