arXiv:2507.14570cs.LGcs.AI2025-07被引 4

单卡10小时训练百亿边图,性能提升超13%。

LPS-GNN : Deploying Graph Neural Networks on Graphs with 100-Billion Edges

  • 设计轻量分区算法LPMetis,缓解邻居爆炸问题。
  • 在真实场景中实现8.24%-13.89%的性能提升。
  • 兼容多种GNN模型,适合大规模工业部署。

图神经网络(GNN)在图挖掘任务中表现强劲,但现有可扩展方案难以兼顾效率与精度。这主要源于迭代消息传递机制带来的高计算开销和大量GPU内存需求,尤其在大规模图中面临邻居爆炸问题。本文提出LPS-GNN框架,可在单张GPU上于10小时内完成百亿边图的表示学习,并在用户获取场景中提升13.8%。我们分析了现有图划分方法,设计了更优的图划分算法LPMetis,其在多个指标上优于当前SOTA。此外,提出子图增强策略以提升预测性能。该框架具备良好兼容性,支持多种GNN算法。已在腾讯平台成功部署,测试于公开及真实数据集,在线上应用中相较SOTA模型性能提升8.24%至13.89%。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) have emerged as powerful tools for various graph mining tasks, yet existing scalable solutions often struggle to balance execution efficiency with prediction accuracy. These difficulties stem from iterative message-passing techniques, which place significant computational demands and require extensive GPU memory, particularly when dealing with the neighbor explosion issue inherent in large-scale graphs. This paper introduces a scalable, low-cost, flexible, and efficient GNN framework called LPS-GNN, which can perform representation learning on 100 billion graphs with a single GPU in 10 hours and shows a 13.8% improvement in User Acquisition scenarios. We examine existing graph partitioning methods and design a superior graph partition algorithm named LPMetis. In particular, LPMetis outperforms current state-of-the-art (SOTA) approaches on various evaluation metrics. In addition, our paper proposes a subgraph augmentation strategy to enhance the model's predictive performance. It exhibits excellent compatibility, allowing the entire framework to accommodate various GNN algorithms. Successfully deployed on the Tencent platform, LPS-GNN has been tested on public and real-world datasets, achieving performance lifts of 8. 24% to 13. 89% over SOTA models in online applications.

图神经网络大规模图高效训练工业部署

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。