Understanding and Bridging the Gaps in Current GNN Performance Optimizations
Graph Neural Network (GNN) has recently drawn a rapid increase of interest in many domains for its effectiveness in learning over graphs. Maximizing its performance is essential for many tasks, but remains preliminarily understood. In this work, we provide an in-depth examination of the state-of-the-art GNN frameworks, revealing five major gaps in the current frameworks in optimizing GNN performance, especially in handling the special complexities of GNN over traditional graph or DNN operations. Based on the insights, we put together a set of optimizations to fill the gaps. These optimizations leverage the start-of-the-art GPU optimization techniques and tailor them to the special properties of GNN. Experimental results show that these optimizations achieve 1.37–15.5X performance improvement over the state-of-the-art frameworks on various GNN models.
Mon 1 MarDisplayed time zone: Eastern Time (US & Canada) change
12:30 - 13:30 | |||
12:30 15mTalk | Understanding and Bridging the Gaps in Current GNN Performance Optimizations Main Conference Kezhao Huang Tsinghua University, Jidong Zhai Tsinghua University, Zhen Zheng Alibaba Group, Youngmin Yi University of Seoul, Xipeng Shen North Carolina State University Link to publication | ||
12:45 15mTalk | A Fast Work-Efficient SSSP Algorithm for GPUs Main Conference Kai Wang University of Texas at Austin, Donald Fussell University of Texas at Austin, Calvin Lin University of Texas at Austin Link to publication | ||
13:00 15mTalk | ShadowVM: Accelerating Data Plane for Data Analytics with Bare Metal CPUs and GPUs Main Conference Zhifang Li East China Normal University, Mingcong Han East China Normal University, Shangwei Wu East China Normal University, Chuliang Weng East China Normal University Link to publication | ||
13:15 15mTalk | BiPart: A Parallel and Deterministic Hypergraph Partitioner Main Conference Sepideh Maleki The University of Texas at Austin, Udit Agarwal UT Austin, Martin Burtscher Texas State University, Keshav Pingali The University of Texas at Austin Link to publication |