arXiv:2508.16769cs.LG2025-08被引 2

针对电路图神经网络训练慢的问题,提出新GPU加速方法。

DR-CircuitGNN: Training Acceleration of Heterogeneous Circuit Graph Neural Network on GPUs

  • 利用行稀疏感知动态激活和优化SpMM内核加速消息传递
  • 在三个电路数据集上实现前向传播3.51倍、反向传播4.09倍提速
  • 适合需要高效训练复杂电路模型的研究者与工业开发者

集成电路设计规模与复杂度不断提升,给电子设计自动化(EDA)带来挑战。图神经网络(GNNs)因电路天然可表示为图而成为潜在解决方案。然而,传统GNN难以捕捉完整设计复杂性。异构图神经网络(HGNNs)通过同时建模拓扑关系与几何特征,能更好理解电路图,但其串行模块化消息传递机制导致计算开销更大,成为性能瓶颈。本文提出DR-CircuitGNN,通过行稀疏感知的动态ReLU与优化的SpMM内核,在GPU上加速HGNN训练。进一步提出并行优化策略:利用多线程CPU初始化与多cudaStreams并发执行,最大化CPU-GPU重叠。实验显示,在三个典型CircuitNet设计(小、中、大)上,前向传播最高提速3.51倍,反向传播达4.09倍;在全尺寸CircuitNet与采样版Mini-CircuitNet上,相比官方DGL cuSPARSE实现,速度提升最高达2.71倍,且相关性分数与误差率几乎无损失。

原文摘要 · Abstract (English)

The increasing scale and complexity of integrated circuit design have led to increased challenges in Electronic Design Automation (EDA). Graph Neural Networks (GNNs) have emerged as a promising approach to assist EDA design as circuits can be naturally represented as graphs. While GNNs offer a foundation for circuit analysis, they often fail to capture the full complexity of EDA designs. Heterogeneous Graph Neural Networks (HGNNs) can better interpret EDA circuit graphs as they capture both topological relationships and geometric features. However, the improved representation capability comes at the cost of even higher computational complexity and processing cost due to their serial module-wise message-passing scheme, creating a significant performance bottleneck. In this paper, we propose DR-CircuitGNN, a fast GPU kernel design by leveraging row-wise sparsity-aware Dynamic-ReLU and optimizing SpMM kernels during heterogeneous message-passing to accelerate HGNNs training on EDA-related circuit graph datasets. To further enhance performance, we propose a parallel optimization strategy that maximizes CPU-GPU concurrency by concurrently processing independent subgraphs using multi-threaded CPU initialization and GPU kernel execution via multiple cudaStreams. Our experiments show that on three representative CircuitNet designs (small, medium, large), the proposed method can achieve up to 3.51x and 4.09x speedup compared to the SOTA for forward and backward propagation, respectively. On full-size CircuitNet and sampled Mini-CircuitNet, our parallel design enables up to 2.71x speed up over the official DGL implementation cuSPARSE with negligible impact on correlation scores and error rates.

图神经网络GPU加速电路设计消息传递

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