跳到正文
原文
arXiv 机器学习· Victor Pekkari, Harry Wik, Alfred Clemedtson, Borun Shi, Xuan-Son Vu·· 7 小时前AI 评分36

图数据库内图神经网络(GNN)训练与推理方法研究

Towards Training and Inference of Graph Neural Networks in Graph Databases

AI 导读

研究人员证实图数据库可直接用于图神经网络(GNN)的训练与推理,直接在库内执行采样、特征检索乃至完整前向传播。通过使用 Cypher 进行即时采样和检索,配合相关优化在保持精度的同时将训练过程加速约 2.5x。此外,在真实世界图上实现的库内推理无需额外内存开销,速度比冷加载图到 RAM 快 >10x。

来源:arXiv 机器学习 · arxiv.org