CoDG-ReRAM: An Algorithm-Hardware Co-design to Accelerate Semi-Structured GNNs on ReRAM

Yixuan Luo, Payman Behnam, Kiran Thorat, Zhuo Liu, Hongwu Peng, Shaoyi Huang, Shu Zhou, Omer Khan, Alexey Tumanov, Caiwen Ding, Tong Geng

Research output: Chapter in Book/Report/Conference proceedingConference contribution

5 Scopus citations

Abstract

Graph Neural Networks (GCNs) have attracted wide attention and are applied to the real world. However, due to the ever-growing graph data with significant irregularities, off-chip communication with poor data locality has become the major bottleneck hurdling the development of GCNs. Fortunately, recent works demonstrate Resistive Random Access Memory (ReRAM) has the potential to perform inherently parallel in-situ computation of Matrix-Vector Multiplication (MVM) in the analog regime fundamentally breaking the communication bottleneck.Inspired by this observation, we propose a novel ReRAM-based GCN acceleration co-design (i.e. algorithm-hardware) framework, CoDG-ReRAM, that can deliver real-time GCN inference with high accuracy. On the algorithm side, we propose a novel model optimization pipeline that simultaneously and efficiently sparsifies and regularizes both graph and parameter matrices in GCNs and creates ReRAM-friendly models. On the hardware side, we take advantage of the software optimization results to provide a more systematic mapping scheme and in-crease computation efficiency to have an energy-efficient ReRAM-based GCN acceleration with low latency. Experimental results show that the proposed work improves performance and energy efficiency by 4× and 5.1 × respectively over SOTA ReRAM-based accelerators of GCNs with a negligible accuracy loss.

Original languageEnglish (US)
Title of host publicationProceedings - 2022 IEEE 40th International Conference on Computer Design, ICCD 2022
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages280-289
Number of pages10
ISBN (Electronic)9781665461863
DOIs
StatePublished - 2022
Externally publishedYes
Event40th IEEE International Conference on Computer Design, ICCD 2022 - Olympic Valley, United States
Duration: Oct 23 2022Oct 26 2022

Publication series

NameProceedings - IEEE International Conference on Computer Design: VLSI in Computers and Processors
Volume2022-October
ISSN (Print)1063-6404

Conference

Conference40th IEEE International Conference on Computer Design, ICCD 2022
Country/TerritoryUnited States
CityOlympic Valley
Period10/23/2210/26/22

Bibliographical note

Publisher Copyright:
© 2022 IEEE.

Keywords

  • Computer Architecture
  • Graph Neural Network
  • Processing In Memory
  • Resistive Random Access Memory

Fingerprint

Dive into the research topics of 'CoDG-ReRAM: An Algorithm-Hardware Co-design to Accelerate Semi-Structured GNNs on ReRAM'. Together they form a unique fingerprint.

Cite this