中文

LazyGNN:基于惰性传播的大规模图神经网络

机器学习 2023-06-02 v2

摘要

近期的研究已表明,通过更深的图神经网络(GNNs)捕获图中的长距离依赖具有诸多优势。但更深的 GNN 由于大规模图中的邻域爆炸问题,面临着长期的可扩展性挑战。在本工作中,我们提出通过更浅的模型而非更深的模型来捕获图中的长距离依赖,从而得到一个高效得多的图表示学习模型 LazyGNN。此外,我们证明了 LazyGNN 与现有的可扩展方法(如采样方法)兼容,可通过开发小批量 LazyGNN 实现进一步加速。综合实验证明了其在大规模基准上的优越预测性能与可扩展性。LazyGNN 的实现见 https://github.com/RXPHD/Lazy_GNN。

关键词

引用

@article{arxiv.2302.01503,
  title  = {LazyGNN: Large-Scale Graph Neural Networks via Lazy Propagation},
  author = {Rui Xue and Haoyu Han and MohamadAli Torkamani and Jian Pei and Xiaorui Liu},
  journal= {arXiv preprint arXiv:2302.01503},
  year   = {2023}
}