arXiv cs.LG· Victor Pekkari, Harry Wik, Alfred Clemedtson, Borun Shi, Xuan-Son Vu·· 19 天前AI 评分35
图神经网络在图数据库中直接进行训练与推理的研究探索
Towards Training and Inference of Graph Neural Networks in Graph Databases
AI 导读
研究人员实现了直接在图数据库内进行图神经网络(GNN)训练与推理,可直接执行采样、特征检索甚至完整的前向传播。通过 Cypher 进行实时采样和检索并结合优化方案,训练过程在保持相同准确率的同时提速约 2.5x。在真实图数据上,数据库内推理相比将图冷加载至 RAM 提速超过 10x,且无需额外内存开销。
来源:arXiv cs.LG · arxiv.org