Sep, 2019

深度图库:面向图的、高性能的图神经网络包

TL;DR本文介绍了 Deep Graph Library(DGL)的设计原则和实现方法,DGL 将 GNNs 的计算模式提炼成几个泛化的稀疏张量操作,适合大规模并行化。DGL以图形为中心编程抽象,能够透明地进行优化。同时DGL采用了框架中立的设计,允许用户在多个深度学习框架中轻松移植和利用现有的组件。我们的评估表明,在各种基准测试中,DGL在速度和内存消耗方面都显著优于其他流行的面向GNNs的框架,并且对于小规模工作负载的开销很小。