arXiv cs.LG· Victor Pekkari, Harry Wik, Alfred Clemedtson, Borun Shi, Xuan-Son Vu·· 7 小时前AI 评分22
图数据库中图神经网络的训练与推理研究
Towards Training and Inference of Graph Neural Networks in Graph Databases
AI 导读
研究显示图数据库可直接用于 GNN 训练与推理,执行采样、特征检索乃至完整前向传播,训练流程可通过 Cypher 实现实时采样与检索。所提优化在保持精度不变的前提下将训练速度提升约 2.5 倍,完全在库内推理比冷加载图到 RAM 快 10 倍以上且无额外内存开销。
来源:arXiv cs.LG · arxiv.org