Dual-Metric Partitioning with Adaptive Kernel Execution for Efficient GCN Acceleration

2026-09-01

AI Summary

本文针对GCN在GPU上运行时的不规则内存访问和负载不平衡问题，通过双度量图分割和自适应内核执行方法来优化，提高了处理大规模图数据的效率。

Abstract

Graph Convolutional Networks (GCNs) are widely used for large graph-structured data, including social, citation, and e-commerce networks, but their deployment is constrained by irregular memory access and severe GPU workload imbalance. These challenges arise in two dimensions: width imbalance from power-law degree distributions and depth imbalance from heterogeneous neighborhood connectivity.We present DualGCN, a GPU acceleration framework addressing both dimensions through dual-metric graph partitioning and adaptive kernel execution. DualGCN combines node degree, reflecting aggregation width, with neighborhood density estimated by anonymous random walks, capturing multihop connectivity and access depth. This hybrid workload metric enables connectivity-aware partitioning of large graphs into sparse and dense regions while reducing workload imbalance from linear to logarithmic complexity. DualGCN then selects partition-specific execution strategies: sparse partitions use warp-level parallelism and coalesced memory access, whereas dense partitions exploit instruction-level parallelism to hide latency and improve GPU utilization. Experiments on twelve real-world graph datasets show that DualGCN consistently accelerates GCN computation, achieving average speedups of 2.53x, 3.8x, and 2.13x over cuSPARSE, GNNAdvisor, and ACCEL, respectively. These results demonstrate that jointly optimizing graph partitioning and kernel execution provides an effective solution for processing large-scale graph and socialnetwork workloads.

Problem

• Graph Convolutional Networks
• workload imbalance
• irregular memory access
• graph partitioning
• GPU acceleration

Innovation

• Dual-Metric Partitioning
• Adaptive Kernel Execution
• Connectivity-Aware Partitioning
• Workload Imbalance Reduction