NGDB-Zoo: Towards Efficient and Scalable Neural Graph Databases Training
zhongwei xie ⋅ Jiaxin Bai ⋅ Shujie LIU ⋅ Haoyu Huang ⋅ LI Yufei ⋅ Yisen Gao ⋅ Hong Ting Tsang ⋅ Yangqiu Song
Abstract
Neural Graph Databases (NGDBs) support complex logical reasoning over incomplete knowledge structures, yet their training efficiency and expressivity are constrained by rigid query-level batching and structure-only embeddings. We present NGDB-Zoo, a unified framework that resolves these bottlenecks by synergizing operator-level training with semantic augmentation. By decoupling logical operators from query topologies, NGDB-Zoo transforms training loops into dynamically scheduled data-flow executions, enabling multi-stream parallelism and achieving a $1.8\times$-$6.8\times$ average throughput compared to baselines. Furthermore, we formalize a decoupled architecture to integrate semantic priors from pre-trained text embeddings without triggering I/O stalls or memory overflows. Experiments on six benchmarks, including $\textit{ogbl-wikikg2}$ and $\textit{ATLAS-Wiki}$, show that NGDB-Zoo scales dense query-embedding training to million-entity graphs while maintaining competitive filtered MRR.
Chat is not available.
Successful Page Load