arXiv:2607.02603cs.DCcs.LG2026-07

用GPU加速图神经网络表达力分析,首次实现百亿边级图的快速计算

Scaling Weisfeiler-Leman Expressiveness Analysis to Massive Graphs with GPUs

  • 基于线性代数重构1-WL稳定着色,设计随机化精炼算法
  • 提出分块处理方案,在保证正确性前提下实现并行加速
  • 在300亿边大规模图上速度提升100倍,适合超大规模图研究

1-WL测试的稳定着色是图神经网络表达能力的上限基准。然而传统算法存在两大瓶颈:一是固有串行性难以利用现代并行硬件;二是全局算法需将整图加载内存,限制了真实场景应用。本文基于1-WL的线性代数视角,提出两个关键贡献:(i) 具有严格概率保证的随机化精炼算法;(ii) 保持正确性的分块处理机制,可将图分解为独立处理子图,并确保返回原图的稳定着色。该方法天然适配GPU高效计算。数值实验表明,基于CUDA的实现相较经典CPU分区精炼算法提速达两个数量级,首次成功在超过300亿边的网络规模图上完成稳定着色计算,而传统方法因超时或内存不足失败。

原文摘要 · Abstract (English)

The stable coloring of the Weisfeiler-Leman (1-WL) test is a cornerstone of Graph Neural Networks because it provides an upper bound to the expressive power of message-passing architectures. Unfortunately, computing it presents two fundamental bottlenecks. First, classic algorithms are inherently sequential and cannot exploit modern massively parallel hardware. Second, these are \emph{global} algorithms, i.e., they require availability in memory of the full graph, severely limiting applicability to real-world instances. We leverage a linear-algebraic interpretation of 1-WL stable coloring and introduce two key contributions: (i)~a randomized refinement algorithm with tight probabilistic guarantees and (ii)~a correctness-preserving batching scheme that decomposes the graph into independently processable subgraphs while provably returning a stable coloring of the original graph. This approach maps directly to GPU-efficient primitives. In numerical experiments, our CUDA implementation delivers speedups up to two orders of magnitude over classical CPU-based partition refinement and, for the first time, successfully computes stable colorings on web-scale graphs with over 30 billion edges, where CPU baselines time out or fail.

图神经网络并行计算GPU加速大规模图

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