paperbot · PL 论文追踪

全部论文 待读 收藏

P2 Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning

Betty:通过批级图划分支持大规模 GNN 训练
ASPLOS Proceedings of the 28th ACM International Conference on Architectural Support for Programming Languages and Operating Systems, Volume 22023匹配度 2/10 · 该论文属于 GNN 训练系统与显存优化,核心是批级图划分和内存管理;与用户关注的静态分析、Agentic/LLM 程序分析、浮点程序分析、贝叶斯程序分析均无直接关联,仅在“运行时与内存管理”标签上稍有交集。引用 43 inbox
Shuangyan Yang, Minjia Zhang, Wenqian Dong, Dong Li
crossref: 2026-09-12

AI 速览

一句话:提出 Betty,用批级图划分、冗余嵌入图划分和内存感知划分缓解 GNN 训练显存瓶颈,支持单 GPU 大规模训练。

图神经网络(GNN)可通过更先进的聚合器、更深的聚合深度和更大的采样率提升效果,但这会显著增加显存占用,常超出 GPU 容量。与 DNN 训练不同,GNN 的 mini-batch 中输入特征与输出标签之间存在复杂依赖,使批级划分困难。为此,论文提出 Betty,一种基于批级图划分的可扩展 GNN 训练方法,并引入冗余嵌入图(REG)划分与内存感知划分两项技术,以缓解分区间的冗余和负载不均衡。作者在大规模真实数据集上评估,结果表明 Betty 能显著缓解显存瓶颈,在单 GPU 上支持更深聚合、更高采样率、更大训练 batch 及更先进聚合器的 GNN 训练。

问题
GNN 训练通过更复杂聚合器、更深聚合深度、更大采样率提升性能,但显存占用常超出 GPU 容量,且 GNN mini-batch 中输入特征与输出标签依赖复杂,使批级划分困难。
方法
Betty 采用批级图划分,并引入冗余嵌入图(REG)划分与内存感知划分,分别缓解分区间的冗余和负载不均衡。
评估
在多个大规模真实图数据集上评估,显示其显著缓解显存瓶颈,可用单 GPU 支持更深聚合、更高采样率、更大 batch 及更复杂聚合器的 GNN 训练。
原文摘要(Abstract)

The Graph Neural Network (GNN) is showing outstanding results in improving the performance of graph-based applications. Recent studies demonstrate that GNN performance can be boosted via using more advanced aggregators, deeper aggregation depth, larger sampling rate, etc. While leading to promising results, the improvements come at a cost of significantly increased memory footprint, easily exceeding GPU memory capacity. In this paper, we introduce a method, Betty, to make GNN training more scalable and accessible via batch-level partitioning. Different from DNN training, a mini-batch in GNN has complex dependencies between input features and output labels, making batch-level partitioning difficult. Betty introduces two noveltechniques, redundancy-embedded graph (REG) partitioning and memory-aware partitioning, to effectively mitigate the redundancy and load imbalances issues across the partitions. Our evaluation of large-scale real-world datasets shows that Betty can significantly mitigate the memory bottleneck, enabling scalable GNN training with much deeper aggregation depths, larger sampling rate, larger training batch sizes, together with more advanced aggregators, with a few as a single GPU.

链接与引用

DOI 原文 · PDF(开放获取) ·

BibTeX
@inproceedings{paperbot9991,
  title = {Betty: Enabling Large-Scale GNN Training with Batch-Level Graph Partitioning},
  author = {Shuangyan Yang and Minjia Zhang and Wenqian Dong and Dong Li},
  booktitle = {ASPLOS},
  year = {2023},
  doi = {10.1145/3575693.3575725}
}